python - 在 SQLAlchemy 中，为什么我的 load_only 不过滤我指定的任何列？

Question

基本上我已经创建了一个数据库，在正常的完整查询中，这是我使用的代码和生成的响应。

db.session.query(User).all()

生成的查询如下:

<User(email='howard@howard.com', fullname='Howard', company='howard', address='None', password='howard')>,  <User(email='emailhoward', fullname='None', company='None', address='None', password='passwordhoward')>

这是合乎逻辑的，因为我正在从表中提取所有内容。但是，当我尝试使用 load_only 专门选择一列时，在本例中为电子邮件列。我使用的代码是:

db.session.query(User).options(load_only(User.address)).all()
db.session.query(User).options(load_only('email')).all()

这两个命令给我相同的结果:

<User(email='howard@howard.com', fullname='Howard', company='howard', address='None', password='howard')>,<User(email='emailhoward', fullname='None', company='None', address='None', password='passwordhoward')>

这非常奇怪，因为我应该在查询中只得到一列。但是，当我使用它时:

db.session.query(User.email).select_from(User).filter_by(email=email).first()[0]

它神奇地只为我返回一列。我需要使用 load_only，因为我有动态表，我想重用相同的函数，而不是维护多组函数。谁能告诉我 load_only 命令有什么问题，或者我做错了什么？

谢谢。

Answer 1

没有问题，只是有点误会；

<User(email='howard@howard.com', fullname='Howard', company='howard', address='None', password='howard')>

是string representation您的 User 模型对象，它是 User.__repr__() 方法在访问延迟列时拉入它们。

使用 load_only()您定义一组列最初为实体加载，同时推迟所有其他列。但是推迟列并不意味着它在某种程度上不可用或包含一些“无值(value)”标记(实际上它确实如此，在引擎盖下)。当第一次引用每个延迟属性时，SQLAlchemy 将发出 SELECT 以获取其值。这应该从日志中显而易见:

In [7]: u = session.query(User).options(load_only(User.email)).first()
2018-05-14 16:04:49,218 INFO sqlalchemy.engine.base.Engine SELECT user.id AS user_id, user.email AS user_email 
FROM user
 LIMIT ? OFFSET ?
2018-05-14 16:04:49,218 INFO sqlalchemy.engine.base.Engine (1, 0)

In [8]: u.fullname
2018-05-14 16:04:53,773 INFO sqlalchemy.engine.base.Engine SELECT user.fullname AS user_fullname 
FROM user 
WHERE user.id = ?
2018-05-14 16:04:53,773 INFO sqlalchemy.engine.base.Engine (2,)
Out[8]: 'Bar'

您可以使用检查 API 检查列是否已延迟。 InstanceState.unloaded保存没有加载值的键集。使用它，您可以将 User.__repr__ 修改为类似以下内容:

class User(Base):
    ...

    def __repr__(self):
        state = inspect(self)
        def ga(attr):
            return (repr(getattr(self, attr))
                    if attr not in state.unloaded
                    else "<deferred>")

        attrs = " ".join([f"{attr.key}={ga(attr.key)}"
                          for attr in state.attrs])
        return f"<User {attrs}>"

或者，您可以遍历 InstanceState.attrs显示 AttributeState.loaded_value ，如果尚未加载值，则计算结果为符号 NO_VALUE:

class User(Base):
    ...

    def __repr__(self):
        state = inspect(self)    
        attrs = " ".join([f"{attr.key}={attr.loaded_value!r}"
                          for attr in state.attrs])
        return f"<User {attrs}>"