replbase 0.0.42__tar.gz → 0.0.44__tar.gz

{replbase-0.0.42 → replbase-0.0.44}/PKG-INFO

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

{replbase-0.0.42 → replbase-0.0.44}/pyproject.toml

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

@@ -119,6 +119,10 @@ class CommandMeta:
             else:
                 self.additional_lines.append(part)
 
+        for arg, param in self.sig_parms.items():
+            if arg not in self.args:
+                self.args[arg] = [self.get_param_str(param)]
+
         self.parse_lines()
         self.parse_flags()
 
@@ -310,9 +314,5 @@ class CommandMeta:
             elif cur_arg:
                 args[cur_arg].append(line)
 
-        for arg, param in self.sig_parms.items():
-            if arg not in args:
-                args[arg].append(self.get_param_str(param))
-
         arg_strs = {arg: "\n".join(lines) for arg, lines in args.items()}
         self.args.update(arg_strs)

{replbase-0.0.42 → replbase-0.0.44}/replbase/repl_base.py

@@ -26,6 +26,7 @@ import stat
 import time
 
 from dataclasses import dataclass, field
+from enum import Enum, EnumType
 from typing import Any, Callable, Literal, get_type_hints
 
 from prompt_toolkit import PromptSession
@@ -370,6 +371,25 @@ class ReplBase:
             key = keys[choice - 1]
             return choices[key]
 
+    def get_enum_val(self, prompt: str, enum_cls: EnumType) -> Enum:
+        """Prompt the user to select an enum value
+
+        Args:
+            prompt (str): Prompt to display
+            enum_cls (EnumType): Enum class to select from
+
+        Returns:
+            Enum: Selected enum value
+        """
+
+        if enum_cls:
+            options = dict(enum_cls.__members__)
+
+            if len(options) == 1:
+                return list(options.values())[0]
+            else:
+                return self.input_choice_dict(prompt, options)
+
     def debug(self, *args) -> None:
         """Print only if debug_enabled == True"""
         if self.debug_enabled: