装饰器
由于函数也是一个对象,而且函数对象可以被赋值给变量,所以,通过变量也能调用该函数。
函数对象有一个__name__
属性,可以拿到函数的名字.
在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)。本质上,decorator就是一个返回函数的高阶函数。
看代码:
def log(fun):
def wrapper(*args,**kw):
print("call %s" % fun.__name__)
return fun(*args,**kw)
return wrapper
@log
def now():
print("2016-12-28")
当我们在调用函数now() 输出结果.
call now
2016-12-28
分析一下:
@log就相当于 now = log(now) ,那么此时now是wrapper函数,你在now后面加个()就相当于调用wrapper函数,wrapper函数是调用原本的函数,就是now()所以就会输出了.
由于@log就相当于 now = log(now),当我们调用 log(now) 的时候返回的是wrapper函数,因此此时now.__name__ 就是wrapper.所以,需要把原始函数的__name__等属性复制到wrapper()函数中,否则,有些依赖函数签名的代码执行就会出错。Python内置的functools.wraps就可以做这件事.所以最终的写法:
import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('call %s():' % func.__name__)
return func(*args, **kw)
return wrapper
偏函数
Python的functools模块提供了很多有用的功能,其中一个就是偏函数(Partial function)。要注意,这里的偏函数和数学意义上的偏函数不一样。
假如我们想把字符串按二进制转换成整数,可以这么写:
>>> int('100',base = 2)
4
>>>
或者你还可以定义函数:
def int2(x,base=2):
return int(x,base)
还有一种更简单的方法,我们可以使用functools.partial帮助我们创建一个偏函数:
>>> import functools
>>> int2 = functools.partial(int, base=2)
>>> int2('1000000')
64
functools.partial的作用就是,把一个函数的某些参数给固定住(也就是设置默认值),返回一个新的函数,调用这个新函数会更简单。
最后,创建偏函数时,实际上可以接收函数对象、*args和**kw这3个参数,当传入:
int2 = functools.partial(int, base=2)
实际上固定了int()函数的关键字参数base,也就是:
int2('10010')
相当于:
kw = { 'base': 2 }
int('10010', **kw)
当传入:
max2 = functools.partial(max, 10)
实际上会把10作为*args的一部分自动加到左边,也就是:
max2(5, 6, 7)
相当于:
args = (10, 5, 6, 7)
max(*args)
结果为10。