是否有一种方法可以将相同的属性逻辑应用于类中的一组属性?例如,我想将相同的@attr1.setter
装饰器应用于attr2
,attr3
和attr4
,而不必为每个属性定义属性。
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