假设我有以下 Python 类
class Foo(object):
a = None
b = None
c = None
def __init__(self, a = None, b = None, c = None):
self.a = a
self.b = b
self.c = c
有什么办法可以简化这个过程吗?每当我向 Foo 类添加新成员时,我都被迫修改构造函数。
请注意
class Foo(object):
a = None
在
Foo
的字典中设置键值对:
Foo.__dict__['a']=None
同时
def __init__(self, a = None, b = None, c = None):
self.a = a
在 Foo 实例对象的字典中设置键值对:
foo=Foo()
foo.__dict__['a']=a
因此,在定义顶部设置类成员与定义下半部分(在
__init__
内)的实例属性设置没有直接关系。
另外,值得注意的是
__init__
是 Python 的 initializer。 __new__
是类 构造函数。
如果您正在寻找一种基于
__init__
的参数自动添加一些实例属性的方法,您可以使用以下方法:
import inspect
import functools
def autoargs(*include,**kwargs):
def _autoargs(func):
attrs,varargs,varkw,defaults=inspect.getargspec(func)
def sieve(attr):
if kwargs and attr in kwargs['exclude']: return False
if not include or attr in include: return True
else: return False
@functools.wraps(func)
def wrapper(self,*args,**kwargs):
# handle default values
for attr,val in zip(reversed(attrs),reversed(defaults)):
if sieve(attr): setattr(self, attr, val)
# handle positional arguments
positional_attrs=attrs[1:]
for attr,val in zip(positional_attrs,args):
if sieve(attr): setattr(self, attr, val)
# handle varargs
if varargs:
remaining_args=args[len(positional_attrs):]
if sieve(varargs): setattr(self, varargs, remaining_args)
# handle varkw
if kwargs:
for attr,val in kwargs.iteritems():
if sieve(attr): setattr(self,attr,val)
return func(self,*args,**kwargs)
return wrapper
return _autoargs
所以当你说
class Foo(object):
@autoargs()
def __init__(self,x,path,debug=False,*args,**kw):
pass
foo=Foo('bar','/tmp',True, 100, 101,verbose=True)
您会自动获取这些实例属性:
print(foo.x)
# bar
print(foo.path)
# /tmp
print(foo.debug)
# True
print(foo.args)
# (100, 101)
print(foo.verbose)
# True
PS。虽然我写这篇文章(为了好玩),但我不建议在严肃的工作中使用
autoargs
。明确是简单、清晰且无误的。我不能对autoargs
说同样的话。
PPS。是我一个人的问题,还是 Stackoverflow 上的很多按钮都坏了?编辑器窗口丢失了所有图标...:(清除浏览器缓存解决了问题。
dataclasses
,这在以下情况下很有帮助:
from dataclasses import dataclass
@dataclass
class Foo:
a: str = None
b: str = None
c: str = None
当您只想存储一些属性时,这可以让您不必编写
__init__
方法。
给你一个很好的
__repr__
方法:
>>> a = Foo()
>>> a
Foo(a=None, b=None, c=None)
如果需要对参数进行计算,可以实现
__post_init__
。
另请参阅命名元组:
from collections import namedtuple
Foo = namedtuple('Foo', ['a', 'b', 'c'])
namedtuple 需要所有字段。
>>> a = Foo(1, 2, 3)
>>> a
Foo(a=1, b=2, c=3)
有优雅的方法可以做到这一点。
有什么办法可以简化这个过程吗?每当我向 Foo 类添加新成员时,我都被迫修改构造函数。
还有一种粗暴的方式。它会起作用,但不推荐。看看并决定。
>>> class Foo(object):
def __init__(self, **attrs):
self.__dict__.update(**attrs)
def __getattr__(self, attr):
return self.__dict__.get(attr, None)
>>> f = Foo(a = 1, b = 2, c = 3)
>>> f.a, f.b
(1, 2)
>>> f = Foo(bar = 'baz')
>>> f.bar
'baz'
>>> f.a
>>>
关键字参数构造函数让您无需显式定义任何参数即可摆脱困境。 警告:这违背了“显式优于隐式”的原则。
仅当您想返回不存在的属性的默认值而不是获取
__getattr__
时,才需要覆盖 AttributeError
。