如何在 Python 中为多个密钥设置并管理独立的过期时间-Python教程-PHP中文网

如何在 Python 中为多个密钥设置并管理独立的过期时间

本文介绍如何使用字典持久化存储密钥及其对应过期时间，实现“首次验证即启动倒计时”的精确过期控制，支持多密钥独立管理，避免重复生成导致时间重置。

本文介绍如何使用字典持久化存储密钥及其对应过期时间，实现“首次验证即启动倒计时”的精确过期控制，支持多密钥独立管理，避免重复生成导致时间重置。

在实际应用中（如软件许可、API 许可证或临时访问凭证），密钥不应每次输入都重新生成过期时间，而应在首次成功验证时绑定唯一过期时刻，并持久记录该状态。原始代码的问题在于：generate_key() 每次调用都生成新时间，且未保存状态；if expiration_time == timedelta: 逻辑错误（类型比较无意义）；验证流程未区分“密钥不存在”“已过期”“有效”三种状态。

正确方案是使用字典 key_expiration_map 作为轻量级内存数据库，以密钥为键、datetime 对象为值，实现状态持久化。以下为优化后的完整实现：

from datetime import datetime, timedelta

# 全局存储：密钥 → 过期时间（datetime对象）
key_expiration_map = {}


def generate_key() -> tuple[str, datetime]:
    """生成固定密钥及1分钟后过期时间（示例用'4255'，生产中应动态生成）"""
    key = "4255"
    expiration_time = datetime.now() + timedelta(minutes=1)
    return key, expiration_time


def is_expired(expiration_time: datetime) -> bool:
    """判断指定过期时间是否已到达"""
    return datetime.now() > expiration_time


def validate_key(user_input: str) -> bool:
    """
    验证用户输入的密钥：
    - 若密钥未注册 → 返回 False 并提示“Invalid Key”
    - 若已注册但过期 → 返回 False 并提示“Key Expired”
    - 若有效 → 打印成功信息与剩余有效期，返回 True
    """
    if user_input not in key_expiration_map:
        print("Invalid Key")
        return False

    expiration_time = key_expiration_map[user_input]
    if is_expired(expiration_time):
        print("Key Expired")
        return False
    else:
        # 可选：显示剩余秒数增强用户体验
        remaining = expiration_time - datetime.now()
        print(f"Successfully Loaded")
        print(f"Expiration: {expiration_time.strftime('%Y-%m-%d %H:%M:%S')} "
              f"(~{int(remaining.total_seconds())}s left)")
        return True


# 初始化：生成密钥并写入映射表（仅执行一次）
correct_key, exp_time = generate_key()
key_expiration_map[correct_key] = exp_time
print("# Generated key:", correct_key)


# 交互式验证循环（支持多次尝试，直到输入有效密钥）
while True:
    user_input = input("Enter your license Key: ").strip()
    if validate_key(user_input):
        break

input("Press any Key to close")

✅ 关键改进说明：