通过SQLAlchemy选择多个多对多关系

时间:2013-07-17 06:47:06

标签: python sql sqlalchemy subquery

我希望能够在SQLAlchemy中查询多个多对多的关系。

我有与群组关联的用户,这些用户与角色相关联。所有协会都是多对多的。我想通过组获取与用户关联的角色列表。

以下是我的条纹模型:

class User(Model):
    id = Column(Integer)
    groups = relationship('Group', secondary=user_group)

class Group(Model):
    id = Column(Integer)
    roles = relationship('Role', secondary=role_group)

class Role(Model):
    id = Column(Integer)

我大致了解将使用什么SQL:

select distinct role.* from role, role_group
where role.id = role_group.role_id
  and role_group.group_id in 
    (select "group".id from "group", user_group 
     WHERE user_group.user_id = 1 
       and "group".id = user_group."group.id")

但是我很难弄清楚如何将其翻译成SQLAlchemy ......而且我不确定即使这里的SQL是最好的方法也不行。

2 个答案:

答案 0 :(得分:3)

试试这个

s = Session()
s.query(Role).join(User.groups).join(Group.roles).filter(User.id==1).all()

它将生成以下sql查询

SELECT role.id AS role_id 
FROM user
JOIN user_group AS user_group_1 ON user.id = user_group_1.user_id
JOIN "group" ON "group".id = user_group_1.group_id
JOIN role_group AS role_group_1 ON "group".id = role_group_1.group_id
JOIN role ON role.id = role_group_1.role_id 
WHERE user.id = 1;

然而sqlalchemy只会返回不同的角色(如果你要查询Role.id,那么sqlalchemy将返回sq查询实际返回的内容)。

答案 1 :(得分:0)

我确实找到了一个我自己的问题的解决方案,它将生成接近我正在寻找的SQL。但是,我相信zaquest的答案要好得多,而且会使用它。这只是为了后人的缘故。

以下是我提出的查询:

s = Session()
s.query(Role).join(Role.groups).filter(
        Group.id.in_(s.query(Group.id).join(User.groups)
                .filter(User.id == self.id)))

这将生成以下SQL

SELECT role.id AS role_id
FROM role JOIN role_group AS role_group_1 ON role.id = role_group_1.role_id 
JOIN "group" ON "group".id = role_group_1.group_id 
WHERE "group".id IN 
  (SELECT "group".id AS group_id 
   FROM "user" JOIN user_group AS user_group_1 ON "user".id = user_group_1.user_id 
   JOIN "group" ON "group".id = user_group_1."group.id" 
   WHERE "user".id = :id_1)

这与我正在寻找的SQL类似。我忘记的关键是我可以使用另一个查询作为SQLAlchemy中in运算符的一部分。