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封閉綁定到原來的未修飾類對象，而不是，但對於一個單元測試，我不會理會。