Python装饰器添加类级变量

时间:2012-08-18 00:01:42

标签: python inheritance metaprogramming decorator python-decorators

我有2个A和B类:

class A(object):
    x = 0

class B(object):
    y = 0

如何使用装饰器 B “继承” A 的类级别变量(在本例中为 x ) ?它可能吗?所需的行为(如果可能),在装饰后, B 将如下所示:

class B(object):
    x = 0
    y = 0

注意:如果有人想/需要知道我为什么要这样做,那只是让SQLAlchemy的具体表继承在代码中看起来更好,尽管我可以看到很多这种行为的用例。

1 个答案:

答案 0 :(得分:3)

当然可以;你可以使用一个类装饰器,它将类A作为参数,然后为你更新装饰类:

import types

class copyattributes(object):
    def __init__(self, source):
        self.source = source

    def __call__(self, target):
        for attr, value in self.source.__dict__.items():
            if attr.startswith('__'):
                continue
            if isinstance(value, (property, types.FunctionType)):
                continue
            setattr(target, attr, value)
        return target

装饰器复制任何真正属性的属性(不是函数或属性),并且不以双下划线开头。

用法:

class A(object):
    x = 0

@copyattributes(A)
class B(object):
    y = 0

测试提示:

>>> class A(object):
...     x = 0
...
>>> @copyattributes(A)
... class B(object):
...     y = 0
... 
>>> B.y
0
>>> B.x
0
>>> dir(B)
['__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__module__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', '__weakref__', 'x', 'y']