replbase 0.0.56__tar.gz → 0.0.58__tar.gz

{replbase-0.0.56 → replbase-0.0.58}/PKG-INFO

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

{replbase-0.0.56 → replbase-0.0.58}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.56"
+version = "0.0.58"
 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.56 → replbase-0.0.58}/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,9 +473,9 @@ class ReplBase:
         if not callable(cmd):
             return
 
-        return CommandMeta(func=cmd, register_cmd=self.add_command).gen_command(
-            hyphenate=hyphenate
-        )
+        cmd_meta = CommandMeta(func=cmd, register_cmd=self.add_command)
+        self.cmd_defs[cmd_name] = cmd_meta
+        return cmd_meta.gen_command(hyphenate=hyphenate)
 
     def setup_cmds(self, *cmd_names: list[str]) -> None:
         """Automatically configure commands based on the provided names