Bootstrap

Python-Flask实现关注,及朋友圈(互相关注)

 

一个用户可以有多个粉丝,Ta的粉丝也可以有多个粉丝。这是数据库的多对多模式,进一步讲,由于这两个"多"都是在同一张表中,因此,称这种关系为自引用关系。

通常,数据库的多对多其实就是两个一对多的关系,我们可以像往常实现一对多那样,在用户表下面添加关注者和被关注者字段,指向一张记录了关注者id和被关注者id的关系表。

Class User(db.Model):
   
    …
#被关注者
    followed = db.relationship('Follow',foreign_key=Follow.follower_id,backref=db.backref('follower',lazy='joined'),lazy='dynamic',cascade='all, delete-orphan')
#关注者
    follower = db.relationship('Follow',foreign_key=Follow.followed_id,backref=db.backref('followed',lazy='joined'),lazy='dynamic',cascade='all, delete-orphan')

字段中lazy=joined模式,可以直接从数据库中取出所有事例,无需一次次查询。

lazy=dynamic,返回的是一个query对象,可以进行后续过滤操作。

delete-orphan,可以监控删除情况,当一方删除后,这个指向也会被消除,all是为了删除没有指向的孤儿记录。

 

创建关系表Follow,记录关注者和被关注者的id。

Class Follow(db.Model):
   __tablename__= 'follow'
   follower_id = db.Column(db.Integer, db.ForeignKey('user.id'),primary_key=True)
   followed_id = db.Column(db.Integer, db.ForeignKey('user.id'),primary_key=True)
   addtime = db.Column(db.DATETIME, default=datetime.now)

实现关注:

def follow(username):
   followed = User.query.filter_by(username = username).first
   followed_id = followed.id
   follow = Follow()
   follow.follower_id = current_user.id
   follow.followed_id = followed_id
   db.session.add(follow)
   db.session.commit()
   

   

实现过程大致如此,把关注者的id和被关注者的id加入Follow表中。

 


查询互粉好友:

      如果我们需要实现朋友圈功能,即:只有相互关注才有访问权限。

      实现该功能需要三步:

  1. 先查询用户的关注列表
  2. 再查询用户的粉丝列表
  3. 找出交集,返回交集结果

1-查询关注列表:

followed = User.query.filter_by(username=username).first()
followed = follow.followed
list1 = []
for i in followed:
    list1.append(i.follower.id)

2-查询粉丝列表:

follower = User.query.filter_by(username=username).first
followers = follow.follower
list2 = []
for j in followers:
    list1.append(j.follower.id)

3-找出交集

friend_list = [x for x in list1 if x in list2]
        
u = Article.query.filter(Article.uuid.in_(friend_list))

return u

4-对交集进行过滤(如分页)

article = u.order_by(Article.addtime.desc()).paginate(
        page, per_page=15, error_out=False
    )

大功告成,这样只会显示只有互粉了,才会在朋友圈显示动态。

;