如何在SQLAlchemy中跨多个表进行联合？

Question

我有一些不同的通知表，我想对所有表进行合并以向用户显示所有通知。但是，工会没有正常工作。

Python代码

def _get_notifications_query(self, unconfirmed_only=True):
    '''
    Return base query to return this users notifications.

    @param unconfirmed_only
    @return Query object
    '''        
    requests = (
        DBSession.query(FriendshipRequestNotification)
        .outerjoin(UserFriendshipRequestNotification,
                   UserFriendshipRequestNotification.notification_id==FriendshipRequestNotification.id)
        .filter(UserFriendshipRequestNotification.user_id==self.id))
    confirmations = (
        DBSession.query(FriendshipConfirmationNotification)
        .outerjoin(UserFriendshipConfirmationNotification,
                   UserFriendshipConfirmationNotification.notification_id==FriendshipConfirmationNotification.id)
        .filter(UserFriendshipConfirmationNotification.user_id==self.id))
    comments = (
        DBSession.query(CommentNotification)
        .outerjoin(UserCommentNotification,
                   UserCommentNotification.notification_id==CommentNotification.id)
        .filter(UserCommentNotification.user_id==self.id))

    if unconfirmed_only:
        requests.filter(UserFriendshipRequestNotification.is_confirmed==False)
        confirmations.filter(UserFriendshipConfirmationNotification.is_confirmed==False)
        comments.filter(UserCommentNotification.is_confirmed==False)

    return requests.union(confirmations, comments)

使用：user._get_notifications_query（unconfirmed_only = False）.all（）

SQL生成

SELECT anon_1.friendship_request_notifications_id AS anon_1_friendship_request_notifications_id, anon_1.friendship_request_notifications_created_at AS anon_1_friendship_request_notifications_created_at, anon_1.friendship_request_notifications_requester_id AS anon_1_friendship_request_notifications_requester_id 
FROM (SELECT friendship_request_notifications.id AS friendship_request_notifications_id, friendship_request_notifications.created_at AS friendship_request_notifications_created_at, friendship_request_notifications.requester_id AS friendship_request_notifications_requester_id 
FROM friendship_request_notifications LEFT OUTER JOIN users_friendship_request_notifications ON users_friendship_request_notifications.notification_id = friendship_request_notifications.id 
WHERE users_friendship_request_notifications.user_id = ? UNION SELECT friendship_confirmation_notifications.id AS friendship_confirmation_notifications_id, friendship_confirmation_notifications.created_at AS friendship_confirmation_notifications_created_at, friendship_confirmation_notifications.accepter_id AS friendship_confirmation_notifications_accepter_id 
FROM friendship_confirmation_notifications LEFT OUTER JOIN users_friendship_confirmation_notifications ON users_friendship_confirmation_notifications.notification_id = friendship_confirmation_notifications.id 
WHERE users_friendship_confirmation_notifications.user_id = ? UNION SELECT comment_notifications.id AS comment_notifications_id, comment_notifications.created_at AS comment_notifications_created_at, comment_notifications.comment_id AS comment_notifications_comment_id 
FROM comment_notifications LEFT OUTER JOIN users_comment_notifications ON users_comment_notifications.notification_id = comment_notifications.id 
WHERE users_comment_notifications.user_id = ?) AS anon_1

我期待这些事情

SELECT * FROM friendship_request_notifications
UNION
SELECT * FROM friendship_confirmation_notifications
UNION 
SELECT * FROM comment_notifications

另外，有什么方法可以对来自SQLAlchemy的聚合并集结果进行排序？

编辑

我应该提到sqlalchemy.sql.union()产生正确的SQL，但是我不知道如何从ORM（返回/计数记录）中利用它。

Answer 1

使用 sqlalchemy 联合不同的表当然是可能的，你只需要指定你从每个表中提取的列，以便它们可以正确组合。假设您有实体A和B，请执行以下操作：

from sqlalchemy.sql.expression import label
...

a = session.query(label('col1', A.someColumn))
b = session.query(label('col1', B.someOtherColumn))
both = a.union_all(b)

在这种情况下，您需要注意的是，如果A.someColumn并且B.someOtherColumn不能强制为相同类型，您的 dbms 可能会吓坏 :)

有关“真实世界”中的示例，请参见第 85、87 和 90 行：

https://gerrit.wikimedia.org/r/#/c/147312/4/wikimetrics/metrics/rolling_active_editor.py

这创建了两个合并在一起的不同表的子查询。然后在连接中稍后使用该子查询，并.c.column像往常一样访问其列。

Answer 2

我认为这不能与联合一起使用，即使假设查询是按照您的预期生成的。您正在查询三种不同的对象类型。当 ORM 从数据库取回行时，我没有找到将行映射到正确类的方法。

在这种情况下，UNION 没有多大意义，因为第三列在所有三个表中都有不同的含义。

您应该分别执行这三个查询，除非您的三种通知类型继承自公共 ORM 映射类。在这种情况下，SQLAlchemy 支持同时查询四种类型，尽管不使用 UNION。