在Python中自动设置类成员变量[重复]

问题描述 投票:0回答:4

假设我有以下 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 类添加新成员时,我都被迫修改构造函数。

python constructor
4个回答
34
投票

请注意

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 上的很多按钮都坏了?编辑器窗口丢失了所有图标...:(清除浏览器缓存解决了问题。


21
投票

Python 3.7 提供了

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)

14
投票

优雅的方法可以做到这一点。

有什么办法可以简化这个过程吗?每当我向 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


© www.soinside.com 2019 - 2024. All rights reserved.