装饰器的调用

• 装饰器的简易版本
• 解决参数问题

装饰器的简易版本

# 给函数添加统计执行时间的功能
def outer(func): 
  # func = index
  def get_time():
    start_time = time.time()
    func()
    end_time = time.time()
    print('函数运行时间%s' % (end_time - start_time)
retrn get_time
index = outer(index)
# 左侧的变量名index指函数名geit_time
index()

解决参数问题

import time


def rigth():
    time.sleep(3)
    print('别搞了,再卷会')

def index(func):
    def data(*args, **kwargs):
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        print('jj卷了%s' % (start_time - end_time))
    return data
rigth = index(rigth)
rigth()

解决返回值问题

import time


def rigth():
    time.sleep(3)
    print('别搞了,卷一会')

    return 'hhhh'

def index(func):
    def data(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        end_time = time.time()
        print('jj卷了%s' % (start_time - end_time))
        return res
    return data
index = index(rigth)
res = index()
print(res)

认证装饰器

# 今天只需要掌握简单版本(每个都校验用户名和密码)
# 记录用户状态的版本多花时间去练
import time
def index():
    time.sleep(1)
    print('百万大奖等你来拿 赶快加入我们吧!!!')
def home():
    time.sleep(1)
    print('学学学 一天到晚就是学 卷死你们这些家伙')
def register():
    time.sleep(1)
    print('注册功能')
# 给index函数添加认证功能
"""
在调用index之前需要用户输入用户名和密码
    正确才可以调用
    错误直接拒绝
"""
# 定义一个用于记录用户是否登录的数据
is_login = {'is_login':False}


def login_auth(func):
    def auth(*args,**kwargs):
        # 1.1 先判断用户是否已经登录
        if is_login.get('is_login'):
            # 3.正常执行函数index
            res = func(*args, **kwargs)
            return res
        # 1.先获取用户的用户名和密码
        username = input('username>>>:').strip()
        password = input('password>>>:').strip()
        # 2.校验用户名和密码是否正确
        if username == 'jason' and password == '123':
            # 3.正常执行函数index
            res = func(*args,**kwargs)
            # 4.将记录用户登录状态的数据修改
            is_login['is_login'] = True
            return res
        else:
            print('用户名或密码错误 无法执行函数')
    return auth
index = login_auth(index)
index()
home = login_auth(home)
home()
register = login_auth(register)
register()

装饰器固定模版

def outer(func):
    def inner(*args, **kwargs):
        print('执行函数之前可以添加的额外功能')
        res = func(*args, **kwargs)  # 执行被装饰的函数
        print('执行函数之后可以添加的额外功能')
        return res  # 将被装饰函数执行之后的返回值返回
    return inner

装饰器语法糖

装饰器语法糖是写规范
    语法糖必须紧贴在被装饰对象的上方
装饰器语法糖内部原理
    会自动将下面紧贴着的被装饰对象名字当做参数传给装饰器函数调用

import time



def index(func):
    def data(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        end_time = time.time()
        print('jb动了%s' % (start_time - end_time))
        return res
    return data
# rigth = index(rigth)
# res = index()


@index
def rigth():
    time.sleep(3)
    print('别搞了,几把')
    return 'hhhh'
rigth()

装饰器修复技术

from functools import wraps
def outer(func):
    @wraps(func)  # 修复技术就是为了让被装饰对象更加不容易被察觉装饰了
    def inner(*args, **kwargs):
        print('执行函数之前可以添加的额外功能')
        res = func(*args, **kwargs)  # 执行被装饰的函数
        print('执行函数之后可以添加的额外功能')
        return res  # 将被装饰函数执行之后的返回值返回
    return inner


@outer  # index = outer(index)
def index():
    print('from index')
print(index)
help(index)

def home():
    """这是一个home函数"""
    print('from home')
# help(index)
# help(home)
# print(index)
# help(len)

溜了溜了