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

6yjfywim  于 4个月前  发布在  Python
关注(0)|答案(1)|浏览(71)

此问题在此处已有答案

pyodbc does not throw on SQL Server error(1个答案)
昨天就关门了。
我是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客户端运行此单个查询,将触发错误:[S 0001][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

我的问题

  • 我是否使用了正确的方法来执行查询?
  • 如果我的代码没有问题,我如何从Python中捕获实际的SQL异常?
eaf3rand

eaf3rand1#

execute命令在准备语句时可能会遇到问题,因为它实际上是多个语句。对于您正在尝试做的事情,我相信您可能想看看executemany
https://github.com/mkleehammer/pyodbc/wiki/Cursor#executemanysql-paras
它采用了不同的形式,并将您的值绑定为列表中的元组。
祝你好运!

相关问题