SQLAlchemy:从查询

时间:2015-08-25 18:57:20

标签: python orm sqlalchemy

我的问题抽象,我有2 tables。用户表和友谊表。

我尝试进行查询以列出所有可用作朋友添加到User 1,Alice的用户,并使用SQLAlchemy排除自己。

考虑到可能会有很多友谊,找到爱丽丝的朋友:

friend_subquery = db.session.query(Friendship).filter_by(User_id=1).subquery()

现在我希望列出所有用户,除了Alice和她的朋友Bob和Jack。

friends = (db.session.query(User).
            filter(User.ID != 1).
            outerjoin(friend_subquery,
                      User.ID != friend_subquery.c.Friend_id))

我的预期结果是获得用户4和5,但这个查询 返回除Alice自己以外的所有人

的条件
User.ID != friend_subquery.c.Friend_id

似乎没有按预期工作。

P.S。我已完成了搜索,阅读文档的作业,但无法弄明白。谢谢你的时间。

1 个答案:

答案 0 :(得分:3)

我认为您的模型定义如下:

class User(db.Model):
    __tablename__ = 'User'

    ID = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100))

    friendships = db.relationship(
        'Friendship',
        foreign_keys='Friendship.User_id',
        backref='friender',
    )
    friendships_of = db.relationship(
        'Friendship',
        foreign_keys='Friendship.Friend_id',
        backref='friendee',
    )


class Friendship(db.Model):
    __tablename__ = 'Friendship'
    ID = db.Column(db.Integer, primary_key=True)
    User_id = db.Column(db.Integer, db.ForeignKey('User.ID'))
    Friend_id = db.Column(db.Integer, db.ForeignKey('User.ID'))

在下面的代码中显示了执行此查询的两种方法。第一个查询依赖于relationship User.friendships_of,而第二个查询使用显式联接:

    # Add users
    u1, u2, u3, u4, u5 = users = [
        User(name="Alice"),
        User(name="Bob"),
        User(name="Jack"),
        User(name="Pluto"),
        User(name="Mike"),
    ]
    db.session.add_all(users)

    # Add friendhips
    u1.friendships.append(Friendship(friendee=u2))
    u1.friendships.append(Friendship(friendee=u3))
    db.session.commit()

    # Find Alice
    u_alice = db.session.query(User).filter(User.name == 'Alice').one()

    # Query (version 1)
    q = (
        db.session.query(User)
        .filter(~User.friendships_of.any(Friendship.User_id == u_alice.ID))
        .filter(User.ID != u_alice.ID)
        .all()
    )
    for x in q:
        print(x)

    # Query (version 2)
    q = (
        db.session.query(User)
        .outerjoin(
            Friendship,
            db.and_(
                u_alice.ID == Friendship.User_id,
                User.ID == Friendship.Friend_id,
            )
        )
        .filter(Friendship.ID == None)
        .filter(User.ID != u_alice.ID)
        .all()
    )
    for x in q:
        print(x)