functools模块

2018-08-29  本文已影响0人  秋幻旎苏

一.update_wrapper

该函数用于更新包装函数(wrapper),使它看起来像被包装的原函数一样。
该函数主要用于装饰器函数的定义中,置于包装函数之前。如果没有对包装函数进行更新,那么被装饰后的函数所具有的元信息就会变为包装函数的元信息,而不是原函数的元信息。

WRAPPER_ASSIGNMENTS = ('__module__', '__name__', '__qualname__', '__doc__',  '__annotations__')
WRAPPER_UPDATES = ('__dict__',)
update_wrapper(wrapper,
               wrapped,
               assigned = WRAPPER_ASSIGNMENTS,
               updated = WRAPPER_UPDATES):

该函数自动在wrapper函数添加一个wrapped属性,保留着wrapped函数

import datetime, time, functools
def logger(duration, func=lambda name, duration: print('{} took {}s'.format(name, duration))):
    def _logger(fn):
        def wrapper(*args,**kwargs):
            start = datetime.datetime.now()
            ret = fn(*args,**kwargs)
            delta = (datetime.datetime.now() - start).total_seconds() 
            if delta > duration:
                func(fn.__name__, duration) 
            return ret
        return functools.update_wrapper(wrapper, fn) 
    return _logger
@logger(5) # add = logger(5)(add) 
def add(x,y):
    time.sleep(1) 
    return x + y
print(add(5, 6), add.__name__, add.__wrapped__, add.__dict__, sep='\n')

二.functools.wraps

def wraps(wrapped,
          assigned = WRAPPER_ASSIGNMENTS,
          updated = WRAPPER_UPDATES):

    return partial(update_wrapper, wrapped=wrapped,
                   assigned=assigned, updated=updated)

这是一个便捷函数,定义一个wrapper函数时,作为函数装饰器调用update_wrapper()。它等价于partial(update_wrapper, wrapped=wrapped, assigned=assigned, updated=updated),返回一个形参wrapped 有默认值的新的update_warpper 函数

import datetime, time, functools
def logger(duration, func=lambda name, duration: print('{} took {}s'.format(name, duration))):
    def _logger(fn):
        @functools.wraps(fn)
        def wrapper(*args,**kwargs):
            start = datetime.datetime.now()
            ret = fn(*args,**kwargs)
            delta = (datetime.datetime.now() - start).total_seconds() 
            if delta > duration:
                func(fn.__name__, duration) 
            return ret
        return wrapper
    return _logger
@logger(5) # add = logger(5)(add) 
def add(x,y):
    time.sleep(1) 
    return x + y
print(add(5, 6), add.__name__, add.__wrapped__, add.__dict__, sep='\n')

三.functools.reduce

reduce(function, sequence, initial=None) -> value

等价于

def reduce(function, iterable, initializer=None):
    it = iter(iterable)
    if initializer is None:
        value = next(it)
    else:
        value = initializer
    for element in it:
        value = function(value, element)
    return value

具体应用如下:

from functools import reduce
def add(x,y):
    return x + y
reduce(add,range(5))

四.functools.partial 偏函数

functools.partial(func, *args, **keywords) -> newfunc
把函数部分的参数固定下来,相当于为部分的参数添加了一个固定的默认值,形成一个新的函数并且返回,是对原函数的封装
partial 函数本质

def partial(func, *args, **keywords):
    def newfunc(*fargs, **fkeywords): # 包装函数
        newkeywords = keywords.copy() newkeywords.update(fkeywords)
        return func(*(args + fargs), **newkeywords)
newfunc.func = func # 保留原函数
newfunc.args = args # 保留原函数的位置参数 
newfunc.keywords = keywords # 保留原函数的关键字参数参数 
return newfunc

五.functools.lru_cache

functools.lru_cache(maxsize=128, typed=False)

装饰器用一个有记忆的调用包装一个函数,它可以保存最近maxsize次调用。当使用同样的参数定期调用费时或I/O绑定的函数时,它可以节省时间。
因为使用字典缓存结果,所以函数的位置和关键字必须是可以哈希的。
为了帮助测量缓存的有效性并且调整maxsize 参数,可以使用cache_info() 返回一个命名元组,包含hits,misses,maxsize和currsize。在多线程环境中,hits和misses是近似值。
装饰器还提供了cache_clean() 用于清除缓存。

上一篇 下一篇

猜你喜欢

热点阅读