Python装饰器添加类级变量

Question

我有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的具体表继承在代码中看起来更好，尽管我可以看到很多这种情况的用例。

Answer 1

你当然可以; 您可以使用将类 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']