Python类:向方法添加动态属性

时间:2012-09-13 10:04:13

标签: python class attributes

假设我们有一个班级:

class Foo (object):
...     def __init__(self,d):
...         self.d=d
...     def return_d(self):
...         return self.d

......和一个字典:

d={'k1':1,'k2':2}

......和一个实例:

inst=Foo(d)

有没有办法动态地将属性添加到return_d,所以:

inst.return_d.k1会返回1吗?

1 个答案:

答案 0 :(得分:9)

您需要做两件事:将return_d声明为属性或属性,并返回一个类似dict的对象,允许对字典键进行属性访问。以下方法可行:

class AttributeDict(dict): 
    __getattr__ = dict.__getitem__

class Foo (object):
    def __init__(self,d):
        self.d=d

    @property
    def return_d(self):
        return AttributeDict(self.d)

简短演示:

>>> foo = Foo({'k1':1,'k2':2})
>>> foo.return_d.k1
1

property decorator将方法转换为属性,__getattr__ hook允许AttributeDict类通过属性访问(.运算符)查找dict键。