0

我一直在使用 SQLAlchemy 连接到 MySQL 数据库。到目前为止，它一直运行良好，但我遇到了一些麻烦。我想在我的一个类被查询后调用一个方法，它会分配一些新的变量。问题是每个查询只有一个用户最终会使用我的“on_query”方法设置变量。

...stuff setting up a connection to the database as well as a SQL session.
   All were done using default values.

class UserTest(Base):
    __tablename__ = "testtable"

    ID = Column(Integer(unsigned=True),
                primary_key=True, nullable=False, autoincrement=True)

    def __init__(self):
        pass

    def on_query(self):
        self.foo = "bar"

query = session.query(UserTest)

for u in query:
    u.on_query()

print [hasattr(u, "foo") for u in query]
#prints [False, False, False, False, False, False, False, False, False, True]

session.close()

我发现如果我要通过迭代查询来创建一个列表，'on_query' 方法会按我的预期工作。

users = [u for u in query]
for u in users:
    u.on_query()
print [hasattr(u, "foo") for u in users]
#prints [True, True, True, True, True, True, True, True, True, True]

这是什么原因？有没有办法阻止它每次迭代都删除“foo”变量？如果可能的话，我更愿意将用户保留在查询对象中。

4

1 回答 1

1

在第一种情况下，查询实际上产生了两次，一次是 for 循环，一次是 print 语句。

由于 for 循环不断替换u引用，因此在循环结束时，只有最后一个保留在 SQLAlchemy 会话中。您可以使用 来验证这一点print session.identity_map。

当 print 语句第二次产生查询时，除了最后一个之外的所有查询都必须再次从数据库中检索，而最后一个则直接从 SQLAlchemy 会话中检索，因此具有foo属性集。

编辑：更透明地做到这一点的一种方法是使用重构器，例如

class UserTest(Base):
    ...

    @orm.reconstructor
    def init_on_load(self):
        self.foo = "bar"
于 2012-07-16T01:11:07.603 回答