Python常用模組之四 funsctools

標籤：

functools 模組中有三個主要的函數 partial(), update_wrapper() 和 wraps()。

1、partial(func[,args][, *keywords])

　　functools.partial 通過封裝手法，允許我們 "重新定義" 函數簽名。用一些預設參數封裝一個可調用對象,返回結果是可調用對象，並且可以像原始對象一樣對待凍結部分函數位置函數或關鍵字參數，簡化函數,更少更靈活的函數參數調用。

from functools import partialint2 = partial(int, base=2)print int2(‘11‘) # 3print int2(‘101‘) # 5

當函數的參數個數太多，需要簡化時，使用functools.partial可以建立一個新的函數，這個新函數可以固定住原函數的部分參數，從而在調用時更簡單。

2、functool.update_wrapper

　　預設partial對象沒有__name__和__doc__, 這種情況下，對於裝飾器函數非常難以debug.使用update_wrapper(),從原始對象拷貝或加入現有partial對象。它可以把被封裝函數的__name__、module、__doc__和 __dict__都複製到封裝函數去(模組層級別常量WRAPPER_ASSIGNMENTS, WRAPPER_UPDATES)。

　　這個函數主要用在裝飾器函數中，裝飾器返回函數反射得到的是封裝函數的函數定義而不是原始函數定義

>>> functools.WRAPPER_ASSIGNMENTS(‘__module__‘, ‘__name__‘, ‘__doc__‘)>>> functools.WRAPPER_UPDATES(‘__dict__‘,)

#!/usr/bin/env python# encoding: utf-8def wrap(func):    def call_it(*args, **kwargs):        """wrap func: call_it"""        print ‘before call‘        return func(*args, **kwargs)    return call_it@wrapdef hello():    """say hello"""    print ‘hello world‘from functools import update_wrapperdef wrap2(func):    def call_it(*args, **kwargs):        """wrap func: call_it2"""        print ‘before call‘        return func(*args, **kwargs)    return update_wrapper(call_it, func)@wrap2def hello2():    """test hello"""    print ‘hello world2‘if __name__ == ‘__main__‘:    hello()    print hello.__name__    print hello.__doc__    print    hello2()    print hello2.__name__    print hello2.__doc__
結果：before callhello worldcall_itwrap func: call_itbefore callhello world2hello2test hello

3、functool.wraps

調用函數裝飾器partial(update_wrapper, wrapped=wrapped, assigned=assigned, updated=updated)的簡寫。

from functools import wrapsdef wrap3(func):    @wraps(func)    def call_it(*args, **kwargs):        """wrap func: call_it2"""        print ‘before call‘        return func(*args, **kwargs)    return call_it@wrap3def hello3():    """test hello 3"""    print ‘hello world3‘結果：before callhello world3hello3test hello 3

詳見：http://www.wklken.me/posts/2013/08/18/python-extra-functools.html

Python常用模組之四 funsctools