python当中的装饰器

Posted 2021-01-27 DESOLATE.X

1.装饰器

　　首先我们来说一下一个软件的设计原则:开闭原则,又被称为开发封闭原则,你的代码对功能的扩展是开放的,你的程序对修改源代码是封闭的.这样的软件设计思路可以更好的维护和开发.

　　开放:对功能扩展开放

　　封闭:对修改代码封闭

　　接下来我们看装饰器.首先我们先模拟一下女娲造人.

def  create():
    print(\'女娲很厉害,捏个泥人吹口气就成了人\')

create_people()

　　现在我们让这个函数加个浇水功能.

def  create():
    print(\'浇水\')
    print(\'女娲很厉害,捏个泥人吹口气就成了人\')

create_people()

　　但是,现在就违反了开闭原则,如果每次要添加新功能,那这样就等于是修改了源代码.因为开闭原则对修改是封闭的,那怎么办,我们可以这样做

def water():
    print(\'先浇水\')
    create()
water()

　　这样做的问题是.函数写好了,重新创建了这个函数,在这之前访问过这个函数的人必须要修改代码来访问新的函数water(),所以现在就用到了装饰器,装饰器的作用就是在不修改原有代码的基础上,给函数扩展功能

def create_people():
    print(\'女娲很厉害随随便便就造人\')
def water(fn):
    def inner():
        print(\'浇浇水\')
        fn()
        print(\'施肥\')
    return inner

# # create_people() # 这个就不行了.
# warter() # 访问浇水就好了

# func = water(create_people)
# func() # 有人问了. 下游访问的不不依然是func么, 不不还是要改么?
create_people = water(create_people)
create_people() # 这回就好了吧

　说一下执行流程:

结论: 我们使用water函数把create_people给包装了一下. 在不修改create_people的前提下.
完成了对create_people函数的功能添加

 

下面我们给出装饰器的完整模型代码

def wrapper(fn):  # fn是目标函数
    def inner(*args,**kwargs):  # 为了目标函数的传参
        \'\'\'这里是执行函数前的内容\'\'\'
            ret = fn(*args,**kwargs)  # 调用目标函数,ret是目标函数的返回值
        \'\'\'这里是执行目标函数后的内容
            return ret  # 把目标函数的返回值返回.保证函数正常的结束
    return inner

@wrapper  # 这里是语法糖
def func():
    print(\'我是目标函数")

func() # 调用目标函数

　　

二.带参数的装饰器

　　注意: 咱们之前的写法是@wrapper 其中wrapper是一个函数. 那么也就是说. 如果我能让wrapper这里换成个函数就行了了. wrapper(True)返回的结果是wrapper也是⼀一个函数啊. 刚刚

好和前面的@组合成一个@wrapper. 依然还是原来那个装饰器. 只不过这里套了3层. 但你要
能看懂. 其实还是原来那个装饰器@wrapper

　　

　　执行步骤:　　　　

　　　　先执行wrapper(True) 然后再@返回值. 返回值恰好是wrapper. 结果就是@wrapper

下面是装饰器传参的通用写法

def wrapper_out(flag):
    def wrapper(fn):
        def inner(*args,**kwargs):
            if flag == True:
                \'\'\'执行目标函数前的位置\'\'\'
                ret = fn(*args,**kwargs)
                \'\'\'执行目标函数后的位置\'\'\'
                return ret
            else:
                ret = fn(*args,**args)
                return ret
        return inner
    return wrapper

@wrapper_out(True)
def func():
    print(\'这里是目标函数\')
    
func()
                

　　

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

def wrapper1(fn):
    def inner(*args,**kwargs):
        print(\'111\')
        ret = fn(*args,**kwargs)
        print(\'222\')
    return inner

def wrapper2(fn):
    def inner(*args,**kwargs):
        print(\'333\')
        ret = fn(*args,**kwargs)
        print(\'444\')
    return inner

def wrapper3(fn):
    def inner(*args,**kwargs):
        print(\'555\')
        ret = fn(*args,**kwargs)
        print(\'666\')
    return inner

@wrapper1
@wrapper2
@wrapper3
def eat():
    print(\'我想吃水果\')

eat()

　　

 　　执行顺序: 首先@wrapper1装饰起来. 然后获取到⼀个新函数是wrapper1中的inner. 然后执行@wrapper2.这个时候. wrapper2装饰的就是wrapper1中的inner了了. 所以. 执行顺序就像:第⼆层装饰器前(第⼀层装饰器前(目标)第一层装饰器后)第二层装饰器后. 程序从左到右执行起来. 就是我们看到的结果

 

 四.  使用wraps装饰函数保留函数的原名称

 

1.wraps

 

装饰器装饰的函数名字会变为inner,为了让装饰的函数名还原,我们用到了@wraps,这样打印出的函数名就还是原来的函数名了, 要引入functools模块

from functools import wraps # 可以改变一个函数的名字, 注释...
 
def wrapper(fn):
    @wraps(fn)  # 把inner的名字改变成原来的func
    def inner(*args, **kwargs):
        print("前")
        ret = fn(*args, **kwargs)
        print("后")
        return ret
 
    return inner
 
@wrapper # func = wrapper(func)
def func():
    print(\'哈哈哈\')
 
print(func.__name__) # func