Conquer the land of clicketing snakes

setup.py

@@ -2,8 +2,10 @@
 
 setup(
     name="rich-click",
+    entry_points={"console_scripts": ["rich-click = rich_click.cli:main"]},
     install_requires=[
         "click",
         "rich",
+        "importlib-metadata; python_version < '3.8'",
     ],
 )

src/rich_click/__init__.py

@@ -9,6 +9,8 @@
 __version__ = "1.1.0.dev0"
 
 from click import *
+from click import group as click_group
+from click import command as click_command
 from .rich_click import RichGroup
 from .rich_click import RichCommand
 
@@ -18,8 +20,6 @@ def group(*args, cls=RichGroup, **kwargs):
 
     Defines the group() function so that it uses the RichGroup class by default
     """
-    from click import group as click_group
-
     return click_group(*args, cls=cls, **kwargs)
 
 
@@ -28,6 +28,4 @@ def command(*args, cls=RichCommand, **kwargs):
 
     Defines the command() function so that it uses the RichCommand class by default
     """
-    from click import command as click_command
-
     return click_command(*args, cls=cls, **kwargs)

src/rich_click/__main__.py

@@ -0,0 +1,15 @@
+"""
+Entry-point module, in case you use `python -m rich_click`.
+
+Why does this file exist, and why `__main__`? For more info, read:
+
+- https://www.python.org/dev/peps/pep-0338/
+- https://docs.python.org/3/using/cmdline.html#cmdoption-m
+"""
+
+import sys
+
+from rich_click.cli import main
+
+if __name__ == "__main__":
+    sys.exit(main(sys.argv[1:]))

src/rich_click/cli.py

@@ -0,0 +1,47 @@
+"""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
+
+import click
+from rich_click import group, command
+
+
+def main(args=None):
+    args = args or sys.argv[1:]
+    if not args:
+        # without args we assume we want to run rich-click on itself
+        # TODO: rewrite using click
+        script_name = "rich-click"
+    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: rich-click [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:]]
+    # patch click before importing the program function
+    click.group = group
+    click.command = command
+    # import the program function
+    module = import_module(module_path)
+    function = getattr(module, function_name)
+    # simply run it: it should be patched as well
+    return function()