存储过程未使用 sqlalchemy/pyodbc 提交

Question

我正在使用 sqlalchemy/pyodbc 连接到 MS SQL 2012 服务器。我选择了 sqlalchemy，因为它使用.read_sql和与熊猫数据框直接集成.to_sql。

在高层次上，我的代码是：

df = dataframe.read_sql("EXEC sp_getsomedata")
<do some stuff here>
finaldf.to_sql("loader_table", engine,...)

这部分很好用，很容易阅读等等。问题是我必须运行一个最终存储过程来将加载器表中的数据插入到实时表中。通常，sqlalchemy 知道在 INSERT/UPDATE/DELETE 之后提交，但是当我运行这个最终存储过程时不想为我做提交。

在尝试了多种方法后，我看到 db 中的事务未提交。我知道 sqlalchemy 非常灵活，我使用了大约 3% 的功能，最简单的方法是什么？我想我需要使用 sqlalchemy 核心而不是 ORM。我看到了使用 sessionmaker 的例子，但我认为它垄断了引擎对象并且不允许熊猫访问它。

connection = engine.connect()
transaction = connection.begin()
connection.execute("EXEC sp_doLoaderStuff")
transaction.commit()
connection.close()

我曾尝试.execute从连接级别调用，从游标级别调用，甚至使用该.raw_connection()方法都没有成功。

connection = engine.raw_connection()
connection.autocommit = True
cursor = connection.cursor()
cursor.execute("EXEC sp_doLoaderStuff")
connection.commit()
connection.close()

任何想法我在这里缺少什么？

Answer 1

完全自作自受。使用正常工作的 raw_connection() 方法的正确工作代码是：

connection = engine.raw_connection()
cursor = connection.cursor()
cursor.execute("EXEC sp_doLoaderStuff")
connection.commit()
connection.close()