python批量导入sql文件
时间: 2024-05-30 14:07:05 浏览: 217
利用Python如何批量修改数据库执行Sql文件
5星 · 资源好评率100%
在 Python 中,你可以使用 pymysql 库连接到 MySQL 数据库,并使用该库提供的方法执行 SQL 语句。如果需要批量导入 SQL 文件,可以将 SQL 文件中的所有语句读入到一个字符串变量中,然后使用 pymysql 库提供的 execute() 方法执行该字符串变量即可。
以下是一个简单的示例代码,可以读取指定目录下的所有 SQL 文件,并批量导入到 MySQL 数据库中:
```python
import os
import pymysql
# 数据库连接信息
host = 'localhost'
port = 3306
user = 'root'
password = '123456'
database = 'test'
# 连接数据库
conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)
cursor = conn.cursor()
# SQL 文件目录
sql_dir = '/path/to/sql/files'
# 遍历 SQL 文件
for filename in os.listdir(sql_dir):
if filename.endswith('.sql'):
# 读取 SQL 文件内容
with open(os.path.join(sql_dir, filename), 'r') as f:
sql = f.read()
# 执行 SQL 语句
cursor.execute(sql)
conn.commit()
# 关闭数据库连接
cursor.close()
conn.close()
```
需要注意的是,如果 SQL 文件中包含多条语句,需要使用分号将它们分隔开。同时,如果 SQL 文件中包含创建数据库或表的语句,需要确保当前用户有足够的权限执行这些操作。
阅读全文