如何用Python或Bash为每条DROP TABLE命令添加前置提示SQL语句

当然可以，用Bash或Python都能轻松完成这个需求，下面是两种具体实现方式：

Bash 实现 #

用sed命令就能快速处理，它支持正则匹配和文本替换，直接在命令行运行即可：

# 处理文件并备份原文件（备份名为your_file.sql.bak）
sed -i.bak 's/^drop table \(.*\);/select '\''dropping table \1'\'';\n&/' your_file.sql

说明： #

• -i.bak：修改文件的同时创建备份，避免误操作丢失原内容
• 正则表达式^drop table \(.*\);匹配以drop table开头的行，捕获表名部分
• 替换部分会先插入select 'dropping table 表名';，再保留原有的drop table语句

Python 实现 #

如果需要更灵活的逻辑（比如处理复杂格式的SQL语句），用Python脚本更合适：

# 输入文件名和输出文件名可根据实际修改
input_file = "your_file.sql"
output_file = "output.sql"

with open(input_file, 'r') as f_in, open(output_file, 'w') as f_out:
    for line in f_in:
        stripped_line = line.strip()
        # 判断是否是DROP TABLE语句（忽略大小写）
        if stripped_line.lower().startswith('drop table'):
            # 提取表名：拆分后取第三个元素，再去掉末尾的分号
            table_name = stripped_line.split()[2].rstrip(';')
            # 写入提示语句
            f_out.write(f"select 'dropping table {table_name}';\n")
        # 写入原语句（跳过空行可根据需求调整）
        if stripped_line:
            f_out.write(stripped_line + '\n')

说明： #

• 脚本会逐行读取原文件，识别出DROP TABLE语句后，插入对应的提示语句
• 支持忽略大小写的匹配，比如DROP TABLE或drop table都能被识别
• 生成的结果会保存到新文件output.sql，不会修改原文件

内容的提问来源于stack exchange，提问作者Vsem Udachi