replbase 0.0.8__tar.gz → 0.0.10__tar.gz

{replbase-0.0.8 → replbase-0.0.10}/PKG-INFO

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

{replbase-0.0.8 → replbase-0.0.10}/pyproject.toml

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

@@ -204,6 +204,22 @@ class ReplBase:
         user_input = self.input(*args)
         return user_input.strip().lower() in true_list
 
+    def input_prefer_int(self, *args) -> int | str:
+        """Parse input as int if possible, otherwise return the string"""
+
+        user_input = self.input(*args).strip().split(".")[0].strip()
+        if is_num_str(user_input):
+            return int(user_input)
+        return user_input
+
+    def input_choice(self, *args, choices: list[str]) -> str:
+        """Prompt for a choice from a list of options"""
+        for idx, choice in enumerate(choices):
+            self.print(f"[{idx+1}]: {choice}")
+        choice = self.input_int(*args)
+        if choice and choice > 0 and choice <= len(choices):
+            return choices[choice - 1]
+
     def debug(self, *args) -> None:
         """Print only if debug_enabled == True"""
         if self.debug_enabled: