如何使用 python 运行 SQL 命令？

Question

我想连接到 MS SQL Server 并使用 python 执行 SQL 命令。我熟悉使用 SQLAlchemy 创建 SQL 表、pandas DataFrames 等，但是如何使用 python/pandas/SQLAlchemy 在 .sql 文件中执行 SQL？有更好的方法吗？

例如，我有一个包含 SQL 文本的文件“update.sql”：

truncate table dev.dbo.jobs
truncate table dev.dbo.workers
go
insert into dev.dbo.jobs select * from test.dbo.jobs
insert into dev.dbo.workers select * from test.dbo.workers

Answer 1

您可以使用 SQLAlchemy 的 connection.execute 来运行原始 SQL 查询。如果您将 sql 语句存储在一个文件中，那么它可能看起来像这样：

from sqlalchemy import create_engine
from sqlalchemy.sql import text

engine = create_engine('urltodb')
conn = engine.connect()

with open('file.sql', 'r') as f:
    for l in f:
        stmt = text(l)
        conn.execute(stmt)

conn.close()