Python装饰器实战：登录认证、日志记录与网页缓存

Python装饰器练习题概述

本文主要介绍Python装饰器的相关练习题及解答，通过详细的示例代码帮助读者深入理解装饰器的用法，对学习和工作具有参考价值。

实现用户登录状态保持的装饰器

第一题要求编写一个装饰器，为多个函数添加用户账号密码的认证功能。

具体需求是：只要登录成功一次，后续调用被装饰的函数时，无需再次输入用户名和密码。

解决方案中定义了一个全局变量FLAG=False，用于标识当前是否已登录。设置此步骤的目的是为了判断用户是否已经登录成功，从而避免重复登录操作。

def login(func):#为多个函数加上的认证功能
  def inner(*args,**kwargs):#加上装饰器
    global FLAG#给函数内的全局变量赋值
    if FLAG:
      ret = func(*args, **kwargs) # func就是被装饰的函数
      return ret
  #'''写登录程序'''
    else:
      username=input('亲输入你的名字：')
      password=input('亲输入你的密码：')
      if username=='lxx' and password=='123':
        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()

实现函数调用日志记录的装饰器

第二题要求编写另一个装饰器，为多个函数添加记录调用功能，确保每次调用函数时，都将该函数的名称追加写入到一个日志文件中。

def log(func):
  def inner(*args,**kwargs):
    with open('log','a',encoding='utf-8') as f:
      f.write(func.__name__)#每次调用函数都将被调用的函数名称写入文件
    ret=func(*args,**kwargs)
    return ret
  return inner

@log
def shoplist_add(): #两个函数有不同的功能
  print('增加一件物品')

@log
def shoplist_del():
  print('删除一件物品')

shoplist_add()
shoplist_del()

进阶：网页内容缓存与下载

第三题为进阶作业，包含以下三个子任务：

1. 编写一个下载网页内容的函数，接收用户传入的URL作为参数，并返回下载页面的结果。

2. 为上述下载函数编写一个装饰器，实现缓存网页内容的功能。

3. 具体实现逻辑：将下载的页面内容存储到本地文件中。如果该文件存在且大小不为零（即文件内有值），则优先从文件中读取网页内容；否则执行下载操作并将结果写入文件。

import os
from urllib.request import urlopen
def cache(func):
  def inner(*args,**kwargs):
    if os.path.getsize('log'):#查看文件大小 如果不为零为 判断文件中是否有内容
      with open('log','rb') as f:
        return f.read()
    ret=func(*args,**kwargs)#请求网页的函数
    with open('log','wb') as f: #wb不需要encoding='utf-8'
      f.write(b'****vfd**' + ret) #下载的页面存放在文件中 将请求的结果写入文件
    return ret
  return inner

@cache
def get(url):
  code=urlopen(url).read
  return code

ret=get('https://shimo.im/docs/VN0BLgAIBdMVSa4S/read')
print(ret)
ret=get('https://shimo.im/docs/VN0BLgAIBdMVSa4S/read')
print(ret)
ret=get('https://shimo.im/docs/VN0BLgAIBdMVSa4S/read')
print(ret)

以上就是本文的全部内容，希望对读者的学习有所帮助，欢迎支持本站。