如果decorator函数位于我无法修改的库中，我应该如何捕获可能引发的decorator函数中的异常？

Question

我正在谷歌应用程序引擎（GAE）上使用python statsd库。不幸的是，GAE在使用套接字时会不时引发ApplicationError:4未知错误。。该错误是一个apiproxy\u错误。ApplicationError。

statsd客户端已经设置为捕获socket.error，但不是套接字可以在GAE上引发的Application ationError。

我专门使用timer，它返回timer的一个实例：https://github.com/jsocol/pystatsd/blob/master/statsd/client.py#L13

Timer的__call__方法允许将其用作装饰器，如下所示：

from statsd import StatsClient

statsd = StatsClient()

@statsd.timer('myfunc')
def myfunc(a, b):
    """Calculate the most complicated thing a and b can do."""

我没有修改计时器的简单能力。\uu调用\uu方法本身来捕获ApplicationError。

我应该如何编写一个包装器或附加装饰器，它仍然允许像@my\u timer\u wrapper（'statsd\u timer\u name'）这样干净的装饰，但它会捕获包装/装饰timer方法中可能发生的附加异常？

这是我的代码库中的一个基础模块，它将被用于很多地方（无论我们想在什么地方使用时间）。因此，尽管这个答案可能有效，但我确实希望避免强制在代码库中使用@statclient.timer的所有用法都在try except块中定义。

我正在考虑做如下事情：

def my_timer_wrapper(wrapped_func, *args, **kwargs):
  @functools.wraps(wrapped_func)
  class Wat(object):
    def __call__(self, *args, **kwargs):
      timer_instance = stats_client.timer(*args, **kwargs)
      try:
        return timer_instance.__call__(wrapped_func)(*args, **kwargs)
      except Exception:
        logger.warning("Caught exception", exc_info=True)
        def foo():
          pass
        return foo

  return Wat()

然后将使用如下方式：

@my_timer_wrapper('stastd_timer_name')
def timed_func():
  do_work()

有更好或更像蟒蛇的方法吗？

Answer 1

这似乎是一个“尽可能简单”的新装饰器在计时器装饰器周围添加额外的try/except的情况。

唯一的问题是，使用需要定义2级嵌套函数的参数的装饰器几乎总是使它们看起来很复杂，即使它们不是：

from functools import wraps

def  shielded_timer(statsd, 
                    exceptions=(apiproxy_errors.ApplicationError),
                    name=None):

    def decorator(func):
        timer_decorator = statsd.timer(name or func.__name__)
        new_func = timer_decorator(func)
        @wraps(func)
        def wrapper(*args, **kw):
            try:
                return new_func(*args, **kw)
            except BaseException as error:
                if isinstance (error, exceptions):
                    # Expected error (ApplicationError by default) ocurred
                    pass
                else:
                    raise
        return wrapper
    return decorator



#####################
statsd = StatsClient()
@shielded_timer(statsd)
def my_func(a,b):
    ...

正如您所看到的，甚至可以很容易地包含额外的细节——在本例中，我已经在装饰时配置了想要的异常，并且可以选择在调用statsd.timer时自动使用装饰函数的名称。