10

このようなPythonデコレータ関数を作成すると

def retry_until_true(tries, delay=60):
    """
    Decorator to rety a function or method until it returns True.
    """
    def deco_retry(f):
        def f_retry(*args, **kwargs):
            mtries  = tries
            rv = f(*args, **kwargs)
            while mtries > 0:
                if rv is True:
                    return True
                mtries -= 1
                time.sleep(delay)
                rv = f(*args, **kwargs)
            return False
        return f_retry
    return deco_retry

こんな感じで使えます

    @retry_until_true(20, delay=30)
    def check_something_function(x, y):
        ...
        return True

しかし、20 と 30 が変数になるように、実行時に「試行」と「遅延」の異なる値をデコレータに渡す方法はありますか?

4

2 に答える 2

11

triesとのインスタンス変数を使用して、クラスをデコレータとして使用できますdelay。

class RetryUntilTrue(object):
    def __init__(self, f=None, tries=10, delay=30):
        self.f = f
        self.tries = tries
        self.delay = delay

    def __call__(self, *args, **kwargs):
        if self.f is None:
            self.f = args[0]
        else:
            tries = self.tries
            while tries:
                if self.f(*args, **kwargs):
                    return True

                tries -= 1
                time.sleep(self.delay)

使用法：

@RetryUntilTrue
def foo(x):
    pass

foo.tries = 20
foo.delay = 1

@RetryUntilTrue(tries=100, delay=9999)
def bar(x):
    pass

bar.tries -= 1
bar.delay = foo.delay
于 2012-08-07T23:36:23.617 に答える
10

もちろん、関数定義を別の関数にネストすることもできます。たとえば、次のようになります。

def explicit_setup_func(tries, delay=60):
    @retry_until_true(tries, delay)
    def check_something_function(x, y):
        # Code

ただし、クラス デコレータ ソリューションはより実用的です。

于 2012-08-07T23:42:29.737 に答える