You've already forked my-video-workflow
减少无用内容输出,添加重试延迟,添加重试次数
This commit is contained in:
@ -1,9 +1,11 @@
|
||||
from time import sleep
|
||||
|
||||
|
||||
class Retry:
|
||||
def __init__(self, max_retry, success_return_value):
|
||||
def __init__(self, max_retry, success_return_value, sleep_sec = 60):
|
||||
self.max_retry = max_retry
|
||||
self.success_return_value = success_return_value
|
||||
self.sleep_sec = 60
|
||||
|
||||
def run(self, func, *args, **kwargs):
|
||||
status = False
|
||||
@ -11,6 +13,7 @@ class Retry:
|
||||
try:
|
||||
return_value = func(*args, **kwargs)
|
||||
except Exception:
|
||||
sleep(self.sleep_sec)
|
||||
continue
|
||||
if return_value == self.success_return_value:
|
||||
status = True
|
||||
|
Reference in New Issue
Block a user