Python-可以将相同的属性逻辑应用于多个属性



是否有一种方法可以将相同的属性逻辑应用于类中的一组属性?例如,我想将相同的@attr1.setter装饰器应用于attr2attr3attr4,而不必为每个属性定义属性。

class Sample:
    def __init__(self):
        self.attr1 = None
        self.attr2 = None
        self.attr3 = None
        self.attr4 = None
    @property
    def attr1(self):
        return self.__attr1
    @attr1.setter
    def attr1(self, val):
        if val < 0:
            self.__attr1 = 0
        else:
            self.__attr1 = val

只需为此创建自己的描述符:

class MyDescriptor:
    def __set_name__(self, owner, name):
        self.name = f'_{name}'
    def __get__(self, instance, owner):
        return getattr(instance, self.name)
    def __set__(self, instance, val):
        if val is None:
            setattr(instance, self.name, None)
        elif val < 0:
            setattr(instance, self.name, 0)
        else:
            setattr(instance, self.name, val)
class Sample:
    attr1 = MyDescriptor()
    attr2 = MyDescriptor()
    attr3 = MyDescriptor()
    attr4 = MyDescriptor()
    def __init__(self):
        self.attr1 = None
        self.attr2 = None
        self.attr3 = None
        self.attr4 = None

现在,在行动中:

In [3]: s = Sample()
In [4]: s.attr1 = -99
In [5]: s.attr1
Out[5]: 0
In [6]: s.attr2
In [7]: s.attr2 = 10
In [8]: s.attr2
Out[8]: 10
In [9]: s.attr2 = -1
In [10]: s.attr2
Out[10]: 0

请参阅描述符HOWTO和一些更相关的文档

注意,我将None的可能性合并到您的Setter逻辑中(您的代码会在实例初始化时提高TypeError,因为Setter检查None < 0是否)。另请注意,您可能不想使用double-underscore name-mangling(这不表示私有),所以我使用常规 single-underscore表示一个变量,而不是公共API的一部分。使用Double-unnunderScore名称响应在此处复杂的事物。

您可以覆盖__getattr____setattr__以表现出您想要的方式。这样,您就不需要定义任何私人变量或初始化任何成员变量。

class Sample:
    def __getattr__(self, attr):
        return self.__dict__.get(attr)
    def __setattr__(self, attr, val):
        if val is not None and val < 0:
            self.__dict__[attr] = 0
        else:
            self.__dict__[attr] = val
s = Sample()
print(s.attr1) # None
s.attr1 = 10
print(s.attr1) # 10
s.attr1 = -10
print(s.attr1) # 0
s.attr1 = None
print(s.attr1) # None

最新更新