如何捕获函数打印内容转为字符串？类中printlist方法场景实现

Absolutely! You can capture all printed output from a function or method and convert it into a string by temporarily redirecting Python's standard output (stdout) to an in-memory buffer. Here's a step-by-step solution tailored to your use case, where you want to use the output of your printlist method as your class's string representation:

Cleanest Approach: Use contextlib.redirect_stdout (Python 3.4+) #

This method uses a context manager to handle redirecting and restoring stdout automatically, so you don't have to worry about cleaning up manually.

from contextlib import redirect_stdout
from io import StringIO

class MyClass:
    def printlist(self):
        print(1)
        print(2)
        print(3)
    
    def __str__(self):
        # Create an in-memory buffer to hold the printed output
        output_buffer = StringIO()
        
        # Redirect stdout to our buffer while running printlist
        with redirect_stdout(output_buffer):
            self.printlist()
        
        # Extract the captured content (optional: strip extra newlines/whitespace)
        captured_output = output_buffer.getvalue().strip()
        return captured_output

# Test it out
my_obj = MyClass()
print(str(my_obj))  # Will print:
# 1
# 2
# 3

For Older Python Versions: Manual Stdout Redirect #

If you're working with Python versions before 3.4, you can manually save and restore stdout using a try-finally block to ensure you never leave stdout redirected accidentally:

import sys
from io import StringIO

class MyClass:
    def printlist(self):
        print(1)
        print(2)
        print(3)
    
    def __str__(self):
        # Save the original stdout to restore later
        original_stdout = sys.stdout
        output_buffer = StringIO()
        
        try:
            # Redirect stdout to our buffer
            sys.stdout = output_buffer
            self.printlist()
            # Grab the captured output
            captured_output = output_buffer.getvalue().strip()
        finally:
            # Always restore the original stdout—critical for normal print behavior!
            sys.stdout = original_stdout
        
        return captured_output

Important Notes #

• This captures only output sent to stdout (the default destination for print()). If your printlist method writes to stderr (e.g., print("error", file=sys.stderr)), you'll need to redirect sys.stderr using the same pattern.
• Omit the .strip() call if you want to preserve the exact output format, including trailing newlines at the end of printlist's output.
• By defining __str__, your object will automatically return this captured string whenever it's converted to a string (like when you call str(my_obj) or pass it to print()), which is exactly what you're aiming for.

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