python装饰器

python学习笔记之装饰器

高阶函数

一个函数能接受另一个函数作为参数传入，这样的一个函数就是高阶函数。

在python中一切皆对象，函数也不例外，函数是可以赋予给一个变量的，看下边代码：

# 求绝对值函数
def fn(x):
    val = x if x >= 0 else -x
    return val

fn(-34)

34

fn(20)

20

在上边的代码中定义了一个函数 fn()，这个函数接收一个参数，函数名是fn，fn(-34)和fn(20)是函数调用。函数名fn是什么呢？其实它就是指向能计算绝对值函数的一个变量。

f1 = fn

type(f1)

function

f1(-12)

12

如上，变量f1已指向了fn函数，调用fn()与调用f1()是一样的效果。得到的结论是变量是可以指向一个函数。

那函数的参数能否接收一个变量呢？以代码来验证：

a = -23

fn(a)

23

上边把变量a指向了-23，再调用fn()函数时把变量a传入，得到了正确的结果，所以函数的参数是能接收一个变量。

总结：在python中，变量可以指向一个函数，函数的参数可以接收一个变量，那么函数的参数也就可以是一个函数。下边以实际代码来验证

def num(x,y):
    return y(x)

def f(a):
    return a ** 2

num(4,f)

16

上边直接调用num()函数，把指向函数f(a)的函数名f，即f实质也是一个变量，作为一个变量传递给了num(x,y)函数中的参数y，这样也得到了正确的答案。所以函数中的参数可以接收一个函数。

总结：高阶函数的实质就是函数的参数能够接收另一个函数。

装饰器

def deco(fn):
    print('ha ha ha')
    return fn

def myfun():
    print('call myfun()')

f4 = deco(myfun) #调用deco(fn)函数，把myfun函数作为参数传入，deco(fn)函数会先执行print语句，再返回fn函数，即这里的myfun函数。再让变量f4指向myfun()这个函数对象

ha ha ha

f4()   #f4是指向myfun()函数对象的一个变量，因为变量可以指向一个函数，所以执行f4()就可以调用myfun()这个函数

call myfun

采用python的魔法也可以实现上边的相同效果，如下：

@deco
def myfun():
    print('call myfun')

ha ha ha

@deco中的@是语法糖，表示下边定义的函数将会被@后的那个函数所修饰，即myfun()函数会被deco(fn)这个函数所修饰，实质就是执行了myfun = deco(myfun)，所以打印出了ha ha ha，并把原来的myfun()函数返回回来，下边执行myfun()时就调用了原来的myfun()函数，输出了call myfun

myfun()

call myfun

再对上边的deco函数进行修改：

def deco(fn):
    def wrap():
        print('ha ha ha')
        print('call {0} funtion'.format(fn.__name__))
        fn()
    return wrap

@deco
def myfun():
    print('call myfun')

myfun()

ha ha ha
call myfun funtion
call myfun

myfun()函数被deco(fn)函数进行修饰后，调用myfun()时就像被施加了魔法一样在执行此函数前附加的执行了一些操作，当然也可以在执行函数后附加一些操作。

上边被装饰的myfun()函数是一个无参的函数，如果被装饰的函数需要接收参数呢？这时装饰器应该这样的来定义：

def deco(fn):
    def wrap(*args,**kwargs):
        print('ha ha ha')
        print('call {0} funtion'.format(fn.__name__))
        return fn(*args,**kwargs)
    return wrap

考虑到被装饰的函数可以接收的参数类型的不确定性，可以用可变位置参数和可变关键字参数来捕捉，即(*args,**kwargs)

@deco
def myfun(x):
    return x ** 2

myfun(5)

ha ha ha
call myfun funtion





25

myfun.__name__

'wrap'

myfun(x)函数被deco(fn)函数装饰后，函数对象的__name__属性会发生改变，发上输出。因为@deco就相当于执行了myfun = deco(myfun)，即变量myfun已经指向了wrap(*args,**kwargs)函数，这时myfun变量指向函数的__name__属性就是wrap，不再是原来的myfun()函数的__name__属性。如果要修正这个问题，可以直接引用python内置的functools.wraps方法，如下：

import functools
def deco(fn):
    @functools.wraps(fn)
    def wrap(*args,**kwargs):
        print('ha ha ha')
        print('call {0} funtion'.format(fn.__name__))
        return fn(*args,**kwargs)
    return wrap

@deco
def myfun(x):
    return x ** 2

myfun(25)

ha ha ha
call myfun funtion





625

myfun.__name__

'myfun'

总结：装饰器其实也是一个函数，此函数可以接收一个函数作为参数，并返回一个函数，即也是一个高阶函数。装饰器这个函数能让一个函数在调用时的前或后额外的执行一些操作来修改原调用的函数。