当前位置:首页 > 行业动态 > 正文

python 如何执行.sql文件

在Python中执行.sql文件,通常需要借助一些第三方库,如sqlite3、pymysql等,这里以sqlite3为例,介绍如何在Python中执行.sql文件。

1、确保已经安装了sqlite3库,如果没有安装,可以使用以下命令进行安装:

pip install pysqlite3

2、创建一个.sql文件,test.sql,在这个文件中,编写SQL语句,

CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL,
    age INTEGER
);
INSERT INTO users (name, age) VALUES ('张三', 25);
INSERT INTO users (name, age) VALUES ('李四', 30);

3、接下来,使用Python的sqlite3库来执行这个.sql文件,以下是一个简单的示例:

import sqlite3
连接到SQLite数据库,如果不存在则创建一个新的数据库文件
conn = sqlite3.connect('test.db')
创建一个游标对象,用于执行SQL语句
cursor = conn.cursor()
读取.sql文件的内容
with open('test.sql', 'r', encoding='utf8') as f:
    sql_script = f.read()
使用游标对象执行SQL脚本
cursor.executescript(sql_script)
提交事务,将更改保存到数据库中
conn.commit()
关闭游标和连接
cursor.close()
conn.close()

4、运行上述Python代码,将会在当前目录下创建一个名为test.db的SQLite数据库文件,并在其中创建一个名为users的表,插入两条数据。

注意:在实际项目中,建议将数据库文件放在项目根目录下,而不是与.py文件同级目录,这样可以避免不同模块之间的文件路径问题,为了提高代码的可读性和可维护性,可以将数据库相关的操作封装到一个单独的模块中。

除了sqlite3库,还有其他一些第三方库可以用于执行.sql文件,如pymysql、psycopg2等,这些库的使用方式大同小异,主要区别在于连接的数据库类型不同(如MySQL、PostgreSQL等),在使用这些库时,需要先安装相应的库,然后按照上述步骤执行.sql文件。

0

随机文章