避免如果__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__)