From fdaa3ad041b42f9d7df271fce459251de8035328 Mon Sep 17 00:00:00 2001 From: David Date: Sat, 13 Dec 2014 00:56:58 +0800 Subject: [PATCH] assign first argument to stop_max_attempt_number --- retrying.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/retrying.py b/retrying.py index 3ed312d..138617a 100644 --- a/retrying.py +++ b/retrying.py @@ -46,7 +46,11 @@ def wrap(f): @six.wraps(f) def wrapped_f(*args, **kw): - return Retrying(*dargs, **dkw).call(f, *args, **kw) + # if dargs contains more than one argument, + # check first one is an integer, then assign it to 'stop_max_attempt_number' + if len(dargs) >= 1 and isinstance(dargs[0], int): + dkw['stop_max_attempt_number'] = dargs[0] + return Retrying(**dkw).call(f, *args, **kw) return wrapped_f