Swift: make generator.run accept options

Author: redsun82

swift/codegen/lib/generator.py

@@ -3,26 +3,29 @@
 import argparse
 import logging
 import sys
+from typing import Set
 
 from . import options, render
 
 
-def _parse(tags):
+def _parse(tags: Set[str]) -> argparse.Namespace:
     parser = argparse.ArgumentParser()
     for opt in options.get(tags):
         opt.add_to(parser)
-    ret = parser.parse_args()
-    log_level = logging.DEBUG if ret.verbose else logging.INFO
-    logging.basicConfig(format="{levelname} {message}", style='{', level=log_level)
-    return ret
+    return parser.parse_args()
 
 
-def run(*modules):
+def run(*modules, **kwargs):
     """ run generation functions in specified in `modules`, or in current module by default
     """
     if modules:
-        opts = _parse({t for m in modules for t in m.tags})
+        if kwargs:
+            opts = argparse.Namespace(**kwargs)
+        else:
+            opts = _parse({t for m in modules for t in m.tags})
+        log_level = logging.DEBUG if opts.verbose else logging.INFO
+        logging.basicConfig(format="{levelname} {message}", style='{', level=log_level)
         for m in modules:
             m.generate(opts, render.Renderer())
     else:
-        run(sys.modules["__main__"])
+        run(sys.modules["__main__"], **kwargs)