python 依次执行多条sql

原创

mob64ca12e36a1d 2023-11-29 09:30:17 ©著作权

文章标签 SQL 数据库连接 Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e36a1d的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python依次执行多条SQL的方法

在数据分析和数据库管理中，有时候我们需要一次性执行多条SQL语句。Python是一种功能强大的编程语言，它提供了多种方法来执行多条SQL语句。本文将介绍两种常见的方法：使用数据库连接和使用SQLAlchemy。

方法一：使用数据库连接

Python的标准库中提供了多个用于数据库连接的模块，如sqlite3、pymysql、psycopg2等。这些模块可以用于连接不同类型的数据库，并执行SQL语句。

以下是一个使用sqlite3模块连接SQLite数据库，并依次执行多条SQL语句的示例：

import sqlite3

# 连接数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

# 创建表
cursor.execute('''
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY,
        name TEXT
    )
''')

# 插入数据
cursor.execute('''
    INSERT INTO users (name) VALUES ('Alice')
''')
cursor.execute('''
    INSERT INTO users (name) VALUES ('Bob')
''')

# 查询数据
cursor.execute('SELECT * FROM users')
result = cursor.fetchall()
for row in result:
    print(row)

# 关闭数据库连接
conn.commit()
conn.close()

在这个示例中，我们首先使用sqlite3模块连接SQLite数据库，然后使用execute()方法执行SQL语句。可以使用多次execute()方法执行多条SQL语句，例如创建表、插入数据和查询数据。最后，我们使用commit()方法提交事务，并使用close()方法关闭数据库连接。

方法二：使用SQLAlchemy

SQLAlchemy是一个流行的Python ORM（对象关系映射）工具，它提供了一个便捷的方式来连接数据库并执行SQL语句。

以下是一个使用SQLAlchemy连接SQLite数据库，并依次执行多条SQL语句的示例：

from sqlalchemy import create_engine, text

# 连接数据库
engine = create_engine('sqlite:///example.db')

# 创建表
create_table_sql = '''
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY,
        name TEXT
    )
'''
with engine.begin() as connection:
    connection.execute(text(create_table_sql))

# 插入数据
insert_data_sql = '''
    INSERT INTO users (name) VALUES (:name)
'''
with engine.begin() as connection:
    connection.execute(text(insert_data_sql), name='Alice')
    connection.execute(text(insert_data_sql), name='Bob')

# 查询数据
select_data_sql = '''
    SELECT * FROM users
'''
with engine.begin() as connection:
    result = connection.execute(text(select_data_sql))
    for row in result:
        print(dict(row))

# 关闭数据库连接
engine.dispose()

在这个示例中，我们首先使用create_engine()函数创建一个数据库连接引擎，然后使用begin()方法开始一个事务。使用execute()方法执行SQL语句，可以通过传递参数的方式插入数据。最后，我们使用dispose()方法关闭数据库连接。