def get_redis_one(to_user, from_user):
to_user_msg = REDIS_DB.get(to_user)
if to_user_msg:
to_user_msg = json.loads(to_user_msg)
count = to_user_msg.get(from_user, 0)
to_user_msg[from_user] = 0
REDIS_DB.set(to_user, json.dumps(to_user_msg))
return count
else:
return 0
def get_msg_one(sender, to_user):
user_msg_redis = REDIS_DB.get(to_user)
if user_msg_redis:
user_msg_dict = json.loads(user_msg_redis)
if user_msg_dict.get(sender):
# return user_msg_dict.get(sender)
user_msg_dict[sender] = 0
else:
user_msg_dict = {sender:0}
user_msg_redis = json.dumps(user_msg_dict)
REDIS_DB.set(to_user,user_msg_redis)
def set_redis(to_user, from_user):
# toy:{app:2}
to_user_msg = REDIS_DB.get(to_user)
if to_user_msg:
to_user_msg = json.loads(to_user_msg)
if to_user_msg.get(from_user):
to_user_msg[from_user] += 1
else:
to_user_msg[from_user] = 1
else:
to_user_msg = {from_user: 1}
REDIS_DB.set(to_user, json.dumps(to_user_msg))
def get_msg_list(user):
user_msg_redis = REDIS_DB.get(user)
if user_msg_redis:
user_msg_dict = json.loads(user_msg_redis)
user_msg_dict["count"] = sum(user_msg_dict.values())
else:
user_msg_dict = {"count":0}
return user_msg_dict
def get_redis_all(to_user):
to_user_msg = REDIS_DB.get(to_user)
if to_user_msg:
to_user_msg = json.loads(to_user_msg)
# {"toy": 1, "toy2": 1, "toy3": 2}
to_user_msg["count"] = sum(to_user_msg.values())
return to_user_msg
else:
return {"count": 0}
def save_msg(sender, to_user):
# 1.查询一下子龙的Redis是否有数据
user_msg_redis = REDIS_DB.get(to_user)
if user_msg_redis:
# 2.将子龙的数据反序列化成字典 { sender : n }
user_msg_dict = json.loads(user_msg_redis)
# 3.判断有没有 sender 的用户发来的消息数量
if user_msg_dict.get(sender):
user_msg_dict[sender] += 1
else:
user_msg_dict[sender] = 1
# 4.如果子龙是刚建立好的用户，他是没有消息的，字典是空
else:
user_msg_dict = {sender: 1}
# 5.序列化用户消息字典user_msg_dict
user_msg_redis = json.dumps(user_msg_dict)
# 6.存回Redis
REDIS_DB.set(to_user, user_msg_redis)
def get_user_msg_one(sender, to_user):
user_msg_redis = REDIS_DB.get(to_user)
if user_msg_redis:
user_msg_dict = json.loads(user_msg_redis)
if user_msg_dict.get(sender):
return user_msg_dict.get(sender)