### Python 中导入 SQL 数据文件的方法 要在 Python 中实现导入 SQL 文件的功能，可以采用多种方法完成。以下是基于 `pymysql` 的解决方案以及一些注意事项。 #### 使用 pymysql 执行 SQL 脚本 可以通过读取 SQL 文件并逐行解析其内容来执行脚本中的命令。这种方法适用于大多数场景下的 SQL 文件导入需求[^4]。 ```python import pymysql def execute_sql_file(file_path, host="127.0.0.1", user="root", password="", db="test_db"): """ 执行 SQL 文件的内容到指定的 MySQL 数据库中。 参数: file_path (str): SQL 文件路径。 host (str): 数据库主机地址，默认为本地。 user (str): 用户名，默认为 root。 password (str): 密码，默认为空。 db (str): 数据库名称，默认为 test_db。 """ connection = None try: # 建立数据库连接 connection = pymysql.connect( host=host, user=user, password=password, database=db, charset='utf8mb4' ) cursor = connection.cursor() # 读取 SQL 文件 with open(file_path, 'r', encoding='utf-8') as sql_file: sql_commands = "" for line in sql_file: # 清除空白字符和注释 stripped_line = line.strip().replace("\n", "").replace("\r", "") if not stripped_line.startswith("--") and stripped_line != "": sql_commands += stripped_line # 当遇到分号时，表示一个完整的 SQL 语句结束 if ";" in stripped_line: pos = stripped_line.rfind(";") if pos + 1 == len(stripped_line): cursor.execute(sql_commands) connection.commit() # 重置 SQL 命令缓冲区 sql_commands = "" print("SQL 文件已成功导入") except Exception as error: print(f"发生错误: {error}") if connection is not None: connection.rollback() finally: if connection is not None: connection.close() # 示例调用 execute_sql_file('example.sql') ``` 以上代码实现了从 SQL 文件中读取内容并通过 `pymysql` 将其逐步应用至目标数据库的操作。需要注意的是，在实际环境中可能需要调整编码方式或其他参数以适配不同的 SQL 文件格式。 --- #### 注意事项 1. **SQL 文件结构**: 确保 SQL 文件遵循标准语法，并且每条指令都以分号 (`;`) 结束。 2. **权限管理**: 需要确认用于连接数据库的账户具有足够的权限去创建表、插入数据等操作。 3. **异常处理**: 如遇任何问题应立即停止事务提交以防破坏现有数据一致性。 ---