为何redirect_stdout无法捕获InteractiveConsole编译代码的exec输出？如何解决？

你遇到的问题核心原因是：InteractiveConsole.compile("2")生成的代码对象，用exec执行时不会自动打印表达式结果。交互式环境里的自动打印是控制台的额外逻辑，exec本身只会执行代码，不会主动输出表达式的返回值，所以redirect_stdout自然捕获不到内容。

解决方案 #

要捕获这类表达式的输出，我们需要模拟交互式控制台的逻辑：判断执行的是表达式还是语句，对表达式的返回值主动打印。可以用以下两种方式实现：

方法一：区分表达式与语句，结合eval和exec

from code import InteractiveConsole
from io import StringIO
from contextlib import redirect_stdout

cons = InteractiveConsole()
code = cons.compile("2")
f = StringIO()

with redirect_stdout(f):
    try:
        # 尝试用eval执行，获取表达式结果
        result = eval(code)
        if result is not None:
            print(result)
    except SyntaxError:
        # 如果是语句（如赋值、循环），用exec执行
        exec(code)

s = f.getvalue()
print("-" * 20)
print(s)

方法二：直接使用InteractiveConsole的run方法处理输出

InteractiveConsole本身提供了run方法，可以直接处理代码执行和输出，我们只需要重定向它的输出流：

from code import InteractiveConsole
from io import StringIO
from contextlib import redirect_stdout

cons = InteractiveConsole()
code_str = "2"
f = StringIO()

with redirect_stdout(f):
    cons.run(code_str)

s = f.getvalue()
print("-" * 20)
print(s)

说明 #

两种方法都能在Jupyter Notebook环境中正常工作：

• 方法一更灵活，适合已经编译好代码对象的场景；
• 方法二更简洁，直接利用InteractiveConsole的内置逻辑处理交互式代码的输出。

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