replbase 0.0.7__tar.gz → 0.0.9__tar.gz

{replbase-0.0.7 → replbase-0.0.9}/PKG-INFO

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

{replbase-0.0.7 → replbase-0.0.9}/pyproject.toml

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

@@ -41,7 +41,7 @@ ColorSystem = Literal["auto", "standard", "256", "truecolor", "windows"]
 
 
 def is_num_str(val: str) -> bool:
-    return bool(re.match(r"^-?\d+(\.\d+)?$", s))
+    return bool(re.match(r"^-?\d+(\.\d+)?$", val))
 
 
 @dataclass
@@ -204,6 +204,14 @@ 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 debug(self, *args) -> None:
         """Print only if debug_enabled == True"""
         if self.debug_enabled: