python - sqlalchemy.exc.ResourceClosedError : This Connection is closed when inserting after select 错误

Question

我正在从 SQLite 数据库执行 select()，然后执行 insert():

engine = create_engine('sqlite:///testdb.db')
metadata = MetaData(bind=engine)
test = Table('test', metadata, autoload=True)

# Select all from pending_data
sel = select([test])
res = engine.execute(sel)

print res

# do an insert into pending_data
test.insert()\
    .values(info='blah')\
    .execute()

当我的代码执行插入行时，出现此错误:

sqlalchemy.exc.ResourceClosedError: This Connection is closed

但是，如果我将 res 转换成这样的列表:

res = list(engine.execute(sel))

我的代码运行良好。这是怎么回事？

Answer 1

SQLAlchemy 有两个必须了解的概念:连接和引擎。一个引擎可以支持多个并发连接。在您的示例中，您将表绑定(bind)到引擎。现在，无论何时调用 .execute，都会为您执行的每个查询创建一个新连接。但是 sqlite3 只允许 1 个同时“连接”。

解决此问题的最佳方法是创建连接并显式使用它，而不是引擎自动创建的连接；并通过 with 语句使用连接，这确保连接将在 block 的末尾关闭:

engine = create_engine('sqlite:///testdb.db')
metadata = MetaData(bind=engine)
test = Table('test', metadata, autoload=True)

with engine.connect() as connection:
    # Select all from pending_data
    sel = select([test])
    res = connection.execute(sel)

    # do an insert into pending_data
    connection.execute(test.insert().values(info='blah'))

---

要了解此行为，会发生错误，因为您在隐式创建和保存的集合中持有事件游标(这由 res 变量引用；直到您使用它或关闭它，或删除对它的引用，游标和连接将处于事件状态，并且数据库将被锁定)。

当您执行 list(res) 时，您正在使用游标，它会被 SQLAlchemy 关闭；如果结果的引用计数下降到 0，也会发生同样的情况。

您也可以尝试以下方法来了解这一点，它们会按您的预期工作:

res = engine.execute(sel)
print(res)
res.close()  # close explicitly

或

res = engine.execute(sel)
print(res)
del res  # drop the only reference to res

因此始终完全使用 ResultProxy 或显式关闭它，或者在完成后删除对它的引用。

如果您重复使用相同的连接，这不是问题；仅当您创建与 sqlite3 数据库的新连接时(postgresql、mysql、oracle 等也可以很好地处理)。