装饰器的简易版本、解决装饰器参数的问题、解决装饰器解决返回值问题、装饰器模板、装饰器的语法糖(@就是标志符号)、装饰器的修复技术、有参装饰器

一、装饰器的简易版本

'''
	核心思想：
	在不改变被装饰对象内部代码和原有调用方式的基础之上在添加额外的功能
# 给被装饰对象添加额外的功能
'''

# 代码体现1：
import time


def index():
    time.sleep(3)
    print('from index')


def home():
    time.sleep(2)
    print('from home')

def outer(func):
    def get_time():
        # 在函数执行之前打一个时间点
        start_time = time.time()
        func()
        # 在函数执行之后打一个时间点
        end_time = time.time()
        print('函数执行了%s秒' % (end_time - start_time))
    return get_time


home = outer(home)
home()

# 利用闭包形式来传参
# res = outer(index)
# res()

二、解决装饰器参数的问题

import time
def index(name, username):
    time.sleep(2)
    print('等待2秒')

def home(name):
    time.sleep(3)
    print('等待3秒')


def outer(func):
    # func = index
    def get_time(*args, **kwargs):
        start_time = time.time()
        func(*args, **kwargs)
        print('函数执行了%s秒' % (time.time() - start_time))

    return get_time


home = outer(home)
home("lin")
# index = outer(index)
# index(name='lin', username='changlin')

三、解决装饰器解决返回值问题

import time


def home(name):
    time.sleep(2)
    print('等待2秒')


def index():
    time.sleep(3)
    print('等待3秒')
    return '3秒后再执行'


def outer(func):
    def inner(*args, **kwargs):
        start_time = time.time()
        ret_func = func(*args, **kwargs)
        print('函数执行了%s秒' % (time.time() - start_time))

        return ret_func

    return inner


home = outer(home)
home('kevin')

# index = outer(index)
# index()

四、课堂练习(登录认证功能)

def home(name):
    print('执行home')

def index():
    print('执行func')

# 要求：登录成功后才能执行index或者home函数

def login_auth(func):
    def auth(*args, **kwargs):
        username = input('username:').strip()
        password = input('password:').strip()

        if username == 'kevin' and password == '123':
            print('登录成功')
            ret_func = func(*args, **kwargs)
            return ret_func
        else:
            print('用户名或密码错误')
    return auth


# home = login_auth(home)
# home('我来啦')
index = login_auth(index)
index()

五、装饰器模板

def outer(func):
	inner(*args, **kwargs):
		print('这是执行函数前的功能')
		ret_func = func(*args, **wkargs)
		print('这是执行函数后的功能')
		return ret_func
	return inner


def index():
	print('执行index')


index = outer(index)
index()

六、装饰器的语法糖(@就是标志符号)

'''
    语法糖的书写规范：
        紧贴着被装饰对象的上方写
    语法糖的内部原理：
        它会自动把被装饰对象的名字当成参数传递给装饰器函数调用
'''
def outer(func):  # index = outer(index)
    def inner(*args, **kwargs):
        print('执行前添加的功能')
        res = func(*args, **kwargs)
        print('执行后添加的功能')
        return res

    return inner


@outer  # outer(index) 此时index就是outer函数的的返回值(内部函数的函数名即inner)
def index():
    print('from index')


@outer  # outer(home)  此时home就是outer函数的的返回值(内部函数的函数名即inner)
def home():
    print('from home')


# index = outer(index)
index()  # index()相当于inner()

七、双层语法糖

import time


def login_auth(func):
    def auth(*args, **kwargs):
        username = input('username:').strip()
        password = input('password:').strip()

        if username == 'kevin' and password == '123':
            print('登录成功')
            res = func(*args, **kwargs)
            return res
        else:
            print('用户名或密码错误')
    return auth


def get_time(func):
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        print('函数执行了%s秒' % (time.time() - start_time))
        return res
    return inner


@login_auth
@get_time
def index():
    time.sleep(3)
    print('等等3秒')


index()

八、双层语法糖

# 判断七句print执行顺序
def outter1(func1):
    print('加载了outter1')
    def wrapper1(*args, **kwargs):
        print('执行了wrapper1')
        res1 = func1(*args, **kwargs)
        return res1
    return wrapper1

def outter2(func2):
    print('加载了outter2')
    def wrapper2(*args, **kwargs):
        print('执行了wrapper2')
        res2 = func2(*args, **kwargs)
        return res2
    return wrapper2

def outter3(func3):
    print('加载了outter3')
    def wrapper3(*args, **kwargs):
        print('执行了wrapper3')
        res3 = func3(*args, **kwargs)
        return res3
    return wrapper3


@outter1
@outter2
@outter3
def index():
    print('from index')

九、装饰器的修复技术

import time
from functools import wraps


def outer(func):
    @wraps(func)
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        print('函数执行%s秒' % (time.time() - start_time))
        return res

    return inner


@outer
def index():
    time.sleep(3)
    print('修复好了')


index()

十、有参装饰器

# 有参装饰器就是带有参数的装饰器

def outer(type):
    # type = 1
    def login_auth(func):
        # func = index
        def auth(*args, **kwargs):
            # 写登录功能
            # 1.
            username = input('username:').strip()
            password = input('password:').strip()
            # 2.验证用户名和密码是否正确
            # with open('a.txt', 'r', encoding='utf8')
            '''根据不同的参数用户名和密码来自不同的位置'''
            if type == 'file':
                print('来自文件')
            elif type == 'mysql':
                print('来自mysql')
            elif type == 'oracle':
                print('来自oracle')
            return auth      
        return login_auth


@outer('file')
def index():
    print('file')
   
   
index() 

    
@outer('mysql')
def mysql():
    print('mysql')
    
    
mysql()

    
@outer('oracle')
def oracle():
    print('oracle')
    

oracle()