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(..)，则传入函数object（command_get_parameter），并用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..

装饰函数

1 个答案: