7.装饰器的一些需求

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

FLAG = False
def login(func):
    def inner(*args,**kwargs):
        global FLAG
        if FLAG:
           ret = func(*args,**kwargs)
           return ret
        else:
           use_name = input('use_name:')
           password = input("password:")
           if use_name == 'yasuo' and password == 'kuaile':
              FLAG = True
              ret = func(*args, **kwargs)
              return ret

           else:
              print('登录失败')

    return inner
@login
def shoplist_add():
    print('添加一个物品')
@login
def shoplist_del():
    print("删除一个物品")
shoplist_add()
shoplist_del()

#编写装饰器，为多个函数加上记录调用功能，要求每次调用函数都将被调用的函数名称写入文件。

def wrapper(func):
    def inner(*args,**kwargs):
        with open('和好','a',encoding='utf-8') as f:
            f.write(func.__name__+'\n')
        ret = func(*args,**kwargs)
        return ret
    return inner
@wrapper
def qq():
    print('世界')
@wrapper
def mm():
    print('青春')
qq()
mm()

#进阶作业
#编写下载网页内容的函数，要求功能是：用户传一个url，函数返回下载页面的结果
#为题目1编写装饰器，实现缓存页面内容的功能：
#具体：实现下载的页面存放于文件中，如果文件内有值(文件大小不为0)，
# 就优先从文件中读取网页内容，否则就去下载

import os
from urllib.request import urlopen
def cache(func):
    def inner(*args,**kwargs):
       if os.path.getsize('和好'):
         with open('和好','rb')as f:
            return f.read()
       ret = func(*args,**kwargs)
       with open('和好','wb') as f:
            f.write(b'*****'+ret)
       return ret
    return inner

@cache
def get(url):
    code = urlopen(url).read()
    return code
ret = get('http://www.baidu.com')
print(ret)
ret = get('http://www.baidu.com')
print(ret)
ret = get('http://www.baidu.com')
print(ret)