将属性添加到python类

时间:2015-07-23 14:11:54

标签: python monkeypatching

请考虑以下代码:

class Foo():

    pass


Foo.entries = dict()


a = Foo()
a.entries['1'] = 1

b = Foo()
b.entries['3'] = 3

print(a.entries)

这将打印:

{'1': 1, '3': 3}

因为条目是作为静态属性添加的。是否有一种方法可以修补类定义以添加新属性(不使用继承)。

我设法找到以下方式,但它看起来很复杂:

def patch_me(target, field, value):
    def func(self):
        if not hasattr(self, '__' + field):
            setattr(self, '__' + field, value())
        return getattr(self, '__' + field)
    setattr(target, field, property(func))

patch_me(Foo, 'entries', dict)

1 个答案:

答案 0 :(得分:1)

通常,属性由__init__()函数或实例化后添加:

foo = Foo()
foo.bar = 'something'  # note case

如果您想自动执行此操作,继承是迄今为止最简单的方法:

class Baz(Foo):
    def __init__(self):
        super().__init__()  # super() needs arguments in 2.x
        self.bar = 'something'

请注意,类不需要出现在Python模块的顶层。您可以在函数内声明一个类:

def make_baz(value):
    class Baz(Foo):
        def __init__(self):
            super().__init__()  # super() needs arguments in 2.x
            self.bar = value()
    return Baz()

此示例将在每次调用make_baz()时创建一个新类。这可能是也可能不是你想要的。这样做可能更简单:

def make_foo(value):
    result = Foo()
    result.bar = value()
    return result

如果您真的开始修补原始类,那么您提供的示例代码或多或少是最简单的方法。您可以考虑使用property()的装饰器语法,但这是一个小的改动。我还应该注意,它将调用双下划线名称修改,这可能是一件好事,因为它意味着你不能与类本身使用的任何名称冲突。