codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: retry支持立即抛出异常

Browse Source
This commit is contained in:
InfinityPacer
2024-06-30 11:44:33 +08:00
parent 0f7a7ef44f
commit 57cdb57331
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/utils/common.py
View File

@ -6,6 +6,8 @@ from typing import Any
from Crypto import Random from Crypto import Random
from Crypto.Cipher import AES from Crypto.Cipher import AES
from app.schemas.exception import ImmediateException
def retry(ExceptionToCheck: Any, def retry(ExceptionToCheck: Any,
tries: int = 3, delay: int = 3, backoff: int = 2, logger: Any = None): tries: int = 3, delay: int = 3, backoff: int = 2, logger: Any = None):
@ -23,6 +25,8 @@ def retry(ExceptionToCheck: Any,
while mtries > 1: while mtries > 1:
try: try:
return f(*args, **kwargs) return f(*args, **kwargs)
except ImmediateException:
raise
except ExceptionToCheck as e: except ExceptionToCheck as e:
msg = f"{str(e)}, {mdelay} 秒后重试 ..." msg = f"{str(e)}, {mdelay} 秒后重试 ..."
if logger: if logger: