Configuration/Setting module added

scripts/ucloud

@@ -3,37 +3,60 @@
 import argparse
 import logging
 import importlib
-import sys
 import os
 import multiprocessing as mp
+import sys
+
+from ucloud.configure.main import update_config, configure_parser
 
 
-COMMANDS = ['api', 'scheduler', 'host', 'filescanner', 'imagescanner', 'metadata']
+def exception_hook(exc_type, exc_value, exc_traceback):
+        logger.error(
+            "Uncaught exception",
+            exc_info=(exc_type, exc_value, exc_traceback)
+        )
+        print(exc_type, exc_value)
 
-if __name__ == "__main__":
+
+if __name__ == '__main__':
     logging.basicConfig(level=logging.DEBUG,
                     format='%(pathname)s:%(lineno)d -- %(levelname)-8s %(message)s',
                     filename='/var/log/ucloud.log', filemode='a')
-
     logger = logging.getLogger("ucloud")
 
-    arg_parser = argparse.ArgumentParser(prog='ucloud',
-                                         description='Open Source Cloud Management Software')
-    arg_parser.add_argument('-c', '--conf-dir', help="Configuration directory")
-    arg_parser.add_argument('component', choices=COMMANDS)
-    arg_parser.add_argument('component_args', nargs='*')
+    sys.excepthook = exception_hook
+    mp.set_start_method('spawn')
+
+    arg_parser = argparse.ArgumentParser()
+    subparsers = arg_parser.add_subparsers(dest="command")
+    
+    api_parser = subparsers.add_parser("api")
+    
+    host_parser = subparsers.add_parser("host")
+    host_parser.add_argument("--hostname", required=True)
+
+    scheduler_parser = subparsers.add_parser("scheduler")
+
+    filescanner_parser = subparsers.add_parser("filescanner")
+
+    imagescanner_parser = subparsers.add_parser("imagescanner")
+
+    metadata_parser = subparsers.add_parser("metadata")
+
+    config_parser = subparsers.add_parser("configure")
+    configure_parser(config_parser)
     args = arg_parser.parse_args()
 
-    if args.conf_dir:
-        os.environ['UCLOUD_CONF_DIR'] = args.conf_dir
+    if not args.command:
+        arg_parser.print_help()
+    else:
+        arguments = vars(args)
+        try:
+            name = arguments.pop('command')
+            mod = importlib.import_module("ucloud.{}.main".format(name))
+            main = getattr(mod, "main")
+            main(**arguments)
 
-    try:
-        mp.set_start_method('spawn')
-        name = args.component
-        mod = importlib.import_module("ucloud.{}.main".format(name))
-        main = getattr(mod, "main")
-        main(*args.component_args)
-
-    except Exception as e:
-        logger.exception(e)
-        print(e)
+        except Exception as e:
+            logger.exception(e)
+            print(e)