如何在Python全局运行环境中装饰print函数且无需修改各文件？

Great question! If you want to globally modify Python's built-in print function using a decorator (without changing every script you run) to add debug info, here's a straightforward way to do it:

The core idea is to wrap the original print function from Python's builtins module with your decorator, then replace the built-in version with your wrapped one. This affects every use of print in the entire Python process.

Step 1: Define the Debug Decorator #

First, create a decorator that adds your desired debug information (like timestamps, caller location, etc.) while preserving all the original behavior of print:

import builtins
import datetime
import inspect

def debug_print_decorator(original_print):
    def wrapper(*args, **kwargs):
        # Grab debug context: timestamp and caller location
        caller_frame = inspect.currentframe().f_back
        caller_file = caller_frame.f_code.co_filename
        caller_line = caller_frame.f_lineno
        current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        
        # Print your debug message first (customize this to your needs)
        original_print(f"[DEBUG {current_time}] Print called from {caller_file}:{caller_line}", **kwargs)
        
        # Call the original print function with the user's input
        return original_print(*args, **kwargs)
    return wrapper

Step 2: Replace the Built-in print #

Once your decorator is ready, overwrite the built-in print with the wrapped version:

# Save the original print if you ever need to restore it later
original_print = builtins.print

# Apply the decorator globally
builtins.print = debug_print_decorator(builtins.print)

How to Make This Apply to Your Entire Environment #

To ensure this modification takes effect for all scripts and code you run, you have a few options:

• Start the interpreter with setup code: Save the above code in a file (e.g., debug_print_setup.py), then launch Python with python -i debug_print_setup.py. Every print in the interactive session will now include debug info.
• Import the setup in your scripts: Add import debug_print_setup at the very top of any script you want to use this modified print in.
• Use a sitecustomize.py file: For a persistent change across all Python runs, place the code in sitecustomize.py (this file runs automatically when Python starts up, if it's in your Python path).

Example Usage #

After setting up the decorator, any call to print will output both your debug info and the original content. For example:

# In a script or interactive session
print("Hello, global debug print!")

Will output:

[DEBUG 2024-05-20 15:45:30] Print called from <stdin>:1
Hello, global debug print!

Key Notes #

• Preserves original behavior: The wrapper uses *args and **kwargs to pass all arguments to the original print, so parameters like sep, end, file, and flush still work as expected.
• Restore original print: If you need to revert to the default print at any point, just run builtins.print = original_print.
• Third-party code impact: This change affects every use of print in the process, including calls from third-party libraries. If you only want to modify your own code's print calls, this approach might be too broad—but it's perfect for full-environment debugging.

内容的提问来源于stack exchange，提问作者MichaelGreen