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

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

本文介绍如何使用字典持久化存储密钥及其对应过期时间，实现“首次验证即启动倒计时”的精准过期控制，支持批量密钥管理，避免每次运行都重置有效期。

本文介绍如何使用字典持久化存储密钥及其对应过期时间，实现“首次验证即启动倒计时”的精准过期控制，支持批量密钥管理，避免每次运行都重置有效期。

在实际开发中（如许可证校验、临时访问令牌等场景），密钥的过期逻辑不应依赖生成时刻，而应以首次成功验证时间为起点——这正是原代码的核心缺陷：generate_key() 每次调用都生成新过期时间，且未持久化记录；同时 if expiration_time == timedelta: 等判断逻辑存在类型错误，导致失效检测完全失效。

正确做法是：将密钥与它的过期时间作为键值对存入字典（如 key_expiration_map），后续每次输入都查表验证，既保证单密钥生命周期独立，又支持多密钥共存管理。

以下是优化后的完整实现：

from datetime import datetime, timedelta

# 全局字典：存储「密钥 → 过期时间」映射
key_expiration_map = {}


def generate_key(key_str: str = "4255", minutes: int = 1) -> str:
    """生成密钥并注册其过期时间（仅首次注册有效）"""
    if key_str not in key_expiration_map:
        expiration_time = datetime.now() + timedelta(minutes=minutes)
        key_expiration_map[key_str] = expiration_time
        print(f"# Registered key '{key_str}' with {minutes}-minute expiry")
    return key_str


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


def validate_key(user_input: str) -> bool:
    """验证用户输入的密钥是否有效且未过期"""
    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:
        print("✅ Successfully Loaded")
        print(f"⏳ Expires at: {expiration_time.strftime('%Y-%m-%d %H:%M:%S')}")
        return True


# 示例：预注册多个密钥（可扩展为从文件/数据库加载）
generate_key("4255", minutes=1)      # 1分钟过期
generate_key("X9A2", minutes=5)      # 5分钟过期
generate_key("TEST", minutes=30)     # 30分钟过期

print("\n? License Key Validator (Ctrl+C to exit)")
while True:
    try:
        user_input = input("\nEnter your license Key: ").strip()
        if not user_input:
            continue
        if validate_key(user_input):
            break
    except KeyboardInterrupt:
        print("\n? Exiting...")
        break

✅ 关键改进说明：