新建用户功能：
|
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
