rgwfuncs 0.0.67__py3-none-any.whl → 0.0.69__py3-none-any.whl

rgwfuncs/interactive_shell_lib.py

@@ -1,7 +1,10 @@
+#!/usr/bin/env python3
 import code
 import readline
 import rlcompleter  # noqa: F401
-import sys  # noqa: F401
+import sys
+import os
+import atexit
 from typing import Dict, Any
 from .df_lib import *  # noqa: F401, F403, E402
 from .algebra_lib import *  # noqa: F401, F403, E402
@@ -9,18 +12,41 @@ from .str_lib import *  # noqa: F401, F403, E402
 from .docs_lib import *  # noqa: F401, F403, E402
 
 
-def interactive_shell(local_vars: Dict[str, Any]) -> None:
+def interactive_shell(local_vars: Dict[str, Any], shell_color: str = '\033[37m') -> None:
     """
     Launches an interactive prompt for inspecting and modifying local variables, making all methods
-    in the rgwfuncs library available by default.
+    in the rgwfuncs library available by default. Persists command history across sessions.
 
     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: non-bright white '\033[37m').
     """
+
+    def setup_readline(shell_color: str = '\033[37m') -> None:
+        """Set up readline for command history persistence with a given shell color."""
+        HISTORY_FILE = os.path.expanduser("~/.rgwfuncs_shell_history")
+        readline.set_history_length(1000)  # Limit history to 1000 lines
+        readline.parse_and_bind("tab: complete")  # Enable tab completion
+        if os.path.exists(HISTORY_FILE):
+            try:
+                readline.read_history_file(HISTORY_FILE)
+            except Exception as e:
+                print(f"{shell_color}Warning: Could not load history file: {e}\033[0m")
+        atexit.register(readline.write_history_file, HISTORY_FILE)
+
+
+    def colorize_output(text: str, shell_color: str) -> str:
+        """Apply shell color to text if it doesn’t already contain ANSI codes."""
+        if '\033[' not in text:
+            return f"{shell_color}{text}\033[0m"
+        return text
+
+
     if not isinstance(local_vars, dict):
         raise TypeError("local_vars must be a dictionary")
 
-    readline.parse_and_bind("tab: complete")
+    # Set up readline for history and completion
+    setup_readline(shell_color)
 
     # Make imported functions available in the REPL
     local_vars.update(globals())
@@ -28,5 +54,22 @@ def interactive_shell(local_vars: Dict[str, Any]) -> None:
     # Create interactive console with local context
     console = code.InteractiveConsole(locals=local_vars)
 
-    # Start interactive session
-    console.interact(banner="Welcome to the rgwfuncs interactive shell.")
+    # 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"
+
+    # Save original stdout write function
+    original_write = sys.stdout.write
+
+    # Define a pure function to wrap stdout.write
+    def colored_write(text: str) -> None:
+        original_write(colorize_output(text, shell_color))
+
+    # Temporarily replace stdout.write
+    sys.stdout.write = colored_write
+
+    # Start the interactive session
+    console.interact(banner=banner, exitmsg=exitmsg)
+
+    # Restore original stdout.write after exiting
+    sys.stdout.write = original_write

{rgwfuncs-0.0.67.dist-info → rgwfuncs-0.0.69.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rgwfuncs
-Version: 0.0.67
+Version: 0.0.69
 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.67.dist-info → rgwfuncs-0.0.69.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=A7EWsYxAfDev_N0-2GjRvAtp0bAwBPHIczXb8Gu9fzI,1107
+rgwfuncs/interactive_shell_lib.py,sha256=BYSeaW6MHXknLMS52Uzu2WhsWqeAHs-zcmpzkEevn2g,2927
 rgwfuncs/str_lib.py,sha256=rtAdRlnSJIu3JhI-tA_A0wCiPK2m-zn5RoGpBxv_g-4,2228
-rgwfuncs-0.0.67.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
-rgwfuncs-0.0.67.dist-info/METADATA,sha256=N1PBQQAjEWCY0f4Fi3O0u5MqRCSKlCtGAnLO9WdK51k,60288
-rgwfuncs-0.0.67.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-rgwfuncs-0.0.67.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
-rgwfuncs-0.0.67.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
-rgwfuncs-0.0.67.dist-info/RECORD,,
+rgwfuncs-0.0.69.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
+rgwfuncs-0.0.69.dist-info/METADATA,sha256=-d8qJ2b_S3qA7y0dzjRHlvkPi-LZE5LosmyDh11peDA,60288
+rgwfuncs-0.0.69.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+rgwfuncs-0.0.69.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
+rgwfuncs-0.0.69.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
+rgwfuncs-0.0.69.dist-info/RECORD,,