保存微博用户之间的关系:

import redis

class RelatoinShip(object):
    """使用无序集合键保存用户关系"""
    def __init__(self, client):
        self.client = client
        
    def follow(self, fans, target):
        """关注某人"""
        # 将用户添加到目标用户粉丝集合中
        target_fans_set = "weibo::user::" + str(target) + "::fans"
        self.client.sadd(target_fans_set, fans)
        # 将目标用户id添加到用户自身的关注集合中
        self_following_set = "weibo::user::" + str(fans) + "::following"
        self.client.sadd(self_following_set, target)
        
    def is_following(self, fans, target):
        """检查fans用户是否关注了target用户"""
        self_following_set = "weibo::user::" + str(fans) + "::following"
        return self.client.sismember(self_following_set, target)
        
    def is_following_each_other(self, user_a, user_b):
        """检查两个用户是否已互相关注"""
        return self.is_following(user_a, user_b) and self.is_following(user_b, user_a)
        
    def get_all_following(self, user):
        """返回用户关注的所有人"""
        self_following_set = "weibo::user::" + str(user) + "::following"
        return self.client.smembers(self_following_set)
        
    def get_all_fans(self, user):
        """返回用户的所有粉丝"""
        target_fans_set = "weibo::user::" + str(user) + "::fans"
        if not self.client.exists(target_fans_set):
            return False
        else:
            return self.client.smembers(target_fans_set)
        
    def common_following(self, user_a, user_b):
        """返回两个用户共同关注的人"""
        user_a_following = "weibo::user::" + str(user_a) + "::following"
        user_b_following = "weibo::user::" + str(user_b) + "::following"
        return self.client.sinter(user_a_following, user_b_following)


if __name__ == "__main__":
    redis_client = redis.StrictRedis()
    relation = RelatoinShip(redis_client)
    relation.follow(10086, 12345)
    relation.follow(10010, 12345)
    print(relation.is_following(10086, 12345))
    print(relation.get_all_following(10086))
    print(relation.get_all_fans(12345))
    print(relation.common_following(10086, 10010))

    关注集合,用于存储用户关注的人的ID,“weibo::user::::following”

    粉丝集合,用于存储用户的粉丝的ID,“weibo::user::::fans”

 

   我们在微博中看到喜欢的人可以点击关注,当然我们自己也可以被人关注,我们可以看到自己所有的粉丝,也可以看到自己关注的所有人,有显示是否已互相关注,微博也会显示自己和好友共同关注的人。

    在这个类中我们使用无序集合来保存用户的关系,集合的特性是去重,绝对没有重复值。无序集合这种数据结构特别适合保存不能有重复值,并且不在乎存储顺序的数据。就像我们在朋友圈点赞时,先点赞的人不一定排在前面,后点赞的人不一定排在后面显示,他们的显示顺序是无序的,但又是唯一的。后面我们在实现点赞(投票)功能时,也使用的无序集合这种数据结构来存储。

    现在可以思考一下如果要取消关注该怎么办呢!?