python基础学习-day17==课后作业练习（无参装饰器）

# 必做题：
# 1.编写函数，（函数执行的时间用time.sleep(n)模拟）

import time

def func(x, y):
    num = x + y
    time.sleep(3)
    print(num)
    return num

# 2.编写装饰器，为函数加上统计时间的功能

import time  # 源码


# 统计时间装饰器
def timer(func):  # func ---> 被装饰对象
    def inner(*args, **kwargs):  # 被装饰对象需要接收的参数

        # 被装饰对象执行之前的时间
        start_time = time.time()
        # *args, **kwargs将接收过来的参数，传给调用的被装饰对象，会得到一个返回值
        res = func(*args, **kwargs)  # 调用被装饰对象， 这行代码就是被装饰对象的执行全过程
        # 被装饰对象执行结束后的时间
        end_time = time.time()

        # 打印被装饰对象的执行时间
        # func.__name__ ---> 可以获取函数的名称
        print(f'当前被装饰对象[{func.__name__}] 执行时间为: [{end_time - start_time}]')
        return res

    return inner


@timer  # timer = inner = timer(func)
def func_demo():
    time.sleep(3)  # 单位: 秒

# 使用装饰器的目的: 1、为被装饰对象添加新的功能 2、前提: 不修改被装饰对象的源代码, 不修改调用方式
# inner = timer(func)
# func = inner
# func()  # --> inner()
# func_demo()

import os

def timer(func):
    def wrapper(*args, **kwargs):
        start = time.time()
        res = func(*args, **kwargs)
        end = time.time()
        print(end - start)
        return res

    return wrapper

num = timer(num)
num(1,2)

# 3.编写装饰器，为函数加上认证的功能

# 全局变量: 用于记录是否有用户登录
login_user = None


# 登录功能
def login():
    username = input('请输入用户名: ').strip()
    password = input('请输入密码: ').strip()
    if username == 'tank' and password == '123':
        print('登录成功')
        # 登录成功后，给全局变量赋值，记录当前用户已经登录
        global login_user
        login_user = username

    else:
        print('登录失败')


# 登录认证装饰器
def login_auth(func):
    def inner(*args, **kwargs):
        # 被装饰对象执行前，为其添加新的功能
        # 在此处做登录认证
        if login_user:
            # 被装饰对象执行  func ---> play, shopping
            res = func(*args, **kwargs)
            return res

            # 被装饰对象执行后，为其添加新的功能
        else:
            print('未有用户登录，无法使用好玩的功能!')
            login()

    return inner

# play 与 shopping 功能使用前 必须先登录 否则，不允许使用，让其执行登录功能
@login_auth
def play():
    # 不使用装饰器
    # if not login_user:
    #     login()
    print('开始play了啊....')
    pass


@login_auth
def shopping():
    pass
# login()
# play()
# shopping()

import os

def login(func):
    def wrapper(*args,**kwargs):
        user_name = input('请输入用户名:')
        user_psd = input('请输入密码：')
        with open('name.txt','rt',encoding='utf-8')as f:
            for line in f:
                username,password = line.strip().split(':')
            if username==user_name and user_psd==password:
                print('登陆成功!')
            #   执行之前的num函数
                res = func(*args,**kwargs)
                return res
            else:
                print('登录失败!')
    return wrapper

num =login(num)
num(1,2)

# 4.编写装饰器，为多个函数加上认证的功能（用户的账号密码来源于文件），要求登录成功一次，后续的函数都无需再输入用户名和密码

# 注意：从文件中读出字符串形式的字典，可以用eval('{"name":"egon","password":"123"}')转成字典格式

# 1、打开db.txt文件，读取用户数据，转成字典类型
with open('db.txt', 'r', encoding='utf-8') as f:
    # 直接读取文件中的数据，得到的数据类型是: str
    user_data = f.read()
    print(user_data, type(user_data))
    '''
    {"name": "tank", "pwd": "123"}
    '''
    # eval(str) ---> 从str字符串中，检测python代码的语法，
    # 如果里面有{key: value}，会将该数据生成对应的类型的内存地址
    user_dic = eval(user_data)
    print(user_dic, type(user_dic))

user_dic = user_dic
# print(user_dic)
# print(type(user_dic))
login_info = None

def login():
    username = input('请输入用户名: ').strip()
    if username in user_dic.get('name'):
        password = input('请输入密码: ').strip()
        if password == user_dic.get('pwd'):
            print('login successful')
            global login_info
            login_info = {username: password}

        else:
            print('密码错误')
    else:
        print('用户不存在')


def login_auth(func):
    def inner(*args, **kwargs):
        if login_info:
            res = func(*args, **kwargs)
            return res
        else:
            login()
    return inner


@login_auth
def withdraw():
    pass
@login_auth
def shopping():
    pass

# 5.编写装饰器，为多个函数加上认证功能，要求登录成功一次，在超时时间内无需重复登录，超过了超时时间，则必须重新登录

import time

# 全局变量，用于计算函数登录后的执行总时长
sum_time = 0  # ---》 func1 --> 3 ---> 6

def time_wait(func):
    # 登录成功后，可以使用的功能时长，单位 秒
    wait_time = 6

    def inner(*args, **kwargs):
        # 执行前时间
        start_time = time.time()
        res = func(*args, **kwargs)
        # 执行后时间
        end_time = time.time()
        global sum_time
        # 第一个函数进来后，func1 执行时长为3, func2 执行时长为3
        sum_time += (end_time - start_time)
        print(f'登录后执行的总时长: {sum_time}')
        # func1执行的时长 >= wait_time
        if sum_time >= wait_time:
            # 需要重新去登录
            login()
            sum_time = 0
        else:
            return res

    return inner


@time_wait
def func1():
    time.sleep(3)

@time_wait
def func2():
    time.sleep(3)

# func1()
# func2()
#
# # 此时总时长为6秒
# func1()

六、# 思考题（选做），叠加多个装饰器，加载顺序与运行顺序
# @deco1 # index=deco1(deco2.wrapper的内存地址)
# @deco2 # deco2.wrapper的内存地址=deco2(deco3.wrapper的内存地址)
# @deco3 # deco3.wrapper的内存地址=deco3(index)
# def index():
#     pass