如何用Python脚本+INI文件安全管理多客户端并发API令牌更新

核心问题分析 #

你遇到的令牌覆盖/丢失问题，本质是单INI文件的多进程读写原子性未保障：即使加了文件锁、只修改对应段，若流程是「读全文件→内存修改→写回文件」，期间如果锁的持有逻辑有漏洞（比如提前释放锁），或者多个进程先后读取旧数据后写回，就会导致后写的进程覆盖前一个的修改。

可行解决方案 #

方案1：拆分单INI为客户端专属文件（推荐） #

彻底规避多客户端的文件冲突，每个客户端的凭据单独存储：

• 每个客户端对应一个独立文件，比如credentials_{client_name}.ini或client_{client_id}.ini
• 操作流程：
  1. 处理目标客户端时，仅打开其专属文件
  2. 用fcntl.LOCK_EX加独占写锁（持有锁直到整个读写操作完成）
  3. 读取文件内容，检查令牌是否过期
  4. 若需刷新，调用Aruba API获取新令牌，更新文件内容
  5. 写入文件，释放锁
• 优势：完全隔离不同客户端的文件操作，无需处理复杂的INI段定位，实现简单且冲突概率为0

方案2：优化单INI文件的原子读写流程 #

若必须保留单INI文件，需严格保障「读取-修改-写入」的原子性：

• 修正后的操作流程：
  1. 打开INI文件，用fcntl.LOCK_EX加独占写锁（全程持有，直到写入完成）
  2. 读取整个文件内容到内存
  3. 定位目标客户端的配置段，更新access_token和refresh_token
  4. 先将修改后的内容写入临时文件（如credentials.ini.tmp）
  5. 用os.replace()原子替换原INI文件（避免写入中断导致原文件损坏）
  6. 关闭文件，释放锁
• 关键细节：
  • 锁必须持有到文件替换完成，不能提前释放
  • 临时文件要和原文件在同一文件系统，确保os.replace()的原子性
  • 令牌刷新失败时，不要覆盖原文件中的有效令牌，保留旧值避免服务中断

方案3：单段原子更新（复杂度高，不推荐） #

针对单INI文件的特定段进行精准更新，避免重写整个文件：

• 流程：
  1. 加独占锁后，读取文件并定位目标客户端段的起始、结束偏移量
  2. 生成修改后的段内容，对比原段长度
  3. 若长度一致：直接定位到起始偏移量，覆盖写入该段
  4. 若长度变化：仍需重写整个文件（文本文件无法局部插入/删除字符而不影响后续内容）
• 劣势：需要手动解析INI段的位置，实现复杂且易出错，仅适合特殊场景

额外注意事项 #

• 锁的可靠性：Linux下用fcntl的独占锁即可，避免用flock（部分场景下兼容性差）
• 重试机制：令牌刷新失败时，添加重试逻辑，最多重试2-3次，失败则保留旧令牌
• 日志记录：记录每次令牌刷新、文件操作的时间、客户端ID、操作结果，方便排查问题

内容的提问来源于stack exchange，提问作者Raul Chiarella