Peewee ORM根据外键获得类似的条目

时间:2014-03-21 06:32:06

标签: python peewee

我在编写查询时遇到问题,因为基于他们拥有的标签在博客中获取类似帖子。我有以下模型:

class Articles(BaseModel):
    name = CharField()
      ...

class Tags(BaseModel):
    name = CharField()

class ArticleTags(BaseModel):
    article = ForeignKeyField(Articles, related_name = "articles")
    tags    = ForeignKeyField(Tags, related_name = "tags")

我想做的是获取具有相似标签的文章,按照常用标签的数量排序。

修改


经过2个小时的摆弄后,我得到了我正在寻找的anwser,我不确定它是否是最有效的方式,但是它正在工作:

如果将来有人可能需要这个功能,那么以下是该功能:

 def get_similar_articles(self,common_tags = 1, limit = 3):
        """
        Get 3 similar articles based on tag used
        Minimum 1 common tags i required
        """
        art = (ArticleTags.select(ArticleTags.tag)\
               .join(Articles)\
               .where(ArticleTags.article == self))

        return Articles.select(Articles, ArticleTags)\
               .join(ArticleTags)\
               .where((ArticleTags.article != self) & (ArticleTags.tag << art))\
               .group_by(Articles)\
               .having(fn.Count(ArticleTags.id) >= common_tags)\
               .order_by(fn.Count(Articles.id).desc())\
               .limit(limit)

1 个答案:

答案 0 :(得分:1)

只是一种风格,表名(和模型类)最好是单数。

# Articles tagged with 'tag1'
Articles.select().join(ArticleTags).join(Tags).where(Tags.name == 'tag1')