在类中捕获异常

Question

是否可以编写一个异常处理程序来捕获类中所有方法产生的运行时错误？我可以通过用 try/except 包围每一个来做到这一点：

class MyError(Exception):
    def __init__(self, obj, method):
        print 'Debug info:', repr(obj.data), method.__name__
        raise

class MyClass:
    def __init__(self, data):
        self.data = data

    def f1(self):
        try:
            all method code here, maybe failing at run time
        except:
            raise MyError(self, self.f1)

我想知道是否有更通用的方法来实现相同的效果 - 对于课堂上任何地方引发的任何错误。我希望能够访问类数据以打印一些调试信息。 另外，如何获取失败的方法名称（示例中为 f1）？

更新：感谢所有有见地的答案，装饰器的想法看起来是要走的路。 关于捕获所有异常的风险：except 分支中的 raise 语句应该重新引发异常而不会丢失任何信息，不是吗？这就是为什么我也把它放在 MyError 中......

Answer 1

警告：如果你想要这样的东西，很可能你不会……但如果你真的想要……

类似：

import functools

def catch_exception(f):
    @functools.wraps(f)
    def func(*args, **kwargs):
        try:
            return f(*args, **kwargs)
        except Exception as e:
            print 'Caught an exception in', f.__name__
    return func

class Test(object):
    def __init__(self, val):
        self.val = val

    @catch_exception
    def calc():
        return self.val / 0

t = Test(3)
t.calc()

展示了如何装饰各个功能。然后，您可以创建一个类装饰器来将此装饰器应用于每个方法（注意classmethod's/staticmethod's/properties 等...）

Answer 2

假设你有一个装饰器catch_exception 就像@Jon Clement 的回答...

class ErrorCatcher(type):
    def __new__(cls, name, bases, dct):
        for m in dct:
            if hasattr(dct[m], '__call__'):
                dct[m] = catch_exception(dct[m])
        return type.__new__(cls, name, bases, dct)

class Test(object):
    __metaclass__ = ErrorCatcher

    def __init__(self, val):
        self.val = val

    def calc(self):
        return self.val / 0

元类在定义Test 时将catch_exception 应用于看似方法的所有内容。

---

为了响应关于每个方法的自定义消息的评论，可以将这样的消息（甚至是用于生成消息的回调函数）附加为属性：

class Test(object):
    __metaclass__ = ErrorCatcher
    def __init__(self, val):
        self.val = val

    def calc(self):
        return self.val / 0

    calc.msg = "Dividing by 0 is ill-advised"

catch_exception 装饰器将在其参数上查找msg 属性，如果找到，则使用它来处理异常。

这种方法可以扩展； msg 可以不是字符串，而是异常类型到字符串的映射。在任何一种情况下，都可以使用任意回调函数替换字符串（当然，在 catch_exception 的支持下），该函数将引发的异常作为参数。

def calc_handler(exc):
    # ...

calc.callback = calc_handler

Answer 3

装饰器将是一个很好的解决方案。

下面是一个例子：

import inspect

def catch_exception_decorator(function):
   def decorated_function:
      try:
         function()
      except:
         raise MyError(self.__class__, inspect.stack()[1][3])
   return decorated_function

class MyClass(object):
    def __init__(self):
         ...

    @catch_exception_decorator
    def f1(self):
         ...

函数顶部的@catch_exception_decorator 是f1 = catch_exception_decorator(f1) 的快捷方式。

除了执行 self.class，您还可以从实例访问类数据，只要您不隐藏变量即可。 inspect.stack()[1][3] 是当前函数的函数名。您可以使用这些来创建异常属性。