Python capture function print output

You can redirect the output of the print function to a string in Python using the io.StringIO class from the io module. Here’s a simple example to demonstrate how you can achieve this:

import io
import sys

def capture_print_output(func, *args, **kwargs):
    # Create a StringIO object to capture the output
    captured_output = io.StringIO()
    # Redirect sys.stdout to the StringIO object
    sys.stdout = captured_output
    
    try:
        # Call the function with the provided arguments
        func(*args, **kwargs)
    finally:
        # Reset sys.stdout to its original value
        sys.stdout = sys.__stdout__
    
    # Get the captured output as a string
    return captured_output.getvalue()

# Example function that prints something
def example_function():
    print("Hello, World!")

# Capture the output of the example_function
output = capture_print_output(example_function)
print("Captured Output:", output)

In this example:

1. We create a StringIO object to capture the output.
2. We redirect sys.stdout to the StringIO object.
3. We call the function whose output we want to capture.
4. We reset sys.stdout to its original value.
5. We retrieve the captured output as a string using the getvalue() method.

Feel free to adapt this code to fit your specific needs! If you have any other questions or need further assistance, just let me know.

Source: Conversation with Copilot, 8/26/2024 (1) How can I redirect print output of a function in python. https://stackoverflow.com/questions/14197009/how-can-i-redirect-print-%20%20output-of-a-function-in-python%20/%2014197079. (2) How do I redirect output of multiple functions to string or stdout in …. https://stackoverflow.com/questions/64199833/how-do-i-redirect-output-of-multiple-functions-to-string-or-stdout-in-python-t. (3) How to assign Print output to a Variable in Python | bobbyhadz. https://bobbyhadz.com/blog/python-assign-string-output-to-variable. (4) Python 101: Redirecting stdout - Mouse Vs Python. https://www.blog.pythonlibrary.org/2016/06/16/python-101-redirecting-stdout/.