Python学习系列之装饰器

装饰器的作用

装饰器用于装饰某个函数、方法或者类,它可以让这个函数执行之前或者执行之后做一些操作

手工实现一个装饰器

def outer(some_func):   #装饰器  $1
    def inner():
        print("before some_func")$3
        ret = some_func()        $2
        print('after some_func') $4
    return inner
 
def foo():              #被装饰的函数
    print(1)
decorator = outer(foo)  #把foo函数当做参数传给outer函数 $5
decorator()             #在执行函数

让我们一步步来分析

1.outer函数是装饰器

2.foo函数是被装饰

3.$5处把foo函数当做参数传进outer函数

4.$1处outer函数接收foo函数这个参数

5.$2处inner函数接收outer函数传进来的参数(此例子为:foo函数),ret用来接收foo函数的结果.(inner函数在本函数内找some_func参数找不到就会向上一级找,就找到了,这是python的作用域(闭包),)

6.$3处是在foo函数的结果之前加一段代码,$4处是在foo函数的结果之后加一段代码

所以说原理就是把函数传到装饰器函数里,被装饰函数执行之前先执行一段你想执行的diamante,被装饰函数执行之后再执行一段你想执行的代码

利用装饰器装饰一个函数

定义被装饰函数

代码:
#!/usr/bin/env python
# _*_ coding: utf-8 _*_
def foo(arg):    # 接受一个参数arg
    print(arg)   # 输出这个参数
foo("Hello World!")  # 调用脚本并且传入参数

现在要在foo函数执行之前和执行之后来执行一些操作

#!/usr/bin/env python
# _*_ coding: utf-8 _*_
def decorator(func):  # 创建一个装饰器函数，接受的参数arg参数就是func函数名
    def inner(*args, **kwargs):
        print("执行函数之前")
        ret = func(*args, **kwargs)
        print("执行函数之后")
        return ret
    return inner
@decorator  # 如果要让某个函数使用装饰器，只需要在这个函数上面加上@+装饰器名
def foo(arg):
    print(arg)
foo("Hello World!")  #执行函数,测试

测试结果

执行函数之前
Hello World!
执行函数之后

多个装饰器装饰同一个函数

#!/usr/bin/env python
# _*_ coding: utf-8 _*_
def decorator1(func):
    def inner():
        print("开始之前执行装饰器01")
        ret = func()
        print("结束之后执行装饰器01")
        return ret
    return inner
def decorator2(func):
    def inner():
        print("decorator2>>>Start...")
        ret = func()
        print("decorator2>>>End...")
        return ret
    return inner
@decorator1
@decorator2
def index():
    print("执行函数...")
index()#执行函数测试

结果

开始之前执行装饰器01
decorator2>>>Start...
执行函数...
decorator2>>>End...
结束之后执行装饰器01