今天我们开始用Redis数据库实现类似微博的后台业务逻辑,我会带领大家一步步的实现类似微博的各种功能效果。这里我所使用的环境是Python3.5+Redis3.2.1,Python2和Python3有很多命令不相同,请大家注意。
检查用户名和邮箱名的唯一性:
import redis
class UniqueSet(object):
"""使用集合键保存用户名和邮箱地址,检查是否已经使用"""
def __init__(self, key, client):
self.key = key
self.client = client
def is_include(self, element):
"""检查元素是否存在集合内"""
return self.client.sismember(self.key, element)
def add(self, element):
"""添加元素到集合中"""
self.client.sadd(self.key, element)
if __name__ == "__main__":
redis_client = redis.StrictRedis()
used_emails = UniqueSet("weibo::used_emails", redis_client)
print(used_emails.is_include("123@qq.com"))
used_emails.add("123@qq.com")
在redis中我们使用"weibo::used_names"和"weibo::used_emails"两个集合来存储所有已经被使用的的名字和邮箱,我们登录微博时一般通过用户名或者邮箱登录,所以这里我们实现的效果就是:
1. 检查用户名或者邮箱名是否已经被使用;
2. 新用户注册成功后,将用户名或者邮箱名添加到集合中去;
3. 这里的集合使用的是无序集合,这个类我们将在后面新建用户时使用到;