深入理解Python类中的布尔上下文重写机制-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00528/article/details/148392538

深入理解Python类中的布尔上下文重写机制

til :memo: Today I Learned 项目地址: https://gitcode.com/gh_mirrors/ti/til

引言

在Python编程中，我们经常需要判断一个对象在布尔上下文中的真假值（Truthiness）。默认情况下，Python中的所有对象实例都被视为真值（Truthy），但有时我们需要根据对象的特定属性或状态来定义其布尔值。本文将深入探讨如何通过重写__bool__()和__len__()方法来控制类的布尔上下文行为。

Python中的真值判断基础

在Python中，任何对象都可以通过bool()函数进行真值判断。Python内置了一些默认规则：

数值类型：0为假，非0为真
序列类型：空序列为假，非空为真
自定义类实例：默认为真

>>> bool(0)      # False
>>> bool(1)      # True
>>> bool([])     # False
>>> bool([1])    # True
>>> bool(object())  # True

自定义类的布尔行为

默认行为

对于自定义类，如果没有实现任何特殊方法，其实例在布尔上下文中总是被视为真值：

class DefaultCart:
    def __init__(self, items=None):
        self.items = items or []

cart = DefaultCart()
print(bool(cart))  # 输出: True

使用`bool()`方法

要自定义布尔行为，可以实现__bool__()方法。当调用bool()函数时，Python会优先查找并调用这个方法：

class BoolCart:
    def __init__(self, items=None):
        self.items = items or []
    
    def __bool__(self):
        print("调用__bool__方法")
        return bool(self.items)

empty_cart = BoolCart()
print(bool(empty_cart))  # 输出: False

non_empty_cart = BoolCart([1,2,3])
print(bool(non_empty_cart))  # 输出: True

使用`len()`方法作为后备

如果类没有实现__bool__()但实现了__len__()，Python会使用__len__()的结果来判断真值（长度为0为假，非0为真）：

class LenCart:
    def __init__(self, items=None):
        self.items = items or []
    
    def __len__(self):
        print("调用__len__方法")
        return len(self.items)

empty_cart = LenCart()
print(bool(empty_cart))  # 输出: False

non_empty_cart = LenCart([1,2,3])
print(bool(non_empty_cart))  # 输出: True

同时实现两种方法

如果同时实现了__bool__()和__len__()，Python会优先使用__bool__()：

class BothCart:
    def __init__(self, items=None):
        self.items = items or []
    
    def __len__(self):
        print("调用__len__方法")
        return len(self.items)
    
    def __bool__(self):
        print("调用__bool__方法")
        return bool(self.items)

cart = BothCart()
print(bool(cart))  # 只会调用__bool__方法