### Python 登录验证装饰器的设计与测试 为了实现一个功能完整的登录验证装饰器并测试其对增删改操作的生效情况，可以按照以下方式设计： #### 1. 定义装饰器逻辑 装饰器的核心在于拦截函数调用前的身份验证过程。如果验证失败，则阻止后续的操作；如果成功，则允许继续执行。 以下是基于 `functools` 的装饰器定义[^2]： ```python import functools def login_required(func): @functools.wraps(func) def wrapper(username, password, *args, **kwargs): # 验证用户名和密码 if username == "root" and password == "123": print("身份验证成功") return func(*args, **kwargs) # 继续执行原函数 else: print("身份验证失败，无法执行操作") return None # 或者抛出异常 return wrapper ``` #### 2. 实现增删改功能 接下来定义一些简单的增删改操作函数，并应用该装饰器进行保护。 ##### (a) 添加数据 ```python @login_required def add_data(data): print(f"新增数据: {data}") return True ``` ##### (b) 删除数据 ```python @login_required def delete_data(key): print(f"删除键为 {key} 的数据") return True ``` ##### (c) 修改数据 ```python @login_required def update_data(key, value): print(f"更新键 {key} 的值为 {value}") return True ``` #### 3. 测试装饰器效果 通过不同的用户名和密码组合来测试装饰器的行为。 ```python if __name__ == "__main__": # 正确的用户名和密码 add_data("root", "123", data="example") # 输出：身份验证成功 新增数据: example # 错误的用户名或密码 delete_data("admin", "wrong_password", key=123) # 输出：身份验证失败，无法执行操作 # 更新操作测试 update_data("root", "123", key="status", value="active") # 输出：身份验证成功 更新键 status 的值为 active ``` 以上代码展示了如何构建一个带有身份验证的装饰器以及它的实际应用场景。当用户提供正确的用户名 (`root`) 和密码 (`123`) 时，装饰器会放行目标函数的执行；否则，它将拒绝访问[^1]。 --- ####