rgwfuncs 0.0.78__py3-none-any.whl → 0.0.79__py3-none-any.whl

rgwfuncs/interactive_shell_lib.py

@@ -2,23 +2,41 @@ import os
 import atexit
 import code
 import readline  # Assuming this is imported elsewhere in your module
+import sys
 from typing import Dict, Any
 from .df_lib import *  # noqa: F401, F403, E402
 from .algebra_lib import *  # noqa: F401, F403, E402
 from .str_lib import *  # noqa: F401, F403, E402
 from .docs_lib import *  # noqa: F401, F403, E402
 
-def interactive_shell(local_vars: Dict[str, Any], shell_color: str = '\033[97m') -> None:
+class CustomConsole(code.InteractiveConsole):
+    """Custom console to make user input white while keeping prompts and output default."""
+    def raw_input(self, prompt=""):
+        # Print the prompt (>>>), then switch to white for input
+        sys.stdout.write(prompt)  # Prompt in default color
+        sys.stdout.write("\033[97m")  # Switch to white for input
+        sys.stdout.flush()
+        try:
+            line = input()  # User types in white
+        except EOFError:
+            raise  # Handle Ctrl+D gracefully
+        finally:
+            sys.stdout.write("\033[0m")  # Reset to default after input
+            sys.stdout.flush()
+        return line
+
+def interactive_shell(local_vars: Dict[str, Any], shell_color: str = '') -> None:
     """
     Launches an interactive prompt for inspecting and modifying local variables, making all methods
     in the rgwfuncs library available by default. Persists command history across sessions.
+    User input is displayed in white; other text uses the terminal's default color.
 
     Parameters:
         local_vars (dict): Dictionary of local variables to be available in the interactive shell.
-        shell_color (str): ANSI color code for shell output (default: bright white '\033[97m').
+        shell_color (str): ANSI color code for shell output (default: empty for terminal default).
     """
 
-    def setup_readline(shell_color: str = '\033[97m') -> None:
+    def setup_readline(shell_color: str = '') -> None:
         """Set up readline for history persistence with a given shell color."""
         HISTORY_FILE = os.path.expanduser("~/.rgwfuncs_shell_history")
         readline.set_history_length(1000)
@@ -27,7 +45,7 @@ def interactive_shell(local_vars: Dict[str, Any], shell_color: str = '\033[97m')
             try:
                 readline.read_history_file(HISTORY_FILE)
             except Exception as e:
-                print(f"{shell_color}Warning: Could not load history file: {e}\033[0m")
+                print(f"{shell_color}Warning: Could not load history file: {e}")
         atexit.register(readline.write_history_file, HISTORY_FILE)
 
     if not isinstance(local_vars, dict):
@@ -39,16 +57,16 @@ def interactive_shell(local_vars: Dict[str, Any], shell_color: str = '\033[97m')
     # Make imported functions available in the REPL
     local_vars.update(globals())
 
-    # Create interactive console with local context
-    console = code.InteractiveConsole(locals=local_vars)
+    # Set prompt to plain text (no color)
+    sys.ps1 = ">>> "
 
-    # Set the prompt to white explicitly
-    import sys
-    sys.ps1 = f"\033[97m>>> "  # White prompt
+    # Create custom console with local context
+    console = CustomConsole(locals=local_vars)
 
-    # Start interactive session with a custom banner and exit message
-    banner = f"{shell_color}Welcome to the rgwfuncs interactive shell.\nUse up/down arrows for command history.\nType 'exit()' or Ctrl+D to quit.\033[0m"
-    exitmsg = f"{shell_color}Goodbye.\033[0m"
+    # Start interactive session with a custom banner and exit message in default color
+    banner = f"{shell_color}Welcome to the rgwfuncs interactive shell.\nUse up/down arrows for command history.\nType 'exit()' or Ctrl+D to quit."
+    exitmsg = f"{shell_color}Goodbye."
 
-    # Run the interactive session without modifying sys.stdout
+    # Run the interactive session
     console.interact(banner=banner, exitmsg=exitmsg)
+

{rgwfuncs-0.0.78.dist-info → rgwfuncs-0.0.79.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rgwfuncs
-Version: 0.0.78
+Version: 0.0.79
 Summary: A functional programming paradigm for mathematical modelling and data science
 Home-page: https://github.com/ryangerardwilson/rgwfunc
 Author: Ryan Gerard Wilson

{rgwfuncs-0.0.78.dist-info → rgwfuncs-0.0.79.dist-info}/RECORD

@@ -2,11 +2,11 @@ rgwfuncs/__init__.py,sha256=LSn54Tlyskcb6Wab_wUpPLB6UGMe5LdrB3GU88mDEbU,1712
 rgwfuncs/algebra_lib.py,sha256=rKFITfpWfgdBswnbMUuS41XgndEt-jUVz2ObO_ik7eM,42234
 rgwfuncs/df_lib.py,sha256=r6T-MwyDq9NAPW1Xf6NzSy7ZFicIKdemR-UKu6TZt5g,71111
 rgwfuncs/docs_lib.py,sha256=y3wSAOPO3qsA4HZ7xAtW8HimM8w-c8hjcEzMRLJ96ao,1960
-rgwfuncs/interactive_shell_lib.py,sha256=VkhhFRD2dMof9rk_LzW9qO3hcoN9H7CmXD5SRZx0vO0,2316
+rgwfuncs/interactive_shell_lib.py,sha256=Pdujo-XdtfnCpudzHz0jSnYes0hNbHfPpREu4Fq1K0M,2979
 rgwfuncs/str_lib.py,sha256=rtAdRlnSJIu3JhI-tA_A0wCiPK2m-zn5RoGpBxv_g-4,2228
-rgwfuncs-0.0.78.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
-rgwfuncs-0.0.78.dist-info/METADATA,sha256=-AkEFWXGxxs3hSb_-CAPXKoTsYNsvWdLssecL69lPaM,60288
-rgwfuncs-0.0.78.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-rgwfuncs-0.0.78.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
-rgwfuncs-0.0.78.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
-rgwfuncs-0.0.78.dist-info/RECORD,,
+rgwfuncs-0.0.79.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
+rgwfuncs-0.0.79.dist-info/METADATA,sha256=gTXvRXdi0hv7sSEW8TQIZdAHL0nrgtFx7kskQyL2FpM,60288
+rgwfuncs-0.0.79.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+rgwfuncs-0.0.79.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
+rgwfuncs-0.0.79.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
+rgwfuncs-0.0.79.dist-info/RECORD,,