Refactoring the sdc-BE-init python scripts

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

diff --git a/catalog-be/src/main/resources/scripts/sdcBePy/users/run.py b/catalog-be/src/main/resources/scripts/sdcBePy/users/run.py
new file mode 100755 (executable)
index 0000000..b933afe
--- /dev/null
+++ b/catalog-be/src/main/resources/scripts/sdcBePy/users/run.py
@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+
+import json
+import os
+import time
+from argparse import ArgumentParser
+
+from sdcBePy.common.bColors import BColors
+from sdcBePy.common.healthCheck import check_backend, RETRY_ATTEMPTS
+from sdcBePy.common.sdcBeProxy import SdcBeProxy
+
+colors = BColors()
+
+
+def load_users(conf_path):
+    with open(conf_path, 'r') as f:
+        return json.load(f)
+
+
+def be_user_init(be_ip, be_port, protocol, conf_path):
+    sdc_be_proxy = SdcBeProxy(be_ip, be_port, protocol)
+    if check_backend(sdc_be_proxy, RETRY_ATTEMPTS):
+        users = load_users(conf_path)
+        for user in users:
+            if sdc_be_proxy.check_user(user['userId']) != 200:
+                result = sdc_be_proxy.create_user(user['firstName'],
+                                                  user['lastName'],
+                                                  user['userId'],
+                                                  user['email'],
+                                                  user['role'])
+                if result == 201:
+                    print('[INFO]: ' + user['userId'] +
+                          ' created, result: [' + str(result) + ']')
+                else:
+                    print('[ERROR]: ' + colors.FAIL + user['userId'] + colors.END_C +
+                          ' error creating , result: [' + str(result) + ']')
+            else:
+                print('[INFO]: ' + user['userId'] + ' already exists')
+    else:
+        print('[ERROR]: ' + time.strftime('%Y/%m/%d %H:%M:%S') + colors.FAIL
+              + 'Backend is DOWN :-(' + colors.END_C)
+        raise Exception("Cannot communicate with the backend!")
+
+
+def get_args():
+    parser = ArgumentParser()
+
+    parser.add_argument('-i', '--ip', required=True)
+    parser.add_argument('-p', '--port', required=True)
+    parser.add_argument('--https', action='store_true')
+    path = os.path.dirname(__file__)
+    parser.add_argument('--conf', default=os.path.join(path, 'data', 'users.json'))
+
+    args = parser.parse_args()
+
+    return [args.ip, args.port, 'https' if args.https else 'http', args.conf]
+
+
+def main():
+    be_user_init(*get_args())
+
+
+if __name__ == "__main__":
+    main()