python语法基础-函数-装饰器-长期维护

Posted 2021-03-11 技术改变命运Andy

###############    装饰器的初成和开放封闭原则    ##############

# 装饰器
# 装饰器非常重要，面试Python的公司必问，
# 原则：开放封闭原则
#

# 需求：计算程序执行的时间，
import time
def func():
    start = time.time()
    time.sleep(0.01)
    print(123)
    end = time.time()
    print(end-start)

func()

# 这种写法如果有100个函数，你都要加一遍这是不行的，太麻烦了，
# 第二版

def func():
    time.sleep(0.01)
    print(123)


def timer(f):
    start = time.time()
    f()
    end = time.time()
    print(end-start)

timer(func)  # 专门写了一个函数用来计算函数的执行时间，


# 闭包的写法

def func():
    time.sleep(0.01)
    print(123)
def timer(f):
    def inner():
        start = time.time()
        f()
        end = time.time()
        print(end-start)
    return inner

# 这就是一个闭包，因为是函数的嵌套，并且函数内部调用函数外部的参数，
# 不修改函数的调用方式，还想要在函数的前后添加功能，
# timer就是一个装饰器，对函数有装饰作用，
func = timer(func)  # inner
func()  # inner()


# 开放封闭原则，适用于对已经固定的额功能，和源码，这个原则是为了程序的稳定性，
# 开放就是对已经有的功能的拓展是开放的，
# 封闭就是对已经有的功能是不改动的，

# 所以使用装饰器是一个非常不错的方式在不改变程序内部的情况下，在程序的前后添加功能，

 

###############    最终的装饰器版本    ##############

# 最终的装饰器版本
# 解决两个问题，
# 1是被装饰函数有返回值
# 2是被装饰函数有参数

import time
def timer(f):
    def inner(*args,**kwargs):  # 这个参数就是为了能接收被装饰器函数的参数，
        start = time.time()
        ret = f(*args,**kwargs)
        end = time.time()
        print(end-start)
        return ret  # 这个地方要返回是因为被装饰函数可能会有返回值，
    return inner

@timer
def func(a):
    time.sleep(0.01)
    print(123)
    print(a)
    return 123
# func = timer(func)  # inner
ret = func(1)  # inner()，这个地方已经是调用的inner了，所以一定要在inner中把被装饰函数的返回值返回回来，
print(ret)


# def outer():
#     def inner():
#         return 123
#     inner()
#
# outer()  # 这个调用是不会有返回值的，因为你只调用了没有返回值，

# 所有有一个装饰器的固定模式
from functools import wraps
def outer(func):  # 或者叫做wrapper，这就是装饰器的意思，一般会起一个和业务相关的名字，
    @wraps(func)  # 加上这一句，再去打印函数的name的时候，就不会是inner了，
    def inner(*args,**kwargs):  # 一般就是一个inner
        # 在被装饰函数之前写的东西
        ret = func(*args,**kwargs)
        # 在被装饰函数之后写的东西
        return ret
    return inner


@outer  # 这一句就是test=outer(test)
def test():
    """
    函数测试，
    :return:
    """
    print(123)


test()  # 这个地方就是inner()了，
print(test.__name__)  # 这个被装饰器装饰了之后会有问题，inner返回的是inner，而不是test
print(test.__doc__)

 

###############    装饰器进阶--带参数的装饰器    ##############

"""

装饰器进阶
1，带参数的装饰器

场景：就是我写了一个计算代码执行时间的装饰器，然后给每一个函数加上了装饰器，
如果我想要去掉装饰器的时候，就需要一个一个的去掉了，我想要用的时候还需要再一个一个的加上，很麻烦，有什么好的办法？

解决方法：就是给装饰器加上一个参数，然后通过控制参数来判断是否开启装饰器，
参数就叫做flag，
这种代码实现就是使用三层装饰器，这样只需要控制flag就可以控制装饰器是否开启了，


"""

import time

flag = False
def timmer_out(flag):
    def timmer(func):
        def inner(*args,**kwargs):
            if flag:
                start = time.time()
                ret = func(*args,**kwargs)
                end = time.time()
                print(end-start)
                return ret
            else:
                ret = func(*args, **kwargs)
                return ret
        return inner
    return timmer

@timmer_out(flag)
def test1():
    time.sleep(1)
    print(123)

@timmer_out(flag)
def test2():
    time.sleep(1)
    print(123)

# @timmer：这一句还好理解就是func=timmer(func)
# @timmer_out(flag)：为了传递参数，我们可以再次在装饰器外部加一层，用来传递参数，
# 这一句要分开看了，第一步是timmer_out(flag)，然后返回了timmer，第二步才是@timmer，执行装饰器，

test1()
test2()

 

###############   装饰器进阶---多个装饰器装饰一个函数    ##############

"""
装饰器进阶：
2，多个装饰器装饰一个函数，


"""
def wrapper1(func):
    def inner():
        print(‘wrapper1 ,before func‘)
        func()
        print(‘wrapper1 ,after func‘)
    return inner

def wrapper2(func):
    def inner():
        print(‘wrapper2 ,before func‘)
        func()
        print(‘wrapper2 ,after func‘)
    return inner

@wrapper2
@wrapper1
def f():
    print(‘in f‘)

f()
"""
执行结果：
wrapper2 ,before func
wrapper1 ,before func
in f
wrapper1 ,after func
wrapper2 ,after func
说明了几个问题：
1，多个装饰器是把函数之前的装饰代码都执行之后，然后才是执行被装饰函数，最后是函数之后的装饰代码
2，函数之前的执行顺序是谁在上面执行谁，这是从上到下，函数之后的执行顺序是谁在下面执行谁，这是从下往上，
就像是俄罗斯套娃，进的时候从外往里，出的时候从里往外，
"""