Python从SQL文件创建数据库首次成功后续失败求助

我一眼就看出问题所在啦——你第一次运行成功是因为数据库和表都还不存在，所有SQL语句都能正常执行，但后续再跑的时候，CREATE DATABASE X``这句会尝试创建已经存在的数据库，直接触发MySQL的报错，导致整个脚本卡壳。

给你两个关键的修复点，保证脚本可以重复安全运行：

1. 修改SQL脚本，避免重复创建报错 #

把你的.sql文件里的创建语句加上IF NOT EXISTS判断，这样即使数据库/表已经存在，也不会抛出错误：

CREATE DATABASE IF NOT EXISTS `X`;
USE `X`;
CREATE TABLE IF NOT EXISTS `profile` (
    `id` int(11) NOT NULL AUTO_INCREMENT,
    -- 补全你未写完的表结构即可
    PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

2. 优化Python代码，增加健壮性 #

原来的代码缺少异常处理和显式提交，而且没有确保连接/游标正确关闭。调整后的代码不仅能处理错误，还能让你看到执行过程的反馈：

import mysql.connector
from mysql.connector import Error

SCHEMA_PATH = 'mysql_schema.sql'

try:
    # 建立数据库连接
    mysql_conn = mysql.connector.connect(
        user='X',
        password='X',
        host='X'
    )
    
    if mysql_conn.is_connected():
        mysql_cursor = mysql_conn.cursor()
        
        # 读取SQL脚本内容
        with open(SCHEMA_PATH, 'rt') as f:
            schema = f.read()
        
        # 执行多语句脚本，遍历每个执行结果
        for result in mysql_cursor.execute(schema, multi=True):
            if result.with_rows:
                print(f"查询结果: {result.fetchall()}")
            else:
                print(f"执行完成，受影响行数: {result.rowcount}")
        
        # 显式提交（DDL语句可能自动提交，但显式提交更稳妥）
        mysql_conn.commit()
        print("数据库初始化完成！")

except Error as e:
    print(f"执行过程中出错: {e}")
    # 出错时回滚操作
    if 'mysql_conn' in locals() and mysql_conn.is_connected():
        mysql_conn.rollback()

finally:
    # 确保游标和连接都关闭
    if 'mysql_cursor' in locals() and mysql_cursor:
        mysql_cursor.close()
    if 'mysql_conn' in locals() and mysql_conn.is_connected():
        mysql_conn.close()

为什么这样改？ #

• IF NOT EXISTS是MySQL的语法，专门用来避免重复创建对象的错误，这是脚本可重复执行的核心。
• 增加异常处理可以帮你快速定位问题，比如密码错误、主机连接失败这些情况都能直接看到报错信息。
• 显式提交和回滚能保证操作的原子性，避免出现半完成的数据库状态。
• 用try-finally确保连接关闭，不会因为程序异常导致连接泄漏。

内容的提问来源于stack exchange，提问作者chase devilla