mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-02-02 18:22:39 +08:00
104 lines
3.1 KiB
Python
104 lines
3.1 KiB
Python
import asyncio
|
|
import inspect
|
|
import time
|
|
from functools import wraps
|
|
from typing import Any, Callable
|
|
|
|
from app.schemas import ImmediateException
|
|
|
|
|
|
def retry(ExceptionToCheck: Any,
|
|
tries: int = 3, delay: int = 3, backoff: int = 2, logger: Any = None):
|
|
"""
|
|
:param ExceptionToCheck: 需要捕获的异常
|
|
:param tries: 重试次数
|
|
:param delay: 延迟时间
|
|
:param backoff: 延迟倍数
|
|
:param logger: 日志对象
|
|
"""
|
|
|
|
def deco_retry(f):
|
|
def f_retry(*args, **kwargs):
|
|
mtries, mdelay = tries, delay
|
|
while mtries > 1:
|
|
try:
|
|
return f(*args, **kwargs)
|
|
except ImmediateException:
|
|
raise
|
|
except ExceptionToCheck as e:
|
|
msg = f"{str(e)}, {mdelay} 秒后重试 ..."
|
|
if logger:
|
|
logger.warn(msg)
|
|
else:
|
|
print(msg)
|
|
time.sleep(mdelay)
|
|
mtries -= 1
|
|
mdelay *= backoff
|
|
return f(*args, **kwargs)
|
|
|
|
async def async_f_retry(*args, **kwargs):
|
|
mtries, mdelay = tries, delay
|
|
while mtries > 1:
|
|
try:
|
|
return await f(*args, **kwargs)
|
|
except ImmediateException:
|
|
raise
|
|
except ExceptionToCheck as e:
|
|
msg = f"{str(e)}, {mdelay} 秒后重试 ..."
|
|
if logger:
|
|
logger.warn(msg)
|
|
else:
|
|
print(msg)
|
|
await asyncio.sleep(mdelay)
|
|
mtries -= 1
|
|
mdelay *= backoff
|
|
return await f(*args, **kwargs)
|
|
|
|
# 根据函数类型返回相应的包装器
|
|
if inspect.iscoroutinefunction(f):
|
|
return async_f_retry
|
|
else:
|
|
return f_retry
|
|
|
|
return deco_retry
|
|
|
|
|
|
def log_execution_time(logger: Any = None):
|
|
"""
|
|
记录函数执行时间的装饰器
|
|
:param logger: 日志记录器对象，用于记录异常信息
|
|
"""
|
|
|
|
def decorator(func: Callable):
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
start_time = time.time()
|
|
result = func(*args, **kwargs)
|
|
end_time = time.time()
|
|
msg = f"{func.__name__} execution time: {end_time - start_time:.2f} seconds"
|
|
if logger:
|
|
logger.debug(msg)
|
|
else:
|
|
print(msg)
|
|
return result
|
|
|
|
@wraps(func)
|
|
async def async_wrapper(*args, **kwargs):
|
|
start_time = time.time()
|
|
result = await func(*args, **kwargs)
|
|
end_time = time.time()
|
|
msg = f"{func.__name__} execution time: {end_time - start_time:.2f} seconds"
|
|
if logger:
|
|
logger.debug(msg)
|
|
else:
|
|
print(msg)
|
|
return result
|
|
|
|
# 根据函数类型返回相应的包装器
|
|
if inspect.iscoroutinefunction(func):
|
|
return async_wrapper
|
|
else:
|
|
return wrapper
|
|
|
|
return decorator
|