replbase 0.0.31__tar.gz → 0.0.33__tar.gz

{replbase-0.0.31 → replbase-0.0.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: replbase
-Version: 0.0.31
+Version: 0.0.33
 Summary: "Combination of other REPL tools into a reusable class that generates a REPL"
 License: MIT
 Author: Joseph Bochinski

{replbase-0.0.31 → replbase-0.0.33}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.31"
+version = "0.0.33"
 description = "\"Combination of other REPL tools into a reusable class that generates a REPL\""
 authors = [ "Joseph Bochinski <stirgejr@gmail.com>",]
 license = "MIT"

{replbase-0.0.31 → replbase-0.0.33}/replbase/repl_base.py

@@ -16,6 +16,7 @@ from __future__ import annotations
 
 import argparse
 import grp
+import inspect
 import math
 import os
 import pwd
@@ -418,6 +419,59 @@ class ReplBase:
         self.commands[cmd_name] = new_cmd
         return new_cmd
 
+    def gen_command(self, cmd_name: str) -> ReplCommand:
+        if not hasattr(self, cmd_name):
+            return
+
+        cmd: Callable = getattr(self, cmd_name)
+        anno = cmd.__annotations__
+        sig = inspect.signature(cmd)
+        parms = sig.parameters
+        help_text = cmd.__doc__
+        repl_cmd = self.add_command(
+            cmd.__name__,
+            cmd,
+            help_txt=help_text,
+            use_parser=True,
+            description=help_text,
+        )
+
+        if not parms:
+            return repl_cmd
+
+        for arg, parm in parms.items():
+            arg_type = anno.get(arg, str)
+            def_val = parm.default
+
+            optional = def_val != parm.empty
+            flag_name = arg.replace("_", "-")
+
+            if optional:
+                flag_name = "--" + flag_name
+
+            cmd_init = {}
+
+            if arg_type == bool:
+                cmd_init["action"] = (
+                    "store_false" if def_val is True else "store_true"
+                )
+
+            if arg_type != str:
+                cmd_init["type"] = arg_type
+
+            if isinstance(arg_type, type) and isinstance(def_val, arg_type):
+                cmd_init["default"] = def_val
+
+            if isinstance(arg_type, list):
+                cmd_init["nargs"] = "+"
+
+            if cmd_init:
+                repl_cmd.parser.add_argument(flag_name, **cmd_init)
+            else:
+                repl_cmd.parser.add_argument(flag_name)
+
+        return repl_cmd
+
     def setup_cmds(self, *cmd_names: list[str]) -> None:
         """Automatically configure commands based on the provided names
             The ReplCommand objects are populated based on function meta
@@ -437,6 +491,10 @@ class ReplBase:
             name = func.__name__
             self.add_command(name, func, help_txt=help_text)
 
+    def setup_cmds_2(self, *cmd_names: list[str]):
+        for name in cmd_names:
+            self.gen_command(name)
+
     def warn(self, msg: str) -> None:
         """Print a message to the REPL preformatted as a warning"""