Python：我可以使用类级装饰器向装饰类添加 mixin 吗？

Question

我想要一个装饰器，除其他外，向它正在装饰的类添加一个 mixin。 （我承认这可能是一个非常糟糕的主意，但请耐心等待。）以下几乎可行：

def add_mixin(cls):
   class inner(cls, NewMixin):
      pass
   return inner

现在如果我这样做：

@add_mixin
class foo:
   ...bunch o' stuff...

foo 类似乎工作得很好，混合了NewMixin。（也许这个方案在面对其他超类时不会完全保留foo 的 MRO，但我认为我不在乎那个。）

我发现的唯一问题是，此时foo.__name__ 是"inner"，我希望它是"foo"！我可以通过添加来解决这个问题 inner.__name__ = cls.__name__ 在add_mixin 的定义中，这似乎涵盖了所有基础（实际上）。

问题：有没有更好的方法来做到这一点？这里还有其他我没有看到的潜伏陷阱吗？

感谢您的帮助。

Answer 1

编辑：在我写答案时，对问题进行了编辑以提及这种方法。

你可以这样做：

def add_mixin(cls):
   class inner(cls, NewMixin):
      pass
   inner.__name__ = cls.__name__
   return inner

然后：

@add_mixin
class foo:
   pass

@add_mixin
class bar:
   pass

f = foo
print(f.__name__)
b = bar
print(b.__name__)
print(f.__name__)

给予

foo
bar
foo

（在引用 bar 后再次打印 foo 以证明 __name__ 没有被覆盖。）

就陷阱而言，我不确定。