replbase 0.0.6__tar.gz → 0.0.7__tar.gz

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

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

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

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

@@ -15,6 +15,7 @@
 
 import argparse
 import os
+import re
 import shlex
 
 from dataclasses import dataclass, field
@@ -39,6 +40,10 @@ ColorSystem = Literal["auto", "standard", "256", "truecolor", "windows"]
 # region Classes
 
 
+def is_num_str(val: str) -> bool:
+    return bool(re.match(r"^-?\d+(\.\d+)?$", s))
+
+
 @dataclass
 class ReplCommand:
     """Class definition for a provided CLI REPL command"""
@@ -184,6 +189,21 @@ class ReplBase:
         """Shortcut to console.input"""
         return self.console.input(*args)
 
+    def input_int(self, *args) -> int:
+        """Parse input as int"""
+
+        user_input = self.input(*args).strip().split(".")[0].strip()
+        if is_num_str(user_input):
+            return int(user_input)
+
+    def input_bool(self, *args, true_list: list[str] = None) -> int:
+        """Parse input as bool"""
+        true_list = true_list or ["y", "yes"]
+        true_list = [val.lower() for val in true_list]
+
+        user_input = self.input(*args)
+        return user_input.strip().lower() in true_list
+
     def debug(self, *args) -> None:
         """Print only if debug_enabled == True"""
         if self.debug_enabled: