Redis实现微博后台业务逻辑系列（三）

423212

新建用户功能：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48

import redis class User(object):     """使用Redis散列键保存用户信息，并尝试登陆"""     def __init__(self, client):         self.client = client         self.key = "weibo::email_to_uid"           def create(self, name, passwd, email):         """创建用户"""         # 生成新的ID         new_id = IdGenerator("weibo::uid", self.client).gen()         # 生成新用户的键         user_key = "weibo::user::" + str(new_id)         # 创建用户，使用散列键保存用户信息，使用hmset时需要将key、value写成字典格式传递给redis         self.client.hmset(user_key, {"id": new_id, "name": name, "passwd": passwd, "email": email})         # 使用散列键保存email和ID的对应关系         self.client.hset(self.key, email, new_id)         return new_id.decode()     def get_by_id(self, id):         """根据用户ID返回用户信息"""         user_key = "weibo::user::" + str(id)         return self.client.hgetall(user_key)               def try_login(self, email, passwd):         """尝试登陆，返回用户信息"""         uid = self.client.hget(self.key, email)         if uid is None:             return None                   # 获取用户信息，并将字典内二进制数据转换成字符串数据并重新存储         user_info = self.get_by_id(uid.decode())         user = dict()         for key,value in user_info.items():             user[key.decode()] = value.decode()                   if user["passwd"] == passwd:             return user              if __name__ == "__main__":     redis_client = redis.StrictRedis()     user = User(redis_client)     uid = user.create("daiby", "abcd1234", "123@qq.com")     print(uid)     user_info = user.try_login("123@qq.com", "abcd1234")     print(user_info)

  在这个类里，我们使用Redis散列键的数据结构来存储用户账户信息。为什么要用散列键来存储呢？Redis散列键的结构是：hmset key field value field value ...，这样我们使用"weibo::user::<id>"这个散列键key来保存用户的ID、邮箱、姓名、密码等信息，再使用"weibo::email_to_uid"这个散列键来保存email和id之间的关联关系，为什么要保存email和id之间的关联关系呢？因为我们使用email名登陆时，通过"weibo::email_to_uid"这个散列键，取得对应的ID了。类中我们实现了三个方法：
    1.创建新用户
    2.根据用户ID返回用户信息

    3.尝试登陆，如果登陆成功则返回用户信息