如何为Python装饰器添加位置参数？解决@语法报错问题

这是个非常典型的装饰器进阶坑——当我们想给装饰器本身传递参数时，两层嵌套的结构就满足不了需求了，得改成三层嵌套的装饰器结构，让我给你一步步拆解清楚：

为什么原来的写法会报错？ #

先回顾下你的问题核心：你改造后的add_a_book需要接收两个参数（被装饰函数func和书籍名book），但Python的@装饰器语法有固定的执行逻辑：

• 当你写@decorator时，Python会自动把下面的函数作为参数传给decorator
• 当你写@decorator(args)时，Python会先执行decorator(args)，然后把返回的结果当成装饰器去处理下面的函数

你的原写法里，add_a_book直接需要func和book两个参数，所以：

• 用@add_a_book("Python")时，只传了book，返回的不是一个能接收func的装饰器，自然报"missing positional argument: 'book'"
• 用@add_a_book时，Python把被装饰函数作为第一个参数传进去，但add_a_book还缺book参数，同样报错
• 第三种写法@add_a_book()不仅没传book，函数定义里写def read_a_book("python")本身就是语法错误，参数不能是字符串字面量

正确的三层嵌套装饰器写法 #

我们需要把装饰器拆成三层：最外层接收装饰器的参数，中间层接收被装饰的函数，最内层是实际执行的包装函数。重构后的代码如下：

def add_a_book(book):
    # 第一层：接收装饰器的参数（这里是book）
    def decorator(func):
        # 第二层：接收被装饰的函数
        def wrapper():
            # 第三层：实际的包装逻辑
            return func() + book
        return wrapper
    return decorator

现在用@语法就能正常工作了：

@add_a_book("Python Cookbook")
def read_a_book():
    return "I am reading the book: "

# 执行测试
print(read_a_book())  # 输出: I am reading the book: Python Cookbook

为什么三层嵌套能行？ #

简单说下执行流程：

1. @add_a_book("Python Cookbook")先执行add_a_book("Python Cookbook")，返回里面的decorator函数
2. Python自动把被装饰的read_a_book函数作为参数传给decorator
3. decorator返回wrapper函数，替换原来的read_a_book
4. 当你调用read_a_book()时，实际执行的是wrapper()，它会调用原函数再拼接上book参数

进阶优化：保留原函数元信息 #

上面的写法有个小问题：被装饰后的函数read_a_book的元信息（比如函数名、文档字符串）会变成wrapper，这对调试和反射不太友好。我们可以用functools.wraps来解决这个问题：

import functools

def add_a_book(book):
    def decorator(func):
        @functools.wraps(func)  # 保留原函数的元信息
        def wrapper():
            return func() + book
        return wrapper
    return decorator

@add_a_book("Python Cookbook")
def read_a_book():
    """返回正在读书的字符串"""
    return "I am reading the book: "

print(read_a_book.__name__)  # 输出: read_a_book（而不是wrapper）
print(read_a_book.__doc__)   # 输出: 返回正在读书的字符串

这样你的装饰器就既通用又规范啦！

内容的提问来源于stack exchange，提问作者Wizard