来源:python中国网 时间:2019-06-06

  装饰器是程序开发中经常会用到的一个功能,用好了装饰器,开发效率如虎添翼。

  场景:

  公司个基础平台部门,基础平台负责提供底层的功能,如:数据库操作、redis调用、监控API等功能。业务部门使用基础功能时,只需调用基础平台提供的功能即可。如下:

# -*- coding: utf-8 -*-

ldef f1():
    print('f1')

def f2():
    print('f2')

  现在需要对基础平台的所有功能进行重构,为平台提供的所有功能添加验证机制,即:执行功能前,先进行验证

# -*- coding: utf-8 -*-
def check_login():
    # 验证1
    # 验证2
    # 验证3
    pass


def f1():

    check_login()
    print('f1')


def f2():

    check_login()
    print('f2')  


  写代码要遵循开放封闭原则,简单来说,它规定已经实现的功能代码不允许被修改,但可以被扩展,如果将开放封闭原则应用在上述需求中,那么就不允许在函数f1、f2、f3、f4的内部进行修改代码,利用装饰器来实现!

# -*- coding: utf-8 -*-
def w1(func):
    def inner():
        # 验证1
        # 验证2
        # 验证3
        func()
    return inner

@w1
def f1():
    print('f1')
@w1
def f2():
    print('f2')


  装饰器的原理是啥?

# -*- coding: utf-8 -*-
#单独以f1为例:

def w1(func):
    def inner():
        # 验证1
        # 验证2
        # 验证3
        func()
    return inner
@w1
def f1():
    print('f1')

  python解释器就会从上到下解释代码,步骤如下:

  1)def w1(func):==>将w1函数加载到内存

  2)@w1

  从表面上看解释器仅仅会解释这两句代码,但是@w1这一句代码里却有大文章,@函数名是python的一种语法糖。

  上例@w1内部会执行一下操作:

  首先,执行w1函数,并将@w1下面的函数作为w1函数的参数,即:@w1等价于w1(f1)所以,内部就会去执行:

# -*- coding: utf-8 -*-
def inner(): 
    #验证 1
    #验证 2
    #验证 3
    f1()     # func是参数,此时 func 等于 f1 
return inner  # 返回的 inner,inner代表的是函数,非执行函数 ,其实就是将原来的 f1 函数塞进另外一个函数中

  其次,将执行完的w1函数返回值赋值给@w1下面的函数的函数名f1即将w1的返回值再重新赋值给f1,即:

# -*- coding: utf-8 -*-
新f1 = def inner(): 
            #验证 1
            #验证 2
            #验证 3
            原来f1()
        return inner

  所以,以后业务部门想要执行f1函数时,就会执行新f1函数,在新f1函数内部先执行验证,再执行原来的f1函数,然后将原来f1函数的返回值返回给了业务调用者。

  如此一来,即执行了验证的功能,又执行了原来f1函数的内容,并将原f1函数返回值返回给业务调用着。

  再议装饰器案例。

# -*- coding: utf-8 -*-
#定义函数:完成包裹数据
def makeBold(fn):
    def wrapped():
        return "" + fn() + ""
    return wrapped

#定义函数:完成包裹数据
def makeItalic(fn):
    def wrapped():
        return "" + fn() + ""
    return wrapped

@makeBold
def test1():
    return "hello world-1"

@makeItalic
def test2():
    return "hello world-2"

@makeBold
@makeItalic
def test3():
    return "hello world-3"

print(test1())
print(test2())
print(test3())


D:python3installpython.exe D:/python/py3script/python66.py
hello world-1
hello world-2
hello world-3

Process finished with exit code 0