Python декоратора для класса

Question

Я пытаюсь сделать это:

import unittest 

def decorator(cls): 
    class Decorator(cls): 
     def __init__(self, *args, **kwargs): 
      super(Decorator, self).__init__(*args, **kwargs) 

    return Decorator 

@decorator 
class myClass(unittest.TestCase): 
    def __init__(self, *args, **kwargs): 
     super(myClass, self).__init__(*args, **kwargs) 
     self.test = 'test' 

    def test_test(self): 
     pass 


myClass().run() 

Но я получаю рекурсии в MyClass.__init__. Есть ли способы избежать этого?

Answer 1

Вы не можете использовать super(myClass, self) в оформленном классе таким образом.

myClass ищется как глобальный, а глобальная myClass является отскока в Decorator, так что вы говорите Python искать в классе MRO для __init__ начиная с Decorator, который myClass, который вызывает super(myClass, self).__init__(), глядя вверх myClass как глобальный, который связан с Decorator и т.д.

Самый простой обходной не использовать super() здесь:

@decorator 
class myClass(unittest.TestCase): 
    def __init__(self, *args, **kwargs): 
     unittest.TestCase.__init__(self, *args, **kwargs) 
     self.test = 'test' 

Это одна из причин, почему в Python 3 была введена версия без аргумента super(), дающая методы вместо __class__.

Вы могли прыгать через некоторые (очень сложно) обручи, чтобы пересобрать метод myClass.__init__(), чтобы дать ему myClass замыкания, связанное с исходным объектом неукрашенного класса вместо этого, но для UnitTest, я бы не стал заморачиваться.