装饰器函数中的异常处理

Question

Python 新手，我有一堆函数可以在某些硬件上执行各种任务。每个函数都有不同数量的参数和返回值。

我想制作一种通用的“重试”包装函数，它会从我的任何函数中捕获异常并进行一些错误处理（例如重试任务）。

据我了解，我应该能够将装饰器函数用作每个函数的通用包装器。这似乎可行，但我似乎无法真正从装饰器函数中调用的函数中获得任何异常。

我查看了各种示例并想出了这个：

def retry(function):
    def _retry(*args, **kwargs):
        try:
            reply = function(*args, **kwargs)
            print "reply: ", reply
            return reply
        except PDError as msg:
            print "_retry", msg
        except:
            print "_retry: another error"
    return _retry

然后我使用我的一个函数的名称来调用它：

value = retry(pd.command_get_parameter(0x00))

它似乎调用了我的函数并正确返回，但我的重试函数中从未捕获到异常。所以我无法处理错误并重试。

我也试过这个：

from functools import wraps

def retry(function):
    @wraps(function)
    def _retry(*args, **kwargs):
    .....

我不确定我做错了什么，或者这是否是最好的方法。有人对如何做到这一点有建议吗？我真的不想为我的每个主要功能分别创建“重试”功能。

Answer 1

将我的评论转换为答案：

你应该像这样使用：

def retry(function):
    @wraps(function)
    def _retry(*args, **kwargs):
        try:
            reply = function(*args, **kwargs)
            print "reply: ", reply
            return reply
        except PDError as msg:
            print "_retry", msg
        except:
            print "_retry: another error"
    return _retry

class SomeClass(object):

    @retry
    def command_get_parameter(..):
        return <some value>
s = SomeClass()
result = s.command_get_parameter(..)  #retry(..) actually invokes this function. 

装饰器接收一个函数，并返回一个装饰函数。装饰是能够在函数被调用之前、之后或捕获异常等之前做某事的东西。如果你使用上面的语法（@retry)，解释器调用retry(..)，传入函数对象（@987654324 @)，并将函数替换为retry(command_get_parameter)返回的函数。

发生了什么有点类似于以下步骤（伪代码）：

new_command_get_parameter = retry(command_get_parameter) #@retry has this effect.
result = new_command_get_parameter(your_input)

不同之处在于，以上两个步骤是由解释器神奇地为您完成的——保持代码更清晰、更易读。

---

目前您正在调用该函数，并将其结果传递给retry(..)，这显然是错误的。此外，它不会按照您希望的方式捕获异常。

---

更新：如果要重试访问实例变量，只需让_retry使用第一个参数为self即可。比如：

def retry(func):
    def _retry(self, *args, **kwargs):
        print "Decorator printing a:", self.a
        print "Decorator printing b:", self.b
        try:
            return func(*args, **kwargs)
        except Exception as e:
            print "Caught exception"
            return "Grr.."
    return _retry


class Temp(object):
    def __init__(self, a, b):
        self.a = a
        self.b = b

    @retry
    def command(self, *args, **kwargs):
        print "In command."
        print "Args:", args
        print "KWargs:", kwargs

        raise Exception("DIE!")

t = Temp(3, 5)

print t.command(3,4,5, a=4, b=8)

输出：

Decorator printing a: 3
Decorator printing b: 5
In command.
Args: (4, 5)
KWargs: {'a': 4, 'b': 8}
Caught exception
Grr..