如何在Python中读取带注释的SQL文件并写入数据库？

import re
import sqlite3
from pathlib import Path

def execute_sql_file_with_comments(file_path, db_connection):
    """
    读取带注释的SQL文件并执行到数据库
    支持：
    - 单行注释（-- 或 #）
    - 多行注释（/* */）
    - 保留分号作为语句分隔符
    """
    # 读取文件内容
    sql_content = Path(file_path).read_text(encoding='utf-8')
    
    # 移除多行注释 /* */
    sql_content = re.sub(r'/\*.*?\*/', '', sql_content, flags=re.DOTALL)
    
    # 移除单行注释 --
    sql_content = re.sub(r'--.*$', '', sql_content, flags=re.MULTILINE)
    
    # 移除单行注释 #
    sql_content = re.sub(r'#.*$', '', sql_content, flags=re.MULTILINE)
    
    # 按分号分割SQL语句，过滤空语句
    statements = [
        stmt.strip() 
        for stmt in sql_content.split(';') 
        if stmt.strip()
    ]
    
    # 执行每条SQL语句
    cursor = db_connection.cursor()
    for stmt in statements:
        try:
            cursor.execute(stmt)
            print(f"执行成功: {stmt[:50]}...")
        except Exception as e:
            print(f"执行失败: {stmt[:50]}... 错误: {e}")
    
    db_connection.commit()
    cursor.close()

# 使用示例
if __name__ == "__main__":
    # 创建SQLite内存数据库（示例）
    conn = sqlite3.connect(':memory:')
    
    # 示例SQL文件内容（实际使用时从文件读取）
    sample_sql = """
    -- 创建用户表
    CREATE TABLE users (
        id INTEGER PRIMARY KEY,
        /* 用户名字段 */
        username TEXT NOT NULL,
        email TEXT UNIQUE,  # 邮箱地址
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    );
    
    /*
      批量插入测试数据
      这是一个多行注释
    */
    INSERT INTO users (username, email) 
    VALUES ('alice', 'alice@example.com');
    
    INSERT INTO users (username, email) 
    VALUES ('bob', 'bob@example.com'); -- 这是Bob的用户
    """
    
    # 将示例SQL写入临时文件
    temp_file = 'temp_sample.sql'
    Path(temp_file).write_text(sample_sql)
    
    # 执行SQL文件
    execute_sql_file_with_comments(temp_file, conn)
    
    # 验证结果
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM users")
    print("\n插入的数据:", cursor.fetchall())
    
    # 清理
    Path(temp_file).unlink(missing_ok=True)
    conn.close()

核心思路：

1. 用正则表达式分三步移除注释：先处理多行注释/* */，再处理单行注释--和#
2. 按分号分割得到独立SQL语句
3. 逐条执行并处理异常

注意点：

• 正则re.DOTALL让.匹配换行符，确保多行注释能正确移除
• 语句分割前先移除注释，避免注释中的分号干扰分割
• 实际生产环境需要更完善的错误处理和事务管理

一句话建议： 用正则过滤注释后按分号分割执行最可靠。

我搞过…最后别人给了思路不要正则
用写语法解析器的思路

入栈出栈 按规范处理 23333

水平不够写得超级难看