首页 > 代码库 > Python的functools模块

Python的functools模块

2024-07-26 15:40:45   219人阅读

这个模块提供了3个有趣的函数，这里介绍下其用法。

首先是partial函数，它可以重新绑定函数的可选参数，生成一个callable的partial对象

  1. >>> int(‘10‘) # 实际上等同于int(‘10‘, base=10)和int(‘10‘, 10)  
  2. 10  
  3. >>> int(‘10‘, 2) # 实际上是int(‘10‘, base=2)的缩写  
  4. 2  
  5. >>> from functools import partial  
  6. >>> int2 = partial(int, 2) # 这里我没写base，结果就出错了  
  7. >>> int2(‘10‘)  
  8. Traceback (most recent call last):  
  9.   File "<stdin>", line 1, in <module>  
  10. TypeError: an integer is required  
  11. >>> int2 = partial(int, base=2) # 把base参数绑定在int2这个函数里  
  12. >>> int2(‘10‘) # 现在缺省参数base被设为2了  
  13. 2  
  14. >>> int2(‘10‘, 3) # 没加base，结果又出错了  
  15. Traceback (most recent call last):  
  16.   File "<stdin>", line 1, in <module>  
  17. TypeError: keyword parameter ‘base‘ was given by position and by name  
  18. >>> int2(‘10‘, base=3)  
  19. 3  
  20. >>> type(int2)  
  21. <type ‘functools.partial‘>  

从中可以看出，唯一要注意的是可选参数必须写出参数名。

接着是update_wrapper函数，它可以把被封装函数的__name__、__module__、__doc__和 __dict__都复制到封装函数去：

  1. #-*- coding: gbk -*-  
  2.   
  3. def thisIsliving(fun):  
  4.   def living(*args, **kw):  
  5.     return fun(*args, **kw) + ‘活着就是吃嘛。‘  
  6.   return living  
  7.  
  8. @thisIsliving  
  9. def whatIsLiving():  
  10.   "什么是活着"  
  11.   return ‘对啊，怎样才算活着呢？‘  
  12.   
  13. print whatIsLiving()  
  14. print whatIsLiving.__doc__  
  15.   
  16. print  
  17.   
  18. from functools import update_wrapper  
  19. def thisIsliving(fun):  
  20.   def living(*args, **kw):  
  21.     return fun(*args, **kw) + ‘活着就是吃嘛。‘  
  22.   return update_wrapper(living, fun)  
  23.  
  24. @thisIsliving  
  25. def whatIsLiving():  
  26.   "什么是活着"  
  27.   return ‘对啊，怎样才算活着呢？‘  
  28.   
  29. print whatIsLiving()  
  30. print whatIsLiving.__doc__  

结果：
对啊，怎样才算活着呢？活着就是吃嘛。
None

对啊，怎样才算活着呢？活着就是吃嘛。
什么是活着
不过也没多大用处，毕竟只是少写了4行赋值语句而已。

最后是wraps函数，它将update_wrapper也封装了进来：

[python]  view plain copy
  1. #-*- coding: gbk -*-  
  2.   
  3. from functools import wraps  
  4.   
  5. def thisIsliving(fun):  
  6.   @wraps(fun)  
  7.   def living(*args, **kw):  
  8.     return fun(*args, **kw) + ‘活着就是吃嘛。‘  
  9.   return living  
  10.  
  11. @thisIsliving  
  12. def whatIsLiving():  
  13.   "什么是活着"  
  14.   return ‘对啊，怎样才算活着呢？‘  
  15.   
  16. print whatIsLiving()  
  17. print whatIsLiving.__doc__  

Python的functools模块


联系
我们