避免如果__name__ ==「__main__」在Python子類使用功能從父

Question

我有一個泛型類（A）這是被繼承了很多像這樣運行：

class A: 
    def run(self): 
     ... 
     self.do_something() 
     ... 

    #abstract function 
    def do_something(self): 
     pass 


class B(A): 
    def do_something(self): 
     ... 

子類是我正在通過將此代碼添加到每個文件直接運行單獨的文件（B是子類的文件名）：

if __name__ == '__main__': 
    B().run() 

我的問題是，我能避免這個代碼添加到所有文件與子類，因爲代碼中唯一改變的是正在使用的類（在這個例子中是B）？

Answer 1

如果您的python版本足夠新，您可以創建一個類裝飾器。

在這種情況下，一個間接的。如果你定義runifmain()集中的地方

def runifmain(mainname): 
    def deco(clas): 
     if mainname == '__main__': 
      clas().run() 
     return clas 
    return deco 


@runifmain(__name__) 
class B(A): 
    [...] 

應該做的工作，只是使用@runifmain(__name__)任何需要它。

Answer 2

如何在主類A中添加額外的功能？

class A(): 
    .. 
    def __RunIfMain__(self,name): 
     if name == '__main__': 
      self.run() 

在子類中，你就必須一行少：

class B(A): 
    .. 

B().__RunIfMain__(__name__)