带参数的Python装饰器写法详解

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《带参数的Python装饰器怎么写？》，以下内容主要包含等知识点，如果你正在学习或准备学习文章，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

带参数的装饰器是返回装饰器的函数，需三层嵌套：外层接收参数，中间层接收函数，内层执行逻辑并调用原函数。如logger(level)可控制日志级别，配合functools.wraps保留函数元信息；retry(max_attempts)示例展示异常重试机制，结构清晰且实用。

带参数的装饰器其实是一个返回装饰器的函数。也就是说，它外层多了一层函数包装，用来接收参数。

基本结构说明

要写一个带参数的装饰器，你需要三层嵌套函数：

• 最外层函数：接收装饰器的参数
• 中间层函数：接收被装饰的函数
• 最内层函数：实际执行逻辑，可以调用原函数并添加额外行为

示例：一个记录日志级别的装饰器

比如你想控制打印日志的级别（info、warning等）：

def logger(level):
    def decorator(func):
        def wrapper(*args, **kwargs):
            print(f"[{level.upper()}] 正在执行 {func.__name__}")
            return func(*args, **kwargs)
        return wrapper
    return decorator

# 使用
@logger("info")
def say_hello(name):
    print(f"Hello, {name}!")

say_hello("Alice")

输出：

[INFO] 正在执行 say_hello
Hello, Alice!

使用 functools.wraps 保留原函数信息

为了不丢失原函数的元数据（如 __name__、__doc__），建议加上 functools.wraps：

from functools import wraps

def logger(level):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            print(f"[{level.upper()}] 正在执行 {func.__name__}")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@logger("debug")
def test():
    """测试函数"""
    print("running test")

print(test.__name__)  # 输出: test（没被wrapper覆盖）
print(test.__doc__)   # 输出: 测试函数

支持多种参数的实用例子

比如限制函数重试次数的装饰器：

def retry(max_attempts=3):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            for i in range(max_attempts):
                try:
                    return func(*args, **kwargs)
                except Exception as e:
                    print(f"第 {i+1} 次尝试失败: {e}")
                    if i == max_attempts - 1:
                        raise
        return wrapper
    return decorator

@retry(max_attempts=2)
def may_fail():
    import random
    if random.random() < 0.8:
        raise ConnectionError("网络错误")
    print("成功！")

# 可能重试一次后成功
may_fail()

基本上就这些。关键点是：带参数的装饰器 = 函数返回装饰器。结构清晰，层层嵌套，配合 wraps 更专业。

到这里，我们也就讲完了《带参数的Python装饰器写法详解》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！