day 11 - 2 裝飾器練習

1、編寫裝飾器，為多個函數加上認證的功能（用戶的賬號密碼來源於文件）要求登錄成功一次，後續的函數都無需再輸入用戶名和密碼

FLAG = False
def login(func):
    def inner(*args,**kwargs):
        global FLAG
        if FLAG:
            ret = func(*args,**kwargs)
            return ret
        else:
            #登錄程式
            username = input('請輸入用戶名：')
            password = input('請輸入密碼：')
            if username == 'ysg' and password == "275":
                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()

2、編寫裝飾器，為多個函數加上記錄調用功能，要求每次調用函數都將被調用的函數名稱寫入文件

def log(func):
    def inner(*arge,**kwargs):
        with open('E:/py/log/log.txt','a',encoding='utf-8') as f:
            f.write(func.__name__ + '\n')
        ret = func(*arge,**kwargs)
        print("")
        return ret
    return inner

@log
def shoplist_add():
    print("購買了一件商品")

@log
def shoplist_del():
    print("刪除了一件商品")

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

進階練習：
1.編寫下載網頁內容的函數，要求功能是：用戶傳入一個url，函數返回下載頁面的結果
2.為題目1編寫裝飾器，實現緩存網頁內容的功能：
具體：實現下載的頁面存放於文件中，如果文件內有值（文件大小不為0），就優先文件中讀取網頁內容，否則，就去下載，然後存到文件中

import os
from urllib.request import urlopen
def cache(func):
    def inner(*arge,**kwargs):
        if os.path.getsize('E:/py/log/url.txt'):
            with open ("E:/py/log/url.txt","rb") as f:
                return f.read()
        ret = func(*arge,**kwargs)
        with open('E:/py/log/url.txt','wb') as f:
            f.write(ret+b"*****")
        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)