feat: add support for importing cli module

seqeralabs · Nov 2, 2023 · d0d0e5f · d0d0e5f
1 parent fafcea2
commit d0d0e5f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/seqerakit/cli.py b/seqerakit/cli.py
@@ -19,6 +19,7 @@
 """
 import argparse
 import logging
+import sys
 
 from pathlib import Path
 from seqerakit import seqeraplatform, helper, overwrite
@@ -122,7 +123,10 @@ def handle_block(self, block, args, destroy=False):
             logger.error(f"Unrecognized resource block in YAML: {block}")
 
 
-def main():
+def main(args=None):
+    if args is None:
+        args = sys.argv[1:]
+
     options = parse_args()
     logging.basicConfig(level=options.log_level)