一个用户可以有多个粉丝,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-查询关注列表:
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
)
大功告成,这样只会显示只有互粉了,才会在朋友圈显示动态。