使用SQLAlchemy

问题描述

非常感谢您，我一直在寻找没有成功的日子。我正在使用sqlAlchemy和MSsql。我只是插入一行，我想找回新创建的记录的ID。

与MSsql一起使用并返回ID的SQL查询是：

INSERT INTO [table](...) OUTPUT inserted.ID VALUES (...)

我使用的炼金术陈述是：

with connection() as conn:
    query = ...
    result_proxy = conn.execute(query)

如果在sql控制台中运行查询，则会获取ID，但是我无法弄清楚如何从execute语句返回的result_proxy中获取该ID。

我尝试过但不起作用的东西：

result_proxy.lastrowid
result_proxy.last_inserted_id
result_proxy.fetchone()
result_proxy.fetchall()

解决方法

您处在正确的轨道上。我只是试过了，对我有用：

import sqlalchemy as sa

# ...

# set up test environment
with engine.begin() as conn:
    conn.execute(sa.text("DROP TABLE IF EXISTS team"))
    conn.execute(
        sa.text(
            "CREATE TABLE team ("
            "id int identity primary key,"
            "prov nvarchar(5),"
            "city nvarchar(50),"
            "team_name nvarchar(50)"
            ")"
        )
    )

# perform the test
sql = """\
INSERT INTO team (prov,city,team_name) 
OUTPUT inserted.id
VALUES (:prov,:city,:team_name)
"""
with engine.begin() as conn:
    new_id = conn.execute(
        sa.text(sql),{"prov": "AB","city": "Calgary","team_name": "Flames"}
    ).scalar()
    print(new_id)  # 1