sqlalchemy在sqlite中使用多个SQL语句执行脚本文件

Question

我正在使用SQLalchemy连接到SQLite数据库。我想执行一些文件，script.sql，其中包含多个 sql语句。

以前，我正在使用sqlite3库并运行

with sqlite3.connect('my_database.db') as conn:
  with open('script.sql') as s:
    conn.executescript(s.read())

移动到sqlalchemy，我发现这个问题，我从中使用session实例编写了新代码：

session.execute(text(s.read())

但是，这返回错误，因为我的脚本包含多个语句。

sqlite3.Warning: You can only execute one statement at a time.

使用sqlalchemy执行大型脚本文件的最佳方法是什么？

Answer 1

Sqlalchemy本身仅调用光标的execute方法，该方法不支持执行多个语句。但是，可以直接访问DB API连接，并调用其executescript方法：

import sqlalchemy as sa

engine = sa.create_engine('sqlite:///so72341960.db')

# Create a table for our example.
tbl = sa.Table(
    'mytable',
    sa.MetaData(),
    sa.Column('id', sa.Integer, primary_key=True),
    sa.Column('name', sa.String),
)

# Ensure we start with a clean table.
tbl.drop(engine, checkfirst=True)
tbl.create(engine)

SQL = """
INSERT INTO mytable (name) VALUES ('Alice');
INSERT INTO mytable (name) VALUES ('Bob');
"""

with engine.begin() as conn:
    dbapi_conn = conn.connection
    dbapi_conn.executescript(SQL)

Answer 2

这是因为SQLite3仅支持一次执行一个语句。如果要执行多个语句，则需要单独命令。例如，

statements = [] // put your statements in this array
for statement in statements:
  session.execute(text(statement))

这将迭代一系列语句，然后依次执行每个语句。