python装饰器

- 装饰器本质

装饰器本质就是一个修饰其他函数的函数，为其他函数添加附加功能

- 装饰器要把握的2个原则

1.不修改被修饰函数的源代码
2.不修改被修饰函数的调用方式

- 装饰器的实现

装饰器 = 高阶函数+函数嵌套+闭包

满足下面任何一个条件都是高阶函数：
（1）函数接收的参数是一盒函数名
（2）函数的返回值是一个函数名

函数嵌套：父函数里面还有子函数
闭包：作用域的一种体现，一层套一层（一层一个函数），给最外层传一个参数，可以一直渗透到最里层

• 只用高阶函数例子：给函数加上统计运行时间的功能

# 方式一：修改了调用方式
import time
def foo():
    print("你好")

def test(func):
    start_time = time.time()
    print(func)
    func()
    stop_time = time.time()
    print("函数%s的运行时间为%s;但是函数调用方式从foo()变成了test(foo)" % (func.__name__, stop_time-start_time))

test(foo)

# 方式二：虽然没有修改调用方式，但是foo1函数被调用了两次func
import time
def foo1():
    print("你好")

def test1(func):
    start_time = time.time()
    print(func)
    func()
    stop_time = time.time()
    print("函数%s的运行时间为%s;但是函数调用方式从foo()变成了test(foo)" % (func.__name__, stop_time-start_time))
    return func

foo1 = test1(foo1)
foo1()

• 高阶函数+函数嵌套实现基本的装饰器架子

# 装饰器架子
def func():
	pass

def timmer(func):
	def wrapper():
		# 函数执行前操作
		func()
		# 函数执行后操作
	return wrapper

func = timmer(func)
func()

• 高阶函数+函数嵌套实现基本的装饰器：给函数加上统计运行时间的功能

import time
def func():
	print("in func")
	time.sleep(2)

def timmer(func):
	def wrapper():
		start_time = time.time()
		func()
		stop_time = time.time()
		print("函数%s的运行时间为%s;但是函数调用方式从foo()变成了test(foo)" % (func.__name__, stop_time-start_time))
	return wrapper
	
func = timmer(func) # 这一句等价于在func定义上面加上@time
func()		

- 装饰器优化：带参数和返回值的装饰器

'''
func = timmer(func)
func(1,2,3)
func=timmer(func)的返回值是wrapper函数，给func加参数等价于给wrapper加参数
所以func(*args, **kwargs) => wrapper(*args, **kwargs)

func的返回值要在wrapper中接收一下，再return回去

'''
def timmer(func):
    # 函数嵌套
    def wrapper(*args, **kwargs):
        start_time = time.time()
        print(func)
        res = func(*args, **kwargs) # 参数+返回值
        stop_time = time.time()
        print("%s函数运行时间为%s" % (func.__name__, stop_time-start_time))
        return res
    return wrapper # 高阶函数

def func(*args, **kwargs):
    time.sleep(1)
    print(args, kwargs)
    print("in func")


func = timmer(func)
res = func(1,2,3)
print(res)

- 修改成@的方式

def timmer(func):
    # 函数嵌套
    def wrapper(*args, **kwargs):
        start_time = time.time()

        res = func(*args, **kwargs)
        stop_time = time.time()
        duration = stop_time-start_time
        print("%s函数运行时间为%s" % (func.__name__, duration))
        return res
    return wrapper # 高阶函数

@timmer # 等价于 func = timmer(func)
def func(*args, **kwargs):
    time.sleep(1)
    print("in func")
    print(args, kwargs)
    return (args, kwargs)

print(func(1, 2, 3))

- 带参数的装饰器–实现认证类型选择

def auth(auth_type="file_auth"):
	def dec_auth(func):
	    def wrapper(*args, **kwargs):
	        # 验证功能
	        print("你输入的认证类型是%s" % auth_type)
	        username = input("用户名：").strip()
	        passwd = input("密码：").strip()
	        if username == "root" and passwd == "123":
	            res = func(*args, **kwargs)
	            return res
	        else:
	            print("用户名密码错误")
	    return wrapper
	return dec_auth
    
# 下面的过程分以下步骤
# （1） auth(auth_type="db_auth")=>dec_auth
# （2）@auth(auth_type="db_auth")  ==》 car = auth(auth_type="db_auth")=dec_auth(car)
@auth(auth_type="db_auth") 
def car(name):
    print("欢迎来到%s的购物车"%name)

car("ybp")