No try: needed for pop/importlib/getattr

2020-01-10 11:35:04 +01:00 · 2020-01-10 11:35:04 +01:00 · d9dd6b48dc
parent b7596e071a
commit d9dd6b48dc
1 changed files with 7 additions and 3 deletions
--- a/scripts/uncloud
+++ b/scripts/uncloud
@ -73,10 +73,14 @@ if __name__ == '__main__':
        arguments = vars(args)
        print(arguments)
        print(etcd_parser)
+
+        name = arguments.pop('command')
+        mod = importlib.import_module('uncloud.{}.main'.format(name))
+        main = getattr(mod, 'main')
+
+        if component in ETCD_COMPONENTS:
+            import etcd3
        try:
-            name = arguments.pop('command')
-            mod = importlib.import_module('uncloud.{}.main'.format(name))
-            main = getattr(mod, 'main')
            main(arguments)
        except UncloudException as err:
            logger.error(err)