如何使用Python/SQLAlchemy和MSSQL在一个SQL中执行多个插入/更新查询？

Question

我是 python 和 SQLAlchemy 的新手，我试图了解如何使用 Python/SQLAlchemy 在一个 SQL 中执行多个插入/更新查询：

要求 在一条 SQL 中执行多个插入/更新：

DECLARE @age INT = 160
INSERT INTO TEST_TABLE VALUES ('QZ_TEST', @age + 1)
INSERT INTO TEST_TABLE VALUES ('QZ_TEST', 'not a number')
INSERT INTO ANOTHER_TABLE VALUES ('QZ_TEST', @age + 2)

理解这个查询看起来很难看，但我们确实有很多类似的查询。 （我们使用的是已有 20 年左右历史的遗留数据库）

Python代码

def OdbcEngineSA(driver, conn_str):
    def connect():
        return pyodbc.connect(conn_str, autocommit=True, timeout=120)
    return sqlalchemy.create_engine(driver + '://', creator=connect)

def get_db_connection():
    return OdbcEngineSA('mssql', 'DSN=mssql;Server=server,15001;database=DEV;UID=user_abc;PWD=pw_')

def main():
    db_connection = get_db_connection()
    sql = """
    DECLARE @age INT = 160
    INSERT INTO TEST_TABLE VALUES ('QZ_TEST', @age + 1)
    INSERT INTO TEST_TABLE VALUES ('QZ_TEST', 'not a number')
    INSERT INTO ANOTHER_TABLE VALUES ('QZ_TEST', @age + 2)
    """
    try:
        db_connection.execute(sql)
        db_connection.commit()
        logger.info('db updated')   
    except Exception as e:
        logger.error('Exception captured as expected: %s', e)
        db_connection.rollback()

请注意

INSERT INTO TEST_TABLE VALUES ('QZ_TEST', 'not a number')

如果我使用 SQL 客户端运行此单个查询，

将触发错误： [S0001][245] 将 varchar 值“不是数字”转换为数据类型 int 时转换失败。

我期待Python捕获异常，但是，Python代码运行时没有任何异常。即使我将 SQL 替换为:

，Python 也没有捕获到异常

BEGIN TRY
        DECLARE @age INT = 160
        INSERT INTO TEST_TABLE VALUES ('QZ_TEST', @age + 1)
        INSERT INTO TEST_TABLE VALUES ('QZ_TEST', 'not a number')
        INSERT INTO ANOTHER_TABLE VALUES ('QZ_TEST', @age + 2)
END TRY
    BEGIN CATCH
        DECLARE @ErrorMessage NVARCHAR(4000)
        DECLARE @ErrorSeverity INT
        DECLARE @ErrorState INT

        SELECT 
            @ErrorMessage = ERROR_MESSAGE(),
            @ErrorSeverity = ERROR_SEVERITY(),
            @ErrorState = ERROR_STATE()

        RAISERROR (@ErrorMessage,
                   @ErrorSeverity, -- Level 16
                   @ErrorState
                   )
    END CATCH

我的问题

• 我是否使用了正确的方法来执行查询？
• 如果我的代码没问题，我如何捕获实际的 SQL 异常 蟒蛇？

Answer 1

execute

命令在准备语句时可能会出现问题，因为它实际上是多个语句。对于您想要做的事情，我相信您可能想看看

executemany

：

https://github.com/mkleehammer/pyodbc/wiki/Cursor#executemanysql-paras

它采用不同的形式，并将您的 INSERT 值绑定为列表中的元组。

祝你好运！