Merge pull request #2 from tomer/patch-1

Allow running generator as Python Module in addition to CLI command

Author: eggplants

openapi_generator_cli/__init__.py

@@ -5,7 +5,7 @@
 import sys
 
 
-def run():
+def run(args=None):
     arguments = ["java"]
 
     if os.getenv("JAVA_OPTS"):
@@ -18,7 +18,18 @@ def run():
     )
     arguments.append(jar_path)
 
-    if len(sys.argv) > 1:
-        arguments.extend(sys.argv[1:])
+    if args and type(args) == list:
+        arguments.extend(args)
 
     subprocess.call(" ".join(arguments), shell=True)
+
+
+def cli():
+    args = []
+    if len(sys.argv) > 1:
+        args = sys.argv[1:]
+    run(args)
+
+
+if __name__ == "__main__":
+    cli()

setup.py

@@ -74,5 +74,5 @@ def run(self):
         "Programming Language :: Python :: Implementation :: PyPy",
     ],
     cmdclass={"upload": UploadCommand},
-    entry_points={"console_scripts": ["openapi-generator=openapi_generator_cli:run"]},
+    entry_points={"console_scripts": ["openapi-generator=openapi_generator_cli:cli"]},
 )