Conquer the land of arguing snakes

gooey/__main__.py

@@ -1,12 +1,13 @@
-# '''
-# Delegates arguments to the main Gooey runner
-#
-# For use when run directly from command line with the -m (module) flag:
-#
-#   e.g. $ python -m gooey
-#
-# '''
-#
-# from gooey import application
-#
-# application.main()
+'''
+Delegates arguments to the main Gooey runner.
+
+For use when run directly from command line with the -m (module) flag:
+
+  e.g. $ python -m gooey
+
+'''
+
+from gooey.cli import main
+
+if __name__ == '__main__':
+    main()

gooey/cli.py

@@ -0,0 +1,41 @@
+'''The command line interface.'''
+
+import sys
+from importlib import import_module
+try:
+    from importlib.metadata import entry_points
+except ImportError:
+    from importlib_metadata import entry_points
+
+from gooey import Gooey
+
+
+def main(args=None):
+    args = args or sys.argv[1:]
+    if not args:
+        # without args we assume we want to run gooey on gooey itself
+        # TODO: rewrite using argparse
+        script_name = "gooey"
+    else:
+        script_name = args[0]
+    scripts = {script.name: script for script in entry_points().get('console_scripts')}
+    if script_name in scripts:
+        # a valid script was passed
+        script = scripts[script_name]
+        module_path = script.module
+        function_name = script.attr
+        prog = script_name
+    elif ':' in script_name:
+        # the path to a function was passed
+        module_path, function_name = args[0].split(':')
+        prog = module_path.split('.', 1)[0]
+    else:
+        print('usage: gooey [SCRIPT | MODULE:FUNCTION] [-- SCRIPT_ARGS...]', file=sys.stderr)
+        sys.exit(1)
+    if len(args) > 1:
+        if args[1] == '--':
+            del args[1]
+    sys.argv = [prog, *args[1:]]
+    module = import_module(module_path)
+    function = getattr(module, function_name)
+    return Gooey(function, use_cmd_args=True)()

setup.py

@@ -16,7 +16,8 @@
     're-wx>=0.0.9',
     'typing-extensions==3.10.0.2',
     'wxpython>=4.1.0',
-    'dataclasses>=0.8; python_version < "3.7"'
+    'dataclasses>=0.8; python_version < "3.7"',
+    'importlib-metadata; python_version < "3.8"',
 ]
 
 setup(
@@ -33,6 +34,7 @@
     install_requires=deps,
     include_package_data=True,
     dependency_links = ["http://www.wxpython.org/download.php"],
+    entry_points={'console_scripts': ['gooey = gooey.cli:main']},
     classifiers = [
         'Development Status :: 4 - Beta',
         'Intended Audience :: Developers',