装饰器的练习题

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

flage=0#未登录状态设置为0
def wapper(fun):
    def inner(*args,**kwargs):
        global flage
        '''登录程序'''

        if flage==1:
            ret = fun(*args, **kwargs)#fun是被装饰的函数
            return ret
        else:
            username=input('请输入用户名')
            password=input('请输入密码')
            if username=='刘德华' and password=='123456':
                flage=1#登录成功设置为1用global
                ret=fun(*args,**kwargs)
                return ret
            else:
                print('登录失败')
    return inner


@wapper#等同于 shopp_add=wapper(shopp_add)
def shopp_add():
    print('增加一个物品')
@wapper#等同于 shopp_del=wapper(shopp_del)
def shopp_del():
    print('删除一个物品')

shopp_add()
shopp_del()

# 2.编写装饰器，为多个函数加上记录调用功能，要求每次调用函数都将被调用的函数名称写入文件
def log(fun):
    def inner(*args,**kwargs):
        res=fun(*args,**kwargs)
        with open('log.txt','a') as f: #a代表给文件追加内容
            f.write(fun.__name__+'\n')
        return res
    return inner

@log
def shoplist_add():
    print('增加一个物品')
@log
def shoplist_del():
    print('删除一个物品')

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

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


import os
from urllib.request import urlopen
def chae(fun):
    def inner(*args,**kwars):
        if os.path.getsize('chace'):#判断文件里面是否有内容
            with open('chace','rb') as f:
                return f.read()#如果文件里面有内容就返回并且结束
        ret=fun(*args,**kwars)
        with open('chace','wb') as f:#因为是bytes类型所以是wb模式
            f.write(b'******'+ret)#将请求的结果写到文件里面
        return ret
    return inner


@chae
def get(url):
    ret=urlopen(url).read()
    return ret

s=get('http://www.baidu.com')
print(s)
s=get('http://www.baidu.com')
print(s)
s=get('http://www.baidu.com')
print(s)

import functools
def ss(text='call'):
        def test(f):
            @functools.wraps(f)
            def testF(*args,**kwargs):
                if text=='call':
                    print('begin %s %s %s'%(text,fun1.__name__,'使用默认的的text'))
                    f(*args,**kwargs)
                    print('end call')
                else:
                    print('begin %s %s' % (text, fun1.__name__))
                    f(*args,**kwargs)
                    print('end call')
            return testF
        return test



@ss('excute')
def fun(x,y):
    print('这是一个被装饰的函数')
    return x,y

@ss()
def fun1(x):
    print('这是一个被装饰的函数')
    return x


fun(1,2)
print('-------------------------')
fun1(4)