2015-12-16 171 views
0

以下是一个基本示例 - 帖子归用户所有并且很受用户欢迎。如何为特定用户选择喜欢的帖子?Peewee:通过多对多关系过滤

import datetime 
import peewee 

class User(peewee.Model): 
    name = peewee.CharField(max_length=200) 

class Post(peewee.Model): 
    user = peewee.ForeignKeyField(User) 
    text = peewee.TextField() 

class Like(peewee.Model): 
    user = peewee.ForeignKeyField(User) 
    post = peewee.ForeignKeyField(Post) 
    time = peewee.DateTimeField(default=datetime.datetime.now) 

我做一个查询从象模型的角度来看,用Like.select()开始查询,并加入Post模型。可以吗?我如何从Post模型的角度来查询,即Post.select()...

这里有一个舒适的完整示例:

def check_likes(): 
    user, _ = User.get_or_create(name="John") 
    post, _ = Post.get_or_create(text="Hello world!", user=user) 
    like, _ = Like.get_or_create(post=post, user=user) 

    print("Created: ", user, post, like) 

    # Get liked posts for user 
    query_likes = (Like.select() 
     .where(Like.user == user) 
     .join(User) 
     .join(Post)) 
    liked_posts = [like.post for like in query_likes] 

    print("Liked posts: ", liked_posts) 

if __name__ == '__main__': 
    User.create_table(True) 
    Post.create_table(True) 
    Like.create_table(True) 

    check_likes() 

UPD。我结束了这样的查询的时刻:

def liked_by(user_id): 
    """Active posts liked by user.""" 
    return (Post.select(Post, User, Like) 
     .join(User, on=(Post.user == User.id)) 
     .switch(Post) 
     .join(Like, peewee.JOIN.LEFT_OUTER, 
       on=(Like.post == Post.id).alias('like')) 
     .where(Like.user == user_id) 
     .order_by(Like.time.desc())) 

回答

1

上述查询应该工作,但你可能改一改:

Like.select(Like, Post)... 

您还需要考虑你是如何加入照顾表。使用 '开关' 的方法:

query_likes = (Like.select(Like, Post) # <<< Add like and post 
    .where(Like.user == user) 
    .join(User) 
    .switch(Like) # <<< You need this. 
    .join(Post)) 
liked_posts = [like.post for like in query_likes] 

你也可以这样做:

Post.select().join(Like).where(Like.user == some_user) 
+0

谢谢查尔斯!在更新后的问题中,我最后用'likes_by()'作为查询。不知道这里是否真的需要'JOIN.LEFT_OUTER'。 我还需要在我的真实代码中指定'on'语句,不确定什么时候需要它,但我使用UUID主键并且好像在某些情况下需要它,所以我决定将它添加到其他地方:) – rudyryk

+0

嗯..似乎在这里实际上不需要'JOIN.LEFT_OUTER'。在另一种情况下可能需要 - 在不过滤的情况下查询所有帖子+由特定用户加入喜欢。对? – rudyryk