2025-10-19 23:05:42 编程来源：ZONE.CI 全球网 0 阅读模式

1. 借助args
2. 更简洁的实现方式
3. 参考资料

之前我们介绍过了无参数和有参数装饰器，那么是否装饰器可以具备可选参数，即既可以使用@func，也可以使用@func(a=1, b=2)，答案是可以的，本篇文章我们介绍一下如何构建一个可选参数的装饰器。

我们所实现的装饰器是一个任务重启装饰器，即任务失败时，能根据指定次数和间隔时间重新运行函数。

1. 借助args

第一种方式是借助于args参数，即通过判断其长度与值的类型，在内部重新定义retry_times和sleep_time变量：

import time
import logging
from functools import wraps
def retry(*args):
    def func_retry(func):
        log = logging.getLogger("Retry")
        @wraps(func)
        def wrapper(*args, **kwargs):
            cnt = 1
            while cnt <= retry_times:
                try:
                    res = func(*args, **kwargs)
                    return res
                except Exception as e:
                    log.error(f"Retry [red]{cnt}[/] times \n", extra={"markup": True})
                    cnt += 1
                    time.sleep(sleep_time)
                    log.error("Messages: [red]{0}[/]".format(e), extra={"markup": True})
            else:
                log.exception(f"Retry {retry_times} times, but all failed!")
                raise Exception(f"Retry {retry_times} times, but all failed!")
        return wrapper
    log = logging.getLogger("Retry")
    # 对应@retry
    if len(args) == 1 and callable(args[0]):
        retry_times, sleep_time = 5, 60
        return func_retry(args[0])
    # 对应@retry(3)
    elif len(args) == 1 and isinstance(args[0], int):
        retry_times, sleep_time = args[0], 60
        return func_retry
    # 对应@retry(2, 3)
    else:
        retry_times, sleep_time = args[0], args[1]
        return func_retry

按照上述方式所实现的装饰器，使用方法如下：

@retry
def retry_func():
    print(1 / 0)
@retry(2)
def retry_func():
    print(1 / 0)
@retry(2, 3)
def retry_func():
    print(1 / 0)

可以看到，这种实现方式并不好，它并没有采用传参的方式，并且使用者对所传入值代表含义也会有疑惑。

2. 更简洁的实现方式

接下来我们介绍一种更好的实现方式，借助于functools模块中的partial函数：

import time
import logging
from functools import wraps, partial
def retry(func = None, *, retry_times: int = 5, sleep_time: int = 60):
    log = logging.getLogger("Retry")
    if func is None:
        return partial(retry, retry_times=retry_times, sleep_time=sleep_time)
    @wraps(func)
    def wrapper(*args, **kwargs):
        cnt = 1
        while cnt <= retry_times:
            try:
                res = func(*args, **kwargs)
                return res
            except Exception as e:
                log.error(f"Retry [red]{cnt}[/] times \n", extra={"markup": True})
                cnt += 1
                time.sleep(sleep_time)
                log.error("Messages: [red]{0}[/]".format(e), extra={"markup": True})
        else:
            log.exception(f"Retry {retry_times} times, but all failed!")
            raise Exception(f"Retry {retry_times} times, but all failed!")
    return wrapper

显然，这种方式实现的装饰器代码简洁多了，具体使用方法如下：

@retry
def retry_func():
    print(1 / 0)
@retry(retry_times=2)
def retry_func():
    print(1 / 0)
@retry(retry_times=2, sleep_time=3)
def retry_func():
    print(1 / 0)