replbase 0.0.15__tar.gz → 0.0.16__tar.gz

{replbase-0.0.15 → replbase-0.0.16}/PKG-INFO

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

{replbase-0.0.15 → replbase-0.0.16}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "replbase"
-version = "0.0.15"
+version = "0.0.16"
 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.15 → replbase-0.0.16}/replbase/repl_base.py

@@ -20,7 +20,7 @@ import re
 import shlex
 
 from dataclasses import dataclass, field
-from typing import Callable, Literal
+from typing import Any, Callable, Literal
 
 from prompt_toolkit import PromptSession
 from prompt_toolkit.history import FileHistory
@@ -222,6 +222,26 @@ class ReplBase:
         if choice and choice > 0 and choice <= len(choices):
             return choices[choice - 1]
 
+    def input_choice_dict(self, prompt: str, choices: dict) -> Any:
+        """Similar to input_choice, but with more control over the options
+
+        Args:
+            prompt (str): String to print to the user
+            choices (dict): Dict of choices; keys will be the displayed options, values will be the associated return value
+
+        Returns:
+            Any: Selected value
+        """
+
+        keys = list(choices.keys())
+        for idx, choice in enumerate(keys):
+            self.print(f"[{idx+1}]: {choice}")
+
+        choice = self.input_int(prompt)
+        if choice and choice > 0 and choice <= len(keys):
+            key = keys[choice]
+            return choices[key]
+
     def debug(self, *args) -> None:
         """Print only if debug_enabled == True"""
         if self.debug_enabled: