避免 if __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版本足够新，你可以创建一个类装饰器。

在这种情况下，是一种间接的。

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


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

如果您在某个地方集中定义 runifmain() 并在需要的地方使用 @runifmain(__name__) ，那么应该可以完成这项工作。

Answer 2

向你的主类 A 添加一个额外的函数怎么样？

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

在你的子类中你会少一行：

class B(A):
  ..

B().__RunIfMain__(__name__)