Python--装饰器，,装饰器的应用场景，较

Python--装饰器，,装饰器的应用场景，较

装饰器的应用场景，较为经典的有插入日志、性能测试、事务处理等。 上代码：

def foo():    print(‘foo‘)

此段代码，如果我们想在不改变原有功能的基础上添加新功能，可以这样做：

def foo():    print(‘foo‘)def wrapper(func):    def inner():        print(‘before‘)        func()        print(‘after‘)    return innerfoo=wrapper(foo)foo()

打印结果：
before
foo
after

Python提供了一个语法糖来降低字符输入量。

def wrapper(func):    def inner():        print(‘before‘)        func()        print(‘after‘)    return inner@wrapper    #wrapper=wrapper(foo)def foo():    print(‘foo‘)foo()

打印结果：
before
foo
after

让我们来关注一下@wrapper的写法，在foo函数定义上加上这一行与另外写foo = wrapper(foo)完全等价，@并没有另外的魔力。

除了字符输入少了一些，还有一个额外的好处：这样看上去更有装饰器的感觉。 这样还不能满足我们的需求，我们想给foo函数传参数的话，如何处理？

上代码：

def wrapper(func):    def inner(sth):        print(‘before‘)        func(sth)        print(‘after‘)    return inner@wrapper    #wrapper=wrapper(foo)def foo(sth):    print(sth)foo(‘apple‘)

打印结果：
before
apple
after

这个是传一个参数的情况，如果我们想传的参数很多，除了多个位置参数之外，还想要传关键字参数，这个时候怎么办呢？ func函数通过*args,**kwargs接收实参即可完美实现！

def wrapper(func):    def inner(*args,**kwargs):        print(‘before‘)        func(*args,**kwargs)        print(‘after‘)    return inner@wrapper    #wrapper=wrapper(foo)def foo(sth1,sth2,sth3,sth4):    print(sth1,sth2,sth3,sth4)foo(‘apple‘,‘orange‘,‘banana‘,‘lemon‘)

打印结果：
before
apple orange banana lemon
after

进一步，我们的foo函数有时候需要使用到返回值，然后如何处理呢？

import timedef wrapper(func):    def inner(*args,**kwargs):        print(‘before‘)        ret = func(*args,**kwargs)        print(‘after‘)        print(ret)        return ret    return inner@wrapper    #foo=wrapper(foo)def foo(arg):    print(arg)    return 111res=foo(‘apple‘)time.sleep(1)print(res)打印结果：beforeappleafter111111（1秒后打印）

此外，functools模块提供有装饰器：wraps（wrapped，assigned，updated）。

函数是有几个特殊属性比如函数名，在被装饰后，上例中的函数名foo会变成包装函数的名字inner，我们可以看下例：

def wrapper(func):    def inner(*args, **kwargs):        print("before...")        ret = func(*args, **kwargs)        print("after/...")        return ret    return inner@wrapper   #  foo = wrapper(foo) -->innerdef foo(sth):    print(sth)    return "aaa"print(foo.__name__)打印结果：inner

这时候wraps就体现了它的作用：它能将装饰过的函数的特殊属性保留。

import functoolsdef wrapper(func):    @functools.wraps(func)    def inner(*args, **kwargs):        print("before...")        ret = func(*args, **kwargs)        print("after/...")        return ret    return inner@wrapper   #  foo = wrapper(foo) -->innerdef foo(sth):    print(sth)    return "aaa"print(foo.__name__)打印结果：foo

关于装饰器，一些有意思的小练习可以拿来做。

一：编写函数，（函数执行的时间是随机的）二：编写装饰器，为函数加上统计时间的功能三：编写装饰器，为函数加上认证的功能四：编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码注意：从文件中读出字符串形式的字典，可以用eval(‘{"name":"egon","password":"123"}‘)转成字典格式五：编写下载网页内容的函数，要求功能是：用户传入一个url，函数返回下载页面的结果六：为题目五编写装饰器，实现缓存网页内容的功能：具体：实现下载的页面存放于文件中，如果文件内有值（文件大小不为0），就优先从文件中读取网页内容，否则，就去下载，然后存到文件中七：还记得我们用函数对象的概念，制作一个函数字典的操作吗，来来来，我们有更高大上的做法，在文件开头声明一个空字典，然后在每个函数前加上装饰器，完成自动添加到字典的操作

Python--装饰器