Python - 装饰器基础知识学习

Python一般函数

类似其他编程余压，Python的函数以def开头，下面给出一个例子：

# step one 普通函数
def myAdd(a,b):
    return a + b

上述函数实现一个简单的两个数相加功能。

Python高阶函数

Python中的函数也是一个对象，也可以作为另一个函数的输入参数或函数返回值，当把函数作为参数输入或返回值时，对应的python函数就称为高阶函数，如下：

# step two 高阶函数
def calc(a,b,f):
    return f(a,b)

上述高阶函数接收三个输入，a，b和f，其中f就是函数对象，返回的就是f(a,b)的值，调用方法：calc(1,2,myAdd)

Python中的高阶函数

举两个python中使用到的高阶函数，filter和map。

filter(function, iterable)

### filter(function,iterable)
'''
# 选出iterable里符合条件的元素
# 返回一个迭代器
'''
i = filter(lambda x : x%2==0, range(10))
print(list(i))

map(function,iterable)

'''
将iterable中的元素一一代入function做改变
'''
i2 = map(lambda x:x+1,range(5))
list(i2)

i3 = map(lambda x,y : x + y,'abc','qwe')
res = list(i3)
print(res)

Python装饰器

装饰器的理解

这里有两个python函数，myAdd和mySub，分别实现两个数的相加与相减。

def myAdd(x,y=10):
    return x + y

def mySub(x,y=10):
    return x - y

如果此时需要在两个函数的基本功能上增加对程序耗时的显示功能，就可以借助高阶函数实现，具体实现如下：

from time import time
def calcTime(func,x,y=10):
    before = time()
    res = func(x,y)
    after = time()
    print('time consuming:',after - before)
    return res

# 测试一下
calcTime(myAdd,10,20)
calcTime(mySub,10,20)

利用calcTime这个高阶函数，可以实现在调用myAdd和mySub函数的同时显示具体的耗时，基本想法实现了，但是有一点与原来的需求有所不同：
原来的需求不是新增加一个函数，去调用新的函数，而是在现有函数的基础上增加功能，所以需要对上面的代码进行一下修改：

from time import time
# 构造一个高阶函数，返回值是f这个函数
def calcTime_Wrapper(func):
    # 将附加的功能写到f函数中去，同时让f函数返回func这个函数
    # func实际上就是用户初期设计的目标函数
    def f(x,y=10):
        before = time()
        res = func(x,y)
        after = time()
        print('time consuming:',after-before)
        return res
    return f

def myAdd(x,y=10):
    return x+y

def mySub(x,y=10):
    return x - y

myAdd = calcTime_Wrapper(myAdd)
mySub = calcTime_Wrapper(mySub)

myAdd(10,10)
mySub(10，20)

不同的是，高阶函数中把额外的功能与调用目标函数的代码放到了f()函数中，高阶函数则将该f()函数作为返回值。
下一步就是将返回的函数赋值给目标函数名即可。

装饰器的编写

上面的改写已经可以实现预期的功能，不过还有两个小问题：

1. f()函数规定了具体的输入参数，利用*args和**kwargs进行修改
2. 原函数的__name__和__doc__无法正确显示，利用funtools进行修改。

此外，在python中可以直接使用@语法糖实现装饰器。

import functools
from time import time

def calcTime(func):
    #将func函数的真实签名赋值到f这个装饰器函数上
    @functools.wraps(func)
    def f(*args,**kwargs):
        before = time()
        rv = func(*args, **kwargs)
        after = time()
        print('time taken: ', after - before)
        return rv
    return f

@calcTime
def myAdd_Main(x,y=10):
    return x + y

@calcTime
def mySub_Main(x,y=10):
    return x - y

print(myAdd_Main.__name__)

上面是python中装饰器的学习笔记，还有一些内容需要进行完善，先挖坑：
1.wrapt模块的使用，如何利用wrapt模块简化
2.涉及到内层函数修改外层函数的变量时，nonlocal的使用

参考链接

https://pouannes.github.io/blog/decorators/
piglei,多次转载后找不到原文