Add basic auth header

[sdc.git] / catalog-be / src / main / resources / scripts / sdcBePy / tosca / main.py

diff --git a/catalog-be/src/main/resources/scripts/sdcBePy/tosca/main.py b/catalog-be/src/main/resources/scripts/sdcBePy/tosca/main.py
index 565ce7e..78032a4 100644 (file)
--- a/catalog-be/src/main/resources/scripts/sdcBePy/tosca/main.py
+++ b/catalog-be/src/main/resources/scripts/sdcBePy/tosca/main.py
@@ -11,7 +11,7 @@ from sdcBePy.common.sdcBeProxy import SdcBeProxy
 def usage():
     print(sys.argv[0],
           '[-s <scheme> | --scheme=<scheme> ] [-i <be host> | --ip=<be host>] [-p <be port> | '
-          '--port=<be port> ] [-u <user userId> | --user=<user userId> ] [-d <true|false> | '
+          '--port=<be port> ] --header=<header> ] [-u <user userId> | --user=<user userId> ] [-d <true|false> | '
           '--debug=<true|false>] [-v <true|false> | --updateVersion=<true|false>]')
 
 
@@ -28,6 +28,7 @@ def parse_param():
 
     parser.add_argument('--ip', "-i")
     parser.add_argument('--port', "-p")
+    parser.add_argument('--header')
     parser.add_argument('--adminUser', "-a")
     parser.add_argument('--https', action='store_true')
     parser.add_argument('--updateVersion', action='store_false')
@@ -36,14 +37,14 @@ def parse_param():
     args, _ = parser.parse_known_args()
 
     return [args.conf, 'https' if args.https else 'http',
-            args.ip, args.port, args.adminUser, args.updateVersion,
+            args.ip, args.port, args.header, args.adminUser, args.updateVersion,
             args.debug]
 
 
 def get_args():
     print('Number of arguments:', len(sys.argv), 'arguments.')
 
-    conf_path, scheme, be_host, be_port, admin_user, update_version, debug = parse_param()
+    conf_path, scheme, be_host, be_port, header, admin_user, update_version, debug = parse_param()
     defaults = load_be_config(conf_path)
 
     # Use defaults if param not provided by the user
@@ -54,22 +55,26 @@ def get_args():
     if admin_user is None:
         admin_user = defaults["adminUser"]
 
-    print('scheme =', scheme, ',be host =', be_host, ', be port =', be_port, ', user =', admin_user,
+    if header is None:
+        print('scheme =', scheme, ',be host =', be_host, ', be port =', be_port, ', user =', admin_user,
           ', debug =', debug, ', update_version =', update_version)
+    else:
+        print('scheme =', scheme, ',be host =', be_host, ', be port =', be_port, ', header =', header, ', user =', admin_user,
+              ', debug =', debug, ', update_version =', update_version)
 
     init_properties(defaults["retryTime"], defaults["retryAttempt"], defaults["resourceLen"])
-    return scheme, be_host, be_port, admin_user, update_version, debug
+    return scheme, be_host, be_port, header, admin_user, update_version, debug
 
 
 def parse_and_create_proxy():
-    scheme, be_host, be_port, admin_user, update_version, debug = get_args()
+    scheme, be_host, be_port, header, admin_user, update_version, debug = get_args()
 
     if debug is False:
         print('Disabling debug mode')
         logger.debugFlag = debug
 
     try:
-        sdc_be_proxy = SdcBeProxy(be_host, be_port, scheme, admin_user, debug=debug)
+        sdc_be_proxy = SdcBeProxy(be_host, be_port, header, scheme, admin_user, debug=debug)
     except AttributeError:
         usage()
         sys.exit(3)