快速体验
- 打开 InsCode(快马)平台 https://www.inscode.net
- 输入框内输入如下内容：
创建一个快速环境初始化工具，功能：1. 读取配置文件确定需要TRUNCATE的表；2. 支持执行前后执行自定义SQL；3. 生成回滚脚本；4. 命令行和API两种调用方式；5. 进度显示和结果通知。使用Python编写，要求支持.env配置，能处理常见的数据库错误情况。- 点击'项目生成'按钮，等待项目生成完整后预览效果
今天在开发过程中遇到一个高频需求：需要频繁清空测试数据库的表数据，但又不想每次都手动执行一堆TRUNCATE语句。于是花了点时间做了个自动化小工具，分享下实现思路和踩坑经验。
为什么需要TRUNCATE工具每次跑测试用例前，最头疼的就是准备测试数据。DELETE语句太慢，DROP TABLE又太暴力，TRUNCATE成了最佳选择。但手动操作既容易出错又浪费时间，特别是当有外键约束时，还得考虑执行顺序。
核心功能设计这个小工具主要解决三个痛点：
- 一键清空指定表（支持表名白名单配置）
- 自动处理外键约束问题
提供执行前后的钩子函数
配置文件设计用.env文件存储数据库连接信息，再单独建个tables.json配置需要清理的表。这样做的好处是：
- 敏感信息不进代码库
- 表名单可版本化管理
不同环境可以配置不同表集合
关键实现细节用Python的click库处理命令行参数，主要逻辑分几步：
- 先检查外键约束，自动生成正确的TRUNCATE顺序
- 执行pre-hook.sql里的初始化语句
- 批量执行TRUNCATE并记录日志
最后跑post-hook.sql的收尾工作
错误处理要点特别要注意几个常见问题：
- 事务中TRUNCATE的行为差异（MySQL和PostgreSQL不同）
- 有外键引用时的级联处理
大表TRUNCATE的锁等待超时 工具里加了自动重试和超时机制，避免卡死。
回滚方案虽然TRUNCATE不能像DELETE那样用事务回滚，但工具会：
- 自动备份表结构
- 记录最后100条数据（如果表不大）
生成逆向SQL脚本
扩展玩法后来发现这个工具还能用来：
- 定期清理日志表
- 做数据脱敏前的清场
- 配合CI/CD做环境初始化
实际用下来，原本需要半小时的手动操作现在10秒搞定。特别是用InsCode(快马)平台测试时，发现他们的在线编辑器可以直接跑Python脚本，连本地环境都不用配，还能一键分享给同事协作。
如果你们团队也需要频繁初始化测试环境，强烈建议试试这个方案。代码我已经放在InsCode上，可以直接fork修改。他们的云环境特别适合这种小工具，不用操心服务器部署，点个按钮就能跑起来。
快速体验
- 打开 InsCode(快马)平台 https://www.inscode.net
- 输入框内输入如下内容：
创建一个快速环境初始化工具，功能：1. 读取配置文件确定需要TRUNCATE的表；2. 支持执行前后执行自定义SQL；3. 生成回滚脚本；4. 命令行和API两种调用方式；5. 进度显示和结果通知。使用Python编写，要求支持.env配置，能处理常见的数据库错误情况。- 点击'项目生成'按钮，等待项目生成完整后预览效果
