Python中能否为REPL直接输出的对象提供专属repr？

你想要的效果完全可以实现，之前用inspect.stack()判断<stdin>的方法之所以失效，是因为不管是直接输出对象，还是输出包含该对象的集合，调用__repr__的上层栈帧来源都是REPL的<stdin>，没法区分两种场景。

这里给两个可行的方案：

方案一：利用sys.displayhook（推荐） #

REPL里直接输出表达式结果时，会调用sys.displayhook来处理显示逻辑。我们可以自定义这个钩子，专门针对目标类对象输出调试信息，其他场景保持默认行为，同时把标准可组合的__repr__留给集合、函数返回等场景使用。

示例代码：

import sys

class Foo:
    def __init__(self, x):
        self.x = x
        self._debug_info = f"调试信息：x={x}，内存地址={id(self)}"

    def __str__(self):
        # 返回可用于重建对象的字符串
        return f"x={self.x}"

    def __repr__(self):
        # 标准可组合的repr，用于集合、函数调用结果等场景
        return f"{self.__class__.__name__}({self.x!r})"

    def _debug_repr(self):
        # REPL专属的调试repr
        return f"{self._debug_info}\n{repr(self)}"

# 保存默认的displayhook
_default_displayhook = sys.displayhook

def custom_displayhook(obj):
    if isinstance(obj, Foo):
        print(obj._debug_repr())
    else:
        _default_displayhook(obj)

sys.displayhook = custom_displayhook

测试效果：

• 直接输入foo = Foo(123)后再输入foo，会输出调试信息+标准repr
• 输入[foo]，列表里只会显示标准的Foo(123)
• 输入print(foo)，会输出__str__定义的内容x=123

这个方案完全基于Python标准接口，不依赖解释器实现细节，兼容性最好。

方案二：改进栈帧检查（依赖CPython实现） #

如果不想修改sys.displayhook，可以优化栈帧检查逻辑，不仅看文件名，还要判断调用者的函数名，区分是REPL直接显示还是其他场景的间接调用。

示例代码：

import inspect

class Foo:
    def __init__(self, x):
        self.x = x
        self._debug_info = f"调试信息：x={x}，内存地址={id(self)}"

    def __str__(self):
        return f"x={self.x}"

    def __repr__(self):
        stack = inspect.stack()
        is_repl_direct = False
        # 遍历上层栈帧，判断是否是REPL直接触发的调用
        for frame_info in stack[1:]:
            # CPython中REPL直接显示会经过displayhook或_run_code函数
            if frame_info.function in ('displayhook', '_run_code'):
                is_repl_direct = True
                break
            # 如果遇到非stdin的调用，说明是其他场景，直接终止检查
            if frame_info.filename != '<stdin>':
                break

        if not is_repl_direct:
            return f"{self.__class__.__name__}({self.x!r})"
        return f"{self._debug_info}\n{self.__class__.__name__}({self.x!r})"

注意：这个方法依赖CPython的内部调用逻辑，在IPython、PyPy等其他Python环境中可能不生效。

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