5分钟了解 Python 装饰器

1、解释器入门
 
ASP站长网写代码要遵循开放封闭原则，那么什么是开放封闭原则呢，简单的说就是：已经实现的功能代码块不允许被修改，但可以被扩展。即：
 
开放：对扩展开发；封闭：已经实现的代码块
 
那么问题来了如何在不更改原有代码前提下实现性能的添加，装饰器就是一个很好的用法
 
例如：我需要在“登录系统前添加验证功能”，在不更改现有的代码该如何实现？
 
def test(name):
    print("登录系统___%s" % name)
 
test("张三")
 
1）方法1：也许有人会想到使用闭包实现，如
 
def outer(func):
    def inner(name):
        print("验证——————————")
        func(name)
    return inner
 
 
def test(name):
    print("登录系统___%s" % name)
 
 
test = outer(test)
test("张三")
 
#输出结果
验证——————————
登录系统___张三
 
那么问题来了，这样的话在每个文件不同作用域里调用test功能之前都需要添加第12行： test = outer(test)，那如果这样的需要添加位置较多的话也不现实，且比较难于查找，出现缺漏的结果也可以自行想象。这里就可以使用方法2
 
2）方法2：使用装饰器
 
def outer(func):
    def inner(name):
        print("验证——————————")
        func(name)
    return inner
 
 
@outer
def test(name):
    print("登录系统___%s" % name)
 
 
test("张三")
 
#输出结果
验证——————————
登录系统___张三
 
这里可以看到只需要在test函数前加上 @outer  就可以实现，也可以这么理解  @outer --> test = outer(test)
 
当装饰的函数有参数时装饰器内部也必须要定义形参，即 第 2 行，在第 4 行时也必须传递参数进行调用
 
多个函数使用同样的验证功能时也是如此
 
def outer(func):
    def inner(name):
        print("验证——————————")
        func(name)
    return inner
 
 
@outer
def test1(name):
    print("登录系统___%s" % name)
 
 
@outer
def test2(name):
    print("查询余额___%s" % name)
 
 
test1("张三")
test2("李四")
 
#输出结果
验证——————————
登录系统___张三
验证——————————
查询余额___李四
 
2、多个装饰器
 
def outer1(func):
    print("_____装饰器outer1____")
    def inner1():
        print("———inner1———————")
        func()
    return inner1
 
def outer2(func):
    print("_____装饰器outer2____")
    def inner2():
        print("———inner2———————")
        func()
    return inner2
 
@outer1
@outer2
def test():
    pass
 
test()
 
#输出结果
_____装饰器outer2____
_____装饰器outer1____
———inner1———————
———inner2———————
 
从结果也可以看到，当有多个装饰器时，装饰器是从内往外装饰，即：1）@outer2-->  test = outer2(test)  2) @outer1--> test = outer1(test)
 
执行到，15，16开始装饰时就会有输出输出23，24行，原因请看上边1）2），在装饰时也会执行outer2()，outer1()两个函数，所以会有输出结果
 
3、装饰器带返回值
 
def outer(func):
    def inner(a, b):
        ret = func(a, b)
        return ret
    return inner
 
@outer
def test(a, b):
    return a + b
 
ret = test(5, 2)
print("-----%d" % ret)
 
#输出结果
-----7
 
当装饰的函数有返回值时需要将返回值返回，如第 3,4 行
 
4、通用装饰器
 
def outer(func):
    def inner(*args, **kwargs):
        ret = func(*args, **kwargs)
        return ret
    return inner
 
@outer
def test1(a):
    return a ** 2
 
@outer
def test2(a, b, c):
    return a * b * c
 
print(test1(3))
print(test2(2, 3, c=5))
 
#输出结果
9
30
 
当多个函数都是用同一个装饰器的时候，参数不一致的问题，就的使用通用装饰器来解决，通用即所有的函数都适用的意思。
 
如上所示，test1与test2函数的参数不一致，也可以用同样的装饰器装饰，第2,3行解决
 
5、带参数的装饰器
 
def f1(flag=1):
    def outer(func):
        def inner(*args, **kwargs):
            if (flag == 2):
                print("______%d" % flag)
                ret = func(*args, **kwargs)
            else:
                print("______%d" % flag)
                ret = func(*args, **kwargs)
            return ret
        return inner
    return outer
 
@f1(2)
def test(a):
    return a ** 2
 
print(test(3))
 
#输出结果
______2