python 用裝飾器寫登錄

# 1.編寫裝飾器，為多個函數加上認證的功能（用戶的賬號密碼來源於文件），
# 要求登錄成功一次，后續的函數都無需再輸入用戶名和密碼
# FLAG = False
# def login(func):
#     def inner(*args,**kwargs):
#         global FLAG
#         '''登錄程序'''
#         if FLAG:
#             ret = func(*args, **kwargs)  # func是被裝飾的函數
#             return ret
#         else:
#             username = input('username : ')
#             password = input('password : ')
#             if username == 'boss_gold' and password == '22222':
#                 FLAG = True
#                 ret = func(*args,**kwargs)      #func是被裝飾的函數
#                 return ret
#             else:
#                 print('登錄失敗')
#     return inner
#
# @login
# def shoplist_add():
#     print('增加一件物品')
#
# @login
# def shoplist_del():
#     print('刪除一件物品')
#
# shoplist_add()
# shoplist_del()

# 2.編寫裝飾器，為多個函數加上記錄調用功能，要求每次調用函數都將被調用的函數名稱寫入文件
# def log(func):
#     def inner(*args,**kwargs):
#         with open('log','a',encoding='utf-8') as f:
#             f.write(func.__name__+'\n')
#         ret = func(*args,**kwargs)
#         return ret
#     return inner
#
# @log
# def shoplist_add():
#     print('增加一件物品')
#
# @log
# def shoplist_del():
#     print('刪除一件物品')

# shoplist_add()
# shoplist_del()
# shoplist_del()
# shoplist_del()
# shoplist_del()
# shoplist_del()

# 進階作業(選做)：
# 1.編寫下載網頁內容的函數，要求功能是：用戶傳入一個url，函數返回下載頁面的結果
# 2.為題目1編寫裝飾器，實現緩存網頁內容的功能：
# 具體：實現下載的頁面存放於文件中，如果文件內有值（文件大小不為0），就優先從文件中讀取網頁內容，否則，就去下載，然后存到文件中
import os
from urllib.request import urlopen
def cache(func):
    def inner(*args,**kwargs):
        if os.path.getsize('web_cache'):
            with open('web_cache','rb') as f:
                return f.read()
        ret = func(*args,**kwargs)  #get()
        with open('web_cache','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)