python 14 装饰器

目录

• 装饰器
  • 1. 开放封闭原则

装饰器

1. 开放封闭原则

扩展是开放的，增加新的功能；修改源码（修改已经实现的功能）是封闭的。
在不改变源码及调用方式的前提下额外增加新的功能。

# 版一：
import time
start_time = time.time()  #起始时间
def func():
    time.sleep(2)   #睡眠，模拟网络延迟
    print("我要飞")
func()
print(time.time()- start_time) # 打印执行世界

# 版二：
import time
def times(s):
    start_time = time.time()
    s()
    print(time.time()- start_time)
def foo():
    time.sleep(2)
    print("我要飞")
times(foo)    #改变了调用方式

# 版三（初版装饰器）：
import time
def times(s):
    def inner():
        start_time = time.time()
        s()
        print(time.time()- start_time)
    return inner
def func():
    time.sleep(1)
    print("我也要飞")
func = times(func)    # 不需改调用方式
func()

# 版四（升级）：
def wrapper(f):
    def inner(a):   #加入参数
        f(a)
    return inner    # 切记不加括号

def func(a):        # 加参数
    print(f"这是{a}的函数")
func = wrapper(func)
func("alex")

# 升级 万能传参：
import time
def wrapper(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        f(*args,**kwargs)
        print(time.time() - start_time)
    return inner    # 切记不加括号

def func(*args,**kwargs):        # 加形参
    time.sleep(1)
    print(f"这是{args,kwargs}的函数")
func = wrapper(func)
func("alex",1,2,a = 3,b = 4)

def foo(*args,**kwargs):        # 加形参
    time.sleep(2)
    print(f"这是{args,kwargs}的函数")
foo = wrapper(foo)
foo("meet",1,2,a = 3,b = 4)

#语法糖   放在被装饰函数的上方
#替代func = wrapper(func) 和 foo = wrapper(foo)

import time
def wrapper(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        f(*args,**kwargs)
        print(time.time() - start_time)
    return inner    # 不加括号
@wrapper   			#语法糖
def func(*args,**kwargs):        # 加形参
    time.sleep(1)
    print(f"这是{args,kwargs}的函数")
# func = wrapper(func)
@wrapper			#语法糖
def foo(*args,**kwargs):        # 加形参
    time.sleep(2)
    print(f"这是{args,kwargs}的函数")
# foo = wrapper(foo)

func("alex",1,2,a = 3,b = 4)
foo("meet",5,6,a = 7,b = 8)

# 标准版：
import time
def wrapper(f):
    def inner(*args,**kwargs):
        start_time = time.time()
        ret = f(*args,**kwargs)
        print(time.time() - start_time)
        return ret  #增加返回值
    return inner    # 切记不加括号
@wrapper  			#语法糖
def func(*args,**kwargs):        # 加形参
    time.sleep(1)
    print(f"这是{args,kwargs}的函数")
    return "这是alex的返回"
print(func("alex",1,2,a = 3,b = 4))  # 会打印"这是alex的返回"

#语法糖  放在被装饰函数的上方

#标准版的装饰器：
def wrapper(func):
    def inner(*args,**kwargs):
        '''执行被装饰函数之前的操作'''
        ret = func(*args,**kwargs)
        '''执行被装饰函数之后的操作'''
        return ret  # 返回inner(),也就是func()打印
    return inner
@wrapper    # 相当于 func = wrapper(func)
def func(*args,**kwargs)
	print(args,kwargs)
	return "返回的内容"     # 返回ret
print(func())  	# 能够打印返回值，也可以传参