python @wraps()装饰器的使用

def dec(f):
    def wrapper(*a, **kw):
        return f(*a, **kw)
    return wrapper

@dec
def add(x, y):
    """add two numbers"""
    return x + y

print(add.__name__)   # wrapper   ← 名字被换掉了！
print(add.__doc__)    # None      ← 文档串也没了

from functools import wraps
def dec(f):
    @wraps(f)            # 就加这一行
    def wrapper(*a, **kw):
        """wrapper doc"""
        print('before')
        return f(*a, **kw)
    return wrapper
@dec
def add(x, y):
    """add two numbers"""
    return x + y
print(add.__name__)      # add
print(add.__doc__)       # add two numbers
print(add.__annotations__)  # 原函数注解也保留

def wraps(wrapped,
          assigned=WRAPPER_ASSIGNMENTS,      # 要拷贝的七大属性
          updated=WRAPPER_UPDATES):          # __dict__ 如何合并
    return partial(update_wrapper,          # 返回一个“等待 wrapper 作为参数”的函数
                   wrapped=wrapped,
                   assigned=assigned,
                   updated=updated)

MY_ATTRS = ('__name__', '__doc__', '__my_flag__')

def my_wraps(f):
    return wraps(f, assigned=MY_ATTRS)

# 或者只想忽略某个属性
from functools import WRAPPER_ASSIGNMENTS
skip_annotation = tuple(a for a in WRAPPER_ASSIGNMENTS if a != '__annotations__')

def dec(f):
    @wraps(f, assigned=skip_annotation)
    def wrapper(*a, **kw):
        ...

def dec(f):
    @wraps(f, updated=())   # 不合并 __dict__，避免覆盖
    def wrapper(*a, **kw):
        ...
    wrapper.decorator = 'my_dec'   # 新增标记
    return wrapper

from functools import wraps

def repeat(n):                      # ← 这一层接收参数
    def dec(f):
        @wraps(f)
        def wrapper(*a, **kw):
            for _ in range(n):
                result = f(*a, **kw)
            return result
        return wrapper
    return dec

@repeat(3)
def hello():
    """say hello"""
    print('hi')

hello()
print(hello.__name__)   # hello
print(hello.__doc__)    # say hello

def count_calls(cls):
    orig_init = cls.__init__

    @wraps(orig_init)          # 保留原 __init__ 签名
    def new_init(self, *a, **kw):
        self._cnt = 0
        orig_init(self, *a, **kw)

    cls.__init__ = new_init
    return cls

>>> hello.__wrapped__        # wraps 自动加的秘密属性
<function hello at 0x...>    # 指向原始函数
>>> inspect.signature(hello) # 拿到的是原签名，不会看到 wrapper 的 *a, **kw