在 sqlalchemy 表中标记列

Question

我想向 sqlalchemy 模型中的列添加自定义属性“可搜索”。目的是仅提取这些列的数据（使用 ModelFoo.__ table__.columns）并将数据放入 solr。有没有办法可以使用自定义属性标记某些列？

class ModelFoo(AppBase):
    __tablename__ = 'foo'
    id = Column("id", Integer, primary_key=True, autoincrement=True)
    os = Column(String, nullable=False, searchable=True)
    platform = Column(String, searchable=True)

默认情况下，我在尝试上述操作时会收到以下错误：

sqlalchemy.exc.ArgumentError: Unknown arguments passed to Column: ['searchable']

我正在寻找一种仅向 solr 添加“可搜索”列的通用方法，类似于以下内容：

for table in Base.metadata.tables.values():
    keys = [str(key) for key in table.columns if key.searchable] 
    solr.add(session.query(*keys).all())

在上面的代码中，我正在寻找一些简短的解决方案或替代方案来让“key.searchable”工作。希望这可以澄清问题。

Answer 1

我使用模型中的单独属性解决了这个问题：

class ModelFoo(Base):
    __tablename__ = 'foo'
    id = Column("id", Integer, primary_key=True, autoincrement=True)
    os = Column(String, nullable=False)
    platform = Column(String)
    search_cols = ["os", "value"]

for k, v in list(Base._decl_class_registry.items()):
    if (hasattr(v, "search_cols")):
        cols = [getattr(v, val) for val in v.search_cols]
        query = sess.query(*cols)
        solr.add(query.all())

Answer 2

我对你的问题有点不清楚，但看起来你正试图从你提供的link 做这样的事情：

si.add(Book.objects.all())

其中Book.objects.all() 是来自 ORM 映射表的记录列表。请注意，文档说对象列表也是可以接受的。我认为这里的解决方案是使用 SQLAlchemy 查询方法来构建仅包含您想要的字段的记录。使用您的示例如下所示：

si.add(session.query(ModelFoo.os, ModelFoo.platform).all())