rgwfuncs 0.0.68__py3-none-any.whl → 0.0.70__py3-none-any.whl

rgwfuncs/__init__.py

@@ -2,7 +2,7 @@
 # Dynamically importing functions from modules
 
 from .df_lib import append_columns, append_percentile_classification_column, append_ranged_classification_column, append_ranged_date_classification_column, append_rows, append_xgb_labels, append_xgb_logistic_regression_predictions, append_xgb_regression_predictions, bag_union_join, bottom_n_unique_values, cascade_sort, delete_rows, drop_duplicates, drop_duplicates_retain_first, drop_duplicates_retain_last, filter_dataframe, filter_indian_mobiles, first_n_rows, from_raw_data, insert_dataframe_in_sqlite_database, last_n_rows, left_join, limit_dataframe, load_data_from_path, load_data_from_query, load_data_from_sqlite_path, load_fresh_data_or_pull_from_cache, mask_against_dataframe, mask_against_dataframe_converse, numeric_clean, order_columns, print_correlation, print_dataframe, print_memory_usage, print_n_frequency_cascading, print_n_frequency_linear, rename_columns, retain_columns, right_join, send_data_to_email, send_data_to_slack, send_dataframe_via_telegram, sync_dataframe_to_sqlite_database, top_n_unique_values, union_join, update_rows
-from .interactive_shell_lib import interactive_shell, setup_readline
+from .interactive_shell_lib import interactive_shell
 from .algebra_lib import cancel_polynomial_expression, compute_constant_expression, compute_constant_expression_involving_matrices, compute_constant_expression_involving_ordered_series, compute_prime_factors, expand_polynomial_expression, factor_polynomial_expression, plot_polynomial_functions, plot_x_points_of_polynomial_functions, python_polynomial_expression_to_latex, simplify_polynomial_expression, solve_homogeneous_polynomial_expression
 from .docs_lib import docs
 from .str_lib import send_telegram_message

rgwfuncs/interactive_shell_lib.py

@@ -2,7 +2,7 @@
 import code
 import readline
 import rlcompleter  # noqa: F401
-import sys  # noqa: F401
+import sys
 import os
 import atexit
 from typing import Dict, Any
@@ -11,33 +11,40 @@ from .algebra_lib import *  # noqa: F401, F403, E402
 from .str_lib import *  # noqa: F401, F403, E402
 from .docs_lib import *  # noqa: F401, F403, E402
 
-# File for command history
-HISTORY_FILE = os.path.expanduser("~/.rgwfuncs_shell_history")
-
-def setup_readline():
-    """Set up readline for command history persistence"""
-    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"Warning: Could not load history file: {e}")
-    atexit.register(readline.write_history_file, HISTORY_FILE)
-
-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. 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 only if it doesn’t already contain ANSI codes, preserving existing codes."""
+        # Check if the text contains any ANSI escape sequences
+        if '\033[' not in text and '\x1b[' not in text:
+            return f"{shell_color}{text}\033[0m"
+        return text  # Return unchanged if it already has ANSI codes
+
     if not isinstance(local_vars, dict):
         raise TypeError("local_vars must be a dictionary")
 
     # Set up readline for history and completion
-    setup_readline()
+    setup_readline(shell_color)
 
     # Make imported functions available in the REPL
     local_vars.update(globals())
@@ -45,6 +52,24 @@ def interactive_shell(local_vars: Dict[str, Any]) -> None:
     # Create interactive console with local context
     console = code.InteractiveConsole(locals=local_vars)
 
-    # Start interactive session with a custom banner
-    banner = "Welcome to the rgwfuncs interactive shell.\nUse up/down arrows for command history.\nType 'exit()' or Ctrl+D to quit."
-    console.interact(banner=banner, exitmsg="Goodbye.")
+    # 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:
+        # Pass text through colorize_output to handle coloring
+        colored_text = colorize_output(text, shell_color)
+        original_write(colored_text)
+
+    # 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.68.dist-info → rgwfuncs-0.0.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rgwfuncs
-Version: 0.0.68
+Version: 0.0.70
 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.70.dist-info/RECORD

@@ -0,0 +1,12 @@
+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=HqYtC-x9J42661d7a-wEOUuEUy0fRZjU3fn-EO6cf7E,3191
+rgwfuncs/str_lib.py,sha256=rtAdRlnSJIu3JhI-tA_A0wCiPK2m-zn5RoGpBxv_g-4,2228
+rgwfuncs-0.0.70.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
+rgwfuncs-0.0.70.dist-info/METADATA,sha256=Ftf_xo3s8-m4kAzqxb2wTYOFMgnoXSlmSyKzXtJdVOg,60288
+rgwfuncs-0.0.70.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+rgwfuncs-0.0.70.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
+rgwfuncs-0.0.70.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
+rgwfuncs-0.0.70.dist-info/RECORD,,

rgwfuncs-0.0.68.dist-info/RECORD

@@ -1,12 +0,0 @@
-rgwfuncs/__init__.py,sha256=fvKoJHhjdgBeoUOrQ5bHAQbJdbTNF7a9hMEx71QF_Qo,1728
-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=Zu0c_7CWsXPyA7FeCzJiw4xtm4_BrU7oC8fvWX6nLqw,1928
-rgwfuncs/str_lib.py,sha256=rtAdRlnSJIu3JhI-tA_A0wCiPK2m-zn5RoGpBxv_g-4,2228
-rgwfuncs-0.0.68.dist-info/LICENSE,sha256=jLvt20gcUZYB8UOvyBvyKQ1qhYYhD__qP7ZDx2lPFkU,1062
-rgwfuncs-0.0.68.dist-info/METADATA,sha256=BPjqpfQUU9azWFph9HHtxthuO65hQj0fJQ5ccbpeAQI,60288
-rgwfuncs-0.0.68.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-rgwfuncs-0.0.68.dist-info/entry_points.txt,sha256=j-c5IOPIQ0252EaOV6j6STio56sbXl2C4ym_fQ0lXx0,43
-rgwfuncs-0.0.68.dist-info/top_level.txt,sha256=aGuVIzWsKiV1f2gCb6mynx0zx5ma0B1EwPGFKVEMTi4,9
-rgwfuncs-0.0.68.dist-info/RECORD,,