functools工具 wraps partial

wraps

''' Python装饰器（decorator）在实现的时候，有一些细节需要被注意。例如，被装饰后的函数其实已经是另外一个函数了（函数名等函数属性会发生改变）。这样有时候会对程序造成一些不便，例如笔者想对flask框架中的一些函数添加自定义的decorator，添加后由于函数名和函数的doc发生了改变，对测试结果有一些影响。 所以，Python的functools包中提供了一个叫wraps的decorator来消除这样的副作用。写一个decorator的时候，最好在实现之前加上functools的wrap，它能保留原有函数的名称和docstring。 '''

from functools import wraps

def deco(func):
    @wraps(func) #加在最内层函数正上方
    def wrapper(*args,**kwargs):
        return func(*args,**kwargs)
    return wrapper

@deco
def index():
    '''哈哈哈哈'''
    print('from index')

print(index.__doc__)  # 哈哈哈哈
index()  # from index

总结: wraps可保留被装饰函数的源信息

partial

偏函数可以帮助我们为函数传递参数

import functools

def func(n1, n2, n3):
    print(n1, n2, n3)

new_func = functools.partial(func, 3)
new_func(6, 9)  # 3 6 9

print(func)  # <function func at 0x0000018EE4766840>
print(new_func)  # functools.partial(<function func at 0x0000018EE4766840>, 3)