replbase 0.0.65__tar.gz → 0.0.66__tar.gz

{replbase-0.0.65 → replbase-0.0.66}/PKG-INFO

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

{replbase-0.0.65 → replbase-0.0.66}/pyproject.toml

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

@@ -115,6 +115,9 @@ class CommandMeta:
     flag_names: dict[str, list[str]] = field(default_factory=dict)
     """ Flag names to use for the ReplCommand.parser arguments embedded in the docstring """
 
+    used_flags: set[str] = field(default_factory=set)
+    """ Reference to keep track of unique flag names """
+
     def __post_init__(self) -> None:
 
         if not self.func:
@@ -253,7 +256,7 @@ class CommandMeta:
             try:
                 arg_type = self.type_hints.get(arg, str)
 
-                flag_name = self.flag_names.get(arg)
+                flag_names = [self.flag_names.get(arg)]
 
                 cmd_init = {"help": self.args.get(arg)}
 
@@ -278,10 +281,10 @@ class CommandMeta:
                     cmd_init["default"] = parm.default
 
                 if cmd_init:
-                    repl_cmd.parser.add_argument(flag_name, **cmd_init)
+                    repl_cmd.parser.add_argument(flag_names, **cmd_init)
 
                 else:
-                    repl_cmd.parser.add_argument(flag_name)
+                    repl_cmd.parser.add_argument(flag_names)
             except TypeError as e:
                 print(
                     f"[ERROR]: {e} for:\n{self.func.__name__}.{arg} with {arg_type} and {parm.default}"
@@ -318,7 +321,14 @@ class CommandMeta:
             if self.is_arg_optional(arg):
                 flag = "--" + flag.replace("_", "-")
 
-            self.flag_names[arg] = flag
+            self.flag_names[arg] = [flag]
+            char = flag[0]
+            if not char.lower() in self.used_flags:
+                self.flag_names[arg].append(f"-{char.lower()}")
+                self.used_flags.add(char.lower())
+            elif char.upper() in self.used_flags:
+                self.flag_names[arg].append(f"-{char.upper()}")
+                self.used_flags.add(char.upper())
 
     def parse_lines(self) -> None:
         """Parse the lines of the docstring into the appropriate sections"""