replbase 0.0.55__tar.gz → 0.0.57__tar.gz

{replbase-0.0.55 → replbase-0.0.57}/PKG-INFO

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

{replbase-0.0.55 → replbase-0.0.57}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.55"
+version = "0.0.57"
 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.55 → replbase-0.0.57}/replbase/cmd_meta.py

@@ -175,9 +175,13 @@ class CommandMeta:
             if doc:
                 return doc
 
-    def gen_command(self) -> ReplCommand:
+    def gen_command(self, hyphenate: bool = True) -> ReplCommand:
         """Generate a ReplCommand object from the provided function
 
+        Args:
+            hyphenate (bool, optional): Use hyphens for flag names.
+                Defaults to True.
+
         Raises:
             ValueError: If no registry command is provided
 
@@ -191,8 +195,14 @@ class CommandMeta:
         help_text = self.summary
         full_help = self.summary + self.additional
 
+        cmd_name = (
+            self.func.__name__.replace("_", "-")
+            if hyphenate
+            else self.func.__name__
+        )
+
         repl_cmd = self.register_cmd(
-            cmd_name=self.func.__name__,
+            cmd_name=cmd_name,
             cmd_func=self.func,
             help_txt=help_text,
             use_parser=True,

{replbase-0.0.55 → replbase-0.0.57}/replbase/repl_base.py

@@ -209,6 +209,8 @@ class ReplBase:
 
     session: PromptSession = None
 
+    cmd_defs: dict[str, CommandMeta] = field(default_factory=dict)
+
     def __post_init__(self) -> None:
         if isinstance(self.commands, dict):
             for cmd_name, cmd in self.commands.items():
@@ -471,10 +473,11 @@ class ReplBase:
         if not callable(cmd):
             return
 
-        if hyphenate:
-            cmd_name = cmd_name.replace("_", "-")
-
-        return CommandMeta(func=cmd, register_cmd=self.add_command).gen_command()
+        cmd_meta = CommandMeta(func=cmd, register_cmd=self.add_command).gen_command(
+            hyphenate=hyphenate
+        )
+        self.cmd_defs[cmd_name] = cmd_meta
+        return cmd_meta
 
     def setup_cmds(self, *cmd_names: list[str]) -> None:
         """Automatically configure commands based on the provided names