使用SQLAlchemy按匹配的标签排序

Question

我有以下表格。我想按标签搜索，并按匹配的标签数量排序。

class DocumentTag(db.Model):
__tablename__ = "document_tag"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(50), nullable=False)
user_document_id = db.Column(db.Integer, db.ForeignKey("user_document.id"))
order = db.Column(db.Integer, nullable=False)
user_document = db.relationship("Document")
class UserDocument(db.Model):
__tablename__ = "user_document"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
document_id = db.Column(db.Integer, db.ForeignKey("document.id"), nullable=False)
date_added = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
last_accessed = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
document_tags = db.relationship(
"DocumentTag",
order_by="DocumentTag.order",
collection_class=ordering_list("order"),
)
document = db.relationship("Document")
document_tag = db.relationship("DocumentTag")

我试着用这样的标签搜索文档。

tags = ['tag_1', 'tag_2'] # this is tags to search
documents_tags_count = (
db.session.query(
UserDocument, func.count(DocumentTag.id).label("num_corres_tags")
)
.join(UserDocument, DocumentTag.user_document) # in order to count combination of document_tags and tags
.filter(
UserDocument.document_tags.any(DocumentTag.name.in_(tags)),
UserDocument.user_id == current_user.id,
)
.group_by(UserDocument)
.order_by(desc("num_corres_tags"))
)

然而，标签num_corres_tags显示的不是经过过滤的标签数量，而是属于文档的所有标签数量。我想做的是通过匹配(过滤(标签的数量来获得有序的结果。

我在这里看到了一些类似的MySQL问题，但我找不到任何SQLAlchemy的解决方案，因为我对SQL真的很陌生。

有什么办法解决我的问题吗？

Answer 1

由于显式联接DocumentTag是为了计数，因此也可以使用该联接进行筛选，而不是单独使用EXISTS子查询表达式：

documents_tags_count = (
db.session.query(
UserDocument, func.count(DocumentTag.id).label("num_corres_tags")
)
.join(UserDocument, DocumentTag.user_document) # in order to count combination of document_tags and tags
.filter(
DocumentTag.name.in_(tags),
UserDocument.user_id == current_user.id,
)
.group_by(UserDocument)
.order_by(desc("num_corres_tags"))
)

现在，结果仅限于匹配的标签，计数应该是您所期望的。