捕获子类 __init__ 中的异常并退出

Question

考虑下面的代码：

class Parent():
    
    def __init__(self, ...):
        # set up logging
        # ....


class Child(Parent):
    
    def __init__(self, ...):
        super(Child, self).__init__(...)

        print(1/0)


c = Child()

由于1/0，上述操作将失败。当__init__ 失败时，是否有可能在Parent 中执行一些逻辑（例如：我想记录一些东西）？也就是说，我想在 Child 类的 __init__ 中捕获异常，然后在整个实例被垃圾收集之前执行诸如日志记录之类的操作。

Answer 1

做到这一点的一种方法是用可以处理发生的任何异常的东西来装饰每个子类的__init__() 方法。您可以在每个子类中显式执行此操作，或者通过使用 Python 3.6 中添加的相对较新的 __init_subclass__() 方法定义元类或自动执行此操作。

后者似乎是使其自动化的最简单方法，所以我将说明如何做到这一点：

from functools import wraps

class Parent:

    def __init__(self, *args, **kwargs):
        # set up logging...
        pass

    def logger(self, msg):
        print('logging msg:', msg)

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        cls.__init__ = cls.log_exceptions(cls.__init__)  # Decorate subclass' __init__.

    @staticmethod
    def log_exceptions(method):
        """ Wrap method in an exception handler that will automatically log any
            exceptions that occur executing it.
        """
        @wraps(method)
        def wrapped(self, *args, **kwargs):
            try:
                return method(self, *args, **kwargs)
            except Exception as exc:
                self.logger(exc)
        return wrapped


class Child(Parent):

    def __init__(self, *args, **kwargs):
        print(f"{type(self).__name__}.__init__() called")
        super().__init__(*args, **kwargs)  # Base class initialization.
        # other initializations...
        print(1 / 0)


c = Child(1, 2, 3)

Answer 2

您必须使有问题的代码的执行由父级实际运行。一种方法是不在__init__ 本身中定义代码，而是在父级执行的挂钩中定义代码。 （dataclasses 模块做了类似的事情。）

class Parent():
    
    def __init__(self, ...):
        # set up logging
        # ....
        
        try:
            self.post_init()
        except Exception:
            ...

    # Do-nothing implementation in case the child doesn't override it
    def post_init(self):
        pass

class Child(Parent):
    
    def post_init(self):
        print(1/0)
    

c = Child()

Answer 3

这样做的一种方法是在您的父类中定义一个记录器方法，以便子类中的每个方法（包括__init__）都可以通过继承logger 方法来获取它。

class Parent:

    def __init__(self, *args):
        print(args, 'Parent')

    def logger(self, msg):
        print('logging', msg)


class Child(Parent):

    def __init__(self, *args):
        super().__init__(*args)

        # other initializations here

        # the code may cause problem
        try:
            print(1 / 0)
        except Exception as e:
            self.logger(e)


c = Child(1, 2, 3)

输出：

(1, 2, 3) Parent
logging division by zero

如果需要，每个 child 类还可以覆盖 logger 方法并在父级的 logger 之外添加一些特定的内容。