python中装饰器的使用

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/niubiqigai/article/details/82849868

python中装饰器的使用

装饰器在不改变函数的原有代码的前提下给函数添加新的功能，是程序开发中经常会用到的一个功能，用好了装饰器，开发效率如虎添翼。

一、装饰器的作用

1. 引入日志
2. 函数执行时间统计
3. 执行函数前预备处理
4. 执行函数后清理功能
5. 权限校验等场景
6. 缓存

二、装饰器示例
例1. 无参数的函数

import time

def time_fun(func):
    def wrapped_func():
        print("%s called at %s" % (func.__name__, time.ctime()))
        func()
    return wrapped_func

@time_fun
def foo():
    print("I am foo")

foo()

输出结果：

foo called at Wed Sep 26 10:14:06 2018
I am foo

上面代码理解装饰器执行行为可理解成(重点理解)

foo = time_fun(foo)
# foo先作为参数赋值给func后,foo接收指向timefun返回的wrapped_func
foo()
# 调用foo(),即等价调用wrapped_func()
# 内部函数wrapped_func被引用，所以外部函数的func变量(自由变量)并没有释放
# func里保存的是原foo函数对象

例2. 被装饰的函数有参数

import time

def time_fun(func):
    def wrapped_func(a, b):
        print("%s called at %s" % (func.__name__, time.ctime()))
        func(a, b)
    return wrapped_func

@time_fun
def foo(a, b):
    print(a+b)

foo(3, 5)

输出结果：

foo called at Wed Sep 26 10:11:54 2018
8

例3. 被装饰的函数有不定长参数

import time

def time_fun(func):
    def wrapped_func(*args, **kwargs):
        print("%s called at %s"%(func.__name__, time.ctime()))
        func(*args, **kwargs)
    return wrapped_func

@time_fun
def foo(a, b, c):
    print(a+b+c)

foo(2, 4, 6)

输出结果：

foo called at Wed Sep 26 10:17:22 2018
12

例4. 装饰器中的return

import time

def time_fun(func):
    def wrapped_func():
        print("%s called at %s" % (func.__name__, time.ctime()))
        return func()
    return wrapped_func

@time_fun
def foo():
    print("I am foo")

@time_fun
def getInfo():
    return '---test---'

foo()
time.sleep(2)

getinfo = getInfo()
time.sleep(2)  # 测试用，这个时候并未输出'---test---'
print(getinfo)

输出结果：

扫描二维码关注公众号，回复： 4798510 查看本文章

foo called at Wed Sep 26 10:30:37 2018
I am foo
getInfo called at Wed Sep 26 10:30:39 2018
---test---

• 总结：
  一般情况下为了让装饰器更通用，可以有return