replbase 0.0.43__tar.gz → 0.0.45__tar.gz

{replbase-0.0.43 → replbase-0.0.45}/PKG-INFO

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

{replbase-0.0.43 → replbase-0.0.45}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.43"
+version = "0.0.45"
 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.43 → replbase-0.0.45}/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,24 @@ 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
+        """
+
+        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: