能否向Python类的__str__方法传入变量以改变输出结果？

Python的__str__是特殊方法，它的调用逻辑是固定的：当你使用print(example)或str(example)时，Python会自动调用example.__str__()，这个方法只能接收self一个参数，没法直接传递额外参数，所以你之前的两种写法都会报错。

下面是几种可行的实现方案：

方案1：通过实例属性控制输出 #

给类添加一个实例属性，让__str__根据这个属性的值返回不同结果：

class Basic:
    def __init__(self):
        self.special_case = False  # 默认状态
    
    def __str__(self):
        return "x" if self.special_case else "y"

example = Basic()
print(example)                    # 输出 y
example.special_case = True
print(example)                    # 输出 x

方案2：定义带参数的普通方法 #

放弃直接修改__str__，而是写一个专门的方法接收参数，返回自定义字符串：

class Basic:
    def __str__(self):
        return "y"  # 保持默认输出逻辑
    
    def get_str(self, special_case=False):
        return "x" if special_case else "y"

example = Basic()
print(example)                    # 输出 y
print(example.get_str(special_case=True))  # 输出 x

方案3：用上下文管理器临时切换状态 #

如果需要临时改变输出状态，用完后自动恢复，可以用上下文管理器：

class Basic:
    def __init__(self):
        self.special_case = False
    
    def __str__(self):
        return "x" if self.special_case else "y"
    
    def use_special_case(self):
        # 定义内部上下文管理器类
        class TempSpecialCase:
            def __enter__(inner_self):
                self.original_state = self.special_case
                self.special_case = True
                return self
            
            def __exit__(inner_self, exc_type, exc_val, exc_tb):
                self.special_case = self.original_state
        
        return TempSpecialCase()

example = Basic()
print(example)  # 输出 y
with example.use_special_case():
    print(example)  # 输出 x
print(example)  # 回到默认输出 y

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