如何自定义set函数的行为 - 问答

在Python中，您可以通过定义一个名为__set__的特殊方法来自定义set函数的行为

class CustomSet:
    def __init__(self, value):
        self._value = value

    def __set__(self, instance, value):
        print("自定义 set 函数被调用")
        if isinstance(value, int):
            self._value = value
        else:
            raise ValueError("值必须是整数")

    def __get__(self, instance, owner):
        return self._value

class MyClass:
    custom_set = CustomSet(0)

# 使用自定义 set 函数
obj = MyClass()
obj.custom_set = 42  # 输出：自定义 set 函数被调用
print(obj.custom_set)  # 输出：42

try:
    obj.custom_set = "not an integer"  # 输出：自定义 set 函数被调用
except ValueError as e:
    print(e)  # 输出：值必须是整数

在这个例子中，我们创建了一个名为CustomSet的类，它有一个__set__方法。当我们尝试设置MyClass实例的custom_set属性时，CustomSet的__set__方法会被调用。我们还定义了一个__get__方法，以便在访问属性时返回正确的值。

请注意，这个例子仅适用于描述符对象，而不是常规的函数或方法。要了解更多关于描述符的信息，请参阅Python官方文档。

0 赞

0 踩