分享程序员开发的那些事...
更新时间：2022-08-13 10:41:38
新建用户功能：
|
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
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.尝试登陆，如果登陆成功则返回用户信息
本文转自戴柏阳的博客博客51CTO博客，原文链接http://blog.51cto.com/daibaiyang119/1962658如需转载请自行联系原作者
daibaiyang119