rgwfuncs 0.0.68__tar.gz → 0.0.70__tar.gz

{rgwfuncs-0.0.68/src/rgwfuncs.egg-info → rgwfuncs-0.0.70}/PKG-INFO

@@ -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.68 → rgwfuncs-0.0.70}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "rgwfuncs"
-version = "0.0.68"
+version = "0.0.70"
 authors = [
   { name = "Ryan Gerard Wilson", email = "ryangerardwilson@gmail.com" },
 ]

{rgwfuncs-0.0.68 → rgwfuncs-0.0.70}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = rgwfuncs
-version = 0.0.68
+version = 0.0.70
 author = Ryan Gerard Wilson
 author_email = ryangerardwilson@gmail.com
 description = A functional programming paradigm for mathematical modelling and data science

{rgwfuncs-0.0.68 → rgwfuncs-0.0.70}/src/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-0.0.70/src/rgwfuncs/interactive_shell_lib.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+import code
+import readline
+import rlcompleter  # 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
+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[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(shell_color)
+
+    # Make imported functions available in the REPL
+    local_vars.update(globals())
+
+    # Create interactive console with local context
+    console = code.InteractiveConsole(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"
+
+    # 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 → rgwfuncs-0.0.70/src/rgwfuncs.egg-info}/PKG-INFO

@@ -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.68/src/rgwfuncs/interactive_shell_lib.py

@@ -1,50 +0,0 @@
-#!/usr/bin/env python3
-import code
-import readline
-import rlcompleter  # noqa: F401
-import sys  # noqa: F401
-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
-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:
-    """
-    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.
-    """
-    if not isinstance(local_vars, dict):
-        raise TypeError("local_vars must be a dictionary")
-
-    # Set up readline for history and completion
-    setup_readline()
-
-    # Make imported functions available in the REPL
-    local_vars.update(globals())
-
-    # 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.")