Question

由于某种原因，当我使用SQLAlchemy的union_all和.all()时，它返回的项目数不正确。

正如您在下面看到的那样，我将每一个细分，以查看错误所在。有谁知道为什么会这样？

>>> pn = PostNotification.query.filter_by(notified_id=1)
>>> cn = CommentNotification.query.filter_by(notified_id=1)
>>> pn.count()
4
>>> cn.count()
2
>>> u = pn.union_all(cn)
>>> u.count()
6
>>> all = u.all()
>>> len(all)
5

这是我的两个模型：

class NotificationMixin:
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(150), nullable=False)
    read = db.Column(db.Boolean, default=False)
    created = db.Column(db.DateTime, index=True, default=datetime.utcnow)

    @declared_attr
    def notifier_id(cls):
        return db.Column(db.Integer, db.ForeignKey('user.id'))

    @declared_attr
    def notified_id(cls):
        return db.Column(db.Integer, db.ForeignKey('user.id'))


class PostNotification(db.Model, NotificationMixin):
    post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
    comment_id = db.Column(db.Integer)

    def __repr__(self):
        return '<PostNotification {}>'.format(self.name)


class CommentNotification(db.Model, NotificationMixin):
    post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
    comment_id = db.Column(db.Integer, db.ForeignKey('post_comment.id'))

    def __repr__(self):
        return '<CommentNotification {}>'.format(self.name)

更新：

Here is a screenshot of the data that represents the two models

当我明确定义列时，使用union_all时没有问题。仅在我db.session.query(PostNotification)和db.session.query(CommentNotification)时返回错误的记录量。

pn = db.session.query(
    PostNotification.id,
    PostNotification.name,
    PostNotification.read,
    PostNotification.created,
    PostNotification.post_id,
    PostNotification.comment_id,
    PostNotification.notifier_id,
    PostNotification.notified_id).filter_by(
        notified_id=1)

cn = db.session.query(
    CommentNotification.id,
    CommentNotification.name,
    CommentNotification.read,
    CommentNotification.created,
    CommentNotification.post_id,
    CommentNotification.comment_id,
    CommentNotification.notifier_id,
    CommentNotification.notified_id).filter_by(
        notified_id=1)

u = pn.union_all(cn).order_by(PostNotification.created.desc())

>>> pn.count()
4
>>> cn.count()
2
u.count()
6
>>> all = u.all()
>>> len(all)
6

与此有关的问题是我丢失了模型，并且我的关系消失了。因此，我必须使用此非常丑陋的解决方法。仅当您看到https://i.stack.imgur.com/UHfo7.jpg中的数据时，这才有意义。

result = []
for row in u:
    if 'post' in row.name.split('_'):
        n = PostNotification.query.filter_by(id=row.id).first()
        result.append(n)
    if 'comment' in row.name.split('_'):
        n = CommentNotification.query.filter_by(id=row.id).first()
        result.append(n)

现在，我的result是降序排列的，两个表都通过union_all合并了，我的关系又恢复了原样。现在的问题是，我显然不能使用result.paginate，因为result现在是list。

Answer 1

联合u并不是多态的，因为它可以识别出哪些行代表PostNotification和哪些CommentNotification实体–它只是将所有行都视为代表主要实体{{ 1}}。

还可能在两个表中都有2个“相同”通知，即它们的主键数值相同。 SQLAlchemy在查询as noted here by the author of SQLAlchemy时会根据主键对模型实体进行重复数据删除，因此PostNotification返回的结果较少。另一方面，len(u.all())在数据库中计数为，因此对所有行进行计数。如果查询属性或多个实体，则不会发生重复数据删除。

Answer 2

好像我想通了。我现在可以直接查询AbstractNotification

db.session.query(AbstractNotification).all()

SQLAlchemy union_all和all（）返回错误的项目数

2 个答案: