PYTHON-FUNCTOOLS模块

野性酷女 2022-10-01 00:40 194阅读 0赞

参考：[https://docs.python.org/2/library/functools.html][https_docs.python.org_2_library_functools.html]  
由于openstack中常常使用functools.partial去封装函数，借此机会了解一下python的functools模块  
在openstack nova/virt/libvirt/driver.py:2657 def spawn(xxx):

def spawn(self, context, instance, image_meta, injected_files, admin_password, network_info=None, block_device_info=None):
            disk_info = blockinfo.get_disk_info(CONF.libvirt.virt_type,
                                                instance,
                                                image_meta,
                                                block_device_info)
            injection_info = InjectionInfo(network_info=network_info,
                                           files=injected_files,
                                           admin_pass=admin_password)
            gen_confdrive = functools.partial(self._create_configdrive,
                                              context, instance,
                                              injection_info)

上述代码中gen\_confdrive使用了functools模块，此时执行gen\_confdrive函数时，效果等于self.\_create\_configdrive（context, instance, injection\_info），相当于封装了一层。

据官方代码：

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

代码中相当于对func进行了封装，也就是新的函数是原先func函数通过传入变量得到的新的函数，如下所示：

import functools
    
    def function(a, b):
        return a + b
    
    newfunc = functools.partial(function, 4, 5)
    print newfunc()

得到的结果是9，其实就是function(4, 5)的结果，当然你也可以通过先传入一个参数4，然后调用的时候再传入5如：newfunc(5),这样的结果也是一样的，所以这个函数只是包装了一层，便于调用：

>>> from functools import partial
    >>> basetwo = partial(int, base=2)
    >>> basetwo.__doc__ = 'Convert base 2 string to an int.'
    >>> basetwo('10010')
    18

接下来讲的是functools.update\_wrapper  
但是我们一般使用functools.wraps，因为functools.wraps等于partial(update\_wrapper, wrapped=wrapped, assigned=assigned, updated=updated).如下代码所示：

def function(a, b):
    ...     return a + b
    ... 
    >>> dir(function)
    ['__call__', '__class__', '__closure__', '__code__', '__defaults__', '__delattr__', '__dict__', '__doc__', '__format__', '__get__', '__getattribute__', '__globals__', '__hash__', '__init__', '__module__', '__name__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__sizeof__', '__str__', '__subclasshook__', 'func_closure', 'func_code', 'func_defaults', 'func_dict', 'func_doc', 'func_globals', 'func_name']
      newfunc = functools.partial(function, 4, 5)
    >>> dir(newfunc)
    ['__call__', '__class__', '__delattr__', '__dict__', '__doc__', '__format__', '__getattribute__', '__hash__', '__init__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__setstate__', '__sizeof__', '__str__', '__subclasshook__', 'args', 'func', 'keywords']

一般通过functools.partial修饰过的函数会少了一些属性，需要通过functools.wraps将诸如**doc**，**name**等属性加上！

[https_docs.python.org_2_library_functools.html]: https://docs.python.org/2/library/functools.html