拍爱

rmessage.py 1.4KB

123456789101112131415161718192021222324252627282930313233
  1. # -*- coding: utf-8 -*-
  2. from utils.redis.connect import r
  3. from utils.redis.rkeys import SYSTEM_MESSAGE_DELETED_INFO, SYSTEM_MESSAGE_READ_INFO
  4. def set_system_message_read_info(user_id):
  5. """ 设置系统消息读取信息 """
  6. from message.models import SystemMessageReadInfo
  7. read_messages = SystemMessageReadInfo.objects.filter(user_id=user_id, status=True)
  8. read_message_ids = [msg.msg_id for msg in read_messages]
  9. r.setexjson(SYSTEM_MESSAGE_READ_INFO % user_id, r.REDIS_EXPIRED_ONE_MONTH, read_message_ids)
  10. return read_message_ids
  11. def get_system_message_read_info(user_id):
  12. """ 获取系统消息读取信息 """
  13. return r.getjson(SYSTEM_MESSAGE_READ_INFO % user_id, default='[]') or set_system_message_read_info(user_id)
  14. def set_system_message_delete_info(user_id):
  15. """ 设置系统消息删除信息 """
  16. from message.models import SystemMessageDeleteInfo
  17. deleted_messages = SystemMessageDeleteInfo.objects.filter(user_id=user_id, status=True)
  18. deleted_message_ids = [msg.msg_id for msg in deleted_messages]
  19. r.setexjson(SYSTEM_MESSAGE_DELETED_INFO % user_id, r.REDIS_EXPIRED_ONE_MONTH, deleted_message_ids)
  20. return deleted_message_ids
  21. def get_system_message_delete_info(user_id):
  22. """ 获取系统消息删除信息 """
  23. return r.getjson(SYSTEM_MESSAGE_DELETED_INFO % user_id, default='[]') or set_system_message_delete_info(user_id)