【Python】装饰器 & 偏函数

【装饰器】

1、最简单的Decorator。

def author(f):

    def addName():

        print('My name is xkfx.\n')

        f()

    return addName

@author

def helloWorld():

    print('Hello, World')

把@log放到now()函数的定义处，相当于执行了语句：helloWorld = author(helloWorld)。

>>> helloWorld()

My name is xkfx.

Hello, World

这个时候查看函数的名字会发现

>>> helloWorld.__name__

'addName'

2、稍微修改一下

def author(f):

    def addName():

        print('My name is xkfx.[edition-2]\n')

        return f()

    return addName

@author

def helloWorld():

    print('Hello, World')

>>> helloWorld()

My name is xkfx.[edition-2]

Hello, World

将直接调用f();改为return f();执行效果是一样的，先打印author name然后执行原函数。

3、这个时候希望由用户输入author名，也就是包装器要接收一个字符串参数，那该怎么办？这就需要一个嵌套的decorator：

def author(authorName):

    def decorator(func):

        def wrapper(*args, **kw): # wrapper是包装纸的意思

            print("author:" + authorName)

            return func() # 调用原函数，注意是有括号的

        return wrapper # 返回包装后的函数

    return decorator # 返回真正的包装器

>>> from test import author

>>> @author('xkfx')

... def printHello():

...     print('Hello\n')

...

>>> printHello()

author:xkfx

Hello

查看被包装后的函数名：

>>> printHello.__name__

'wrapper'

需要把原始函数的__name__等属性复制到wrapper()函数中，否则，有些依赖函数签名的代码执行就会出错。而做这件事情本身也属于“拓展函数功能”的范畴，用Python库内置的装饰器完成就可以了。只需记住在定义wrapper()的前面加上@functools.wraps(func)即可，如下所示：

import functools

def author(authorName):

    def decorator(func):

        @functools.wraps(func)

        def wrapper(*args, **kw): # wrapper是包装纸的意思

            print("author:" + authorName)

            return func() # 调用原函数，注意是有括号的

        return wrapper # 返回包装后的函数

    return decorator # 返回真正的包装器

再走一遍流程：

>>> printHello.__name__

'printHello'

4、练习。

import functools

def log(func):

    @functools.wraps(func)

    def wrapper(*args, **kw):

        print('begin call')

        result = func(*args, **kw)

        print('after call')

        return result

    return wrapper

@log

def f():

    pass

需求二不会做。

【偏函数】

当函数的参数太多的时候，可以选择将部分参数固定来简化函数，固定的方式就是使用functools模块中的partial函数。

一个简单的例子，固定谈话对象：

def say(words, name):

    print(words + ',' + name)

原函数需要两个参数：

>>> say('hello', 'xkfx')

hello,xkfx

经简化后得到的“偏”函数只需要传入一个参数就够了：

>>> say2 = functools.partial(say, name='xkfx')

>>> say2('hi')

hi,xkfx

当然固定多个参数肯定是可行的，（实际上是将要固定关键字参数传入了一个隐式的dict）

>>> say2 = functools.partial(say, name='xkfx', age=19)

>>> say2('hi')

hi,xkfx

Your age is 19

巴特西

【Python】装饰器 & 偏函数

最新文章

热门文章