functools模块2个常用函数

　　functools是python2.5被引人的,一些工具函数放在此包里，python3中增加了更多工具函数，业务开发时大多情况下用不到，此处介绍使用频率较高的2个函数。

　　partial函数(偏函数)

　　这个函数在基础教程篇介绍过，可查看偏函数，partial函数可以把一个函数的某些参数预先设置默认值，返回一个新的函数，调用这个新函数会更简单。

# -*- coding: utf-8 -*-

import functools

def showarg(*args, **kw):
    print(args)
    print(kw)

p1=functools.partial(showarg, 1,2,3)
p1()
p1(4,5,6)
p1(a='python', b='dong')

p2=functools.partial(showarg, a=3,b='linux')
p2()
p2(1,2)
p2(a='python', b='dong')

(1, 2, 3)
{}
(1, 2, 3, 4, 5, 6)
{}
(1, 2, 3)
{'a': 'python', 'b': 'dong'}
()
{'a': 3, 'b': 'linux'}
(1, 2)
{'a': 3, 'b': 'linux'}
()
{'a': 'python', 'b': 'dong'}

　　wraps函数

　　使用装饰器时，有一些细节需要被注意。例如，被装饰后的函数其实已经是另外一个函数了（函数名等函数属性会发生改变）。添加后对测试结果有一些影响，例如:

# -*- coding: utf-8 -*-

def note(func):
    "note function"
    def wrapper():
        "wrapper function"
        print('note something')
        return func()
    return wrapper

@note
def test():
    "test function"
    print('I am test')

test()
print(test.__doc__)

note something
I am test
wrapper function

　　所以，Python的functools包中提供了一个叫wraps的装饰器来消除这样的副作用。例如：  

# -*- coding: utf-8 -*-
import functools
def note(func):
    "note function"
    @functools.wraps(func)
    def wrapper():
        "wrapper function"
        print('note something')
        return func()
    return wrapper

@note
def test():
    "test function"
    print('I am test')

test()
print(test.__doc__)

note something
I am test
test function

很赞哦！

python编程网提示：转载请注明来源www.python66.com。
有宝贵意见可添加站长微信(底部)，获取技术资料请到公众号(底部)。同行交流请加群