拍爱

retrieve.py 2.4KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. # -*- coding: utf-8 -*-
  2. from group.models import GroupInfo, GroupPhotoInfo, GroupUserInfo, PhotoCommentInfo, PhotoThumbUpInfo
  3. from utils.redis.connect import r
  4. from utils.redis.rgroup import set_group_photo_data, set_group_users_info
  5. from utils.redis.rkeys import (GROUP_LAST_PHOTO_PK, GROUP_PHOTO_WATCHER_SET, GROUP_USERS_APPLYING_SET,
  6. GROUP_USERS_DELETED_SET, GROUP_USERS_PASSED_SET, GROUP_USERS_REFUSED_SET)
  7. def retrieve_group_data():
  8. """ 群组数据 """
  9. groups = GroupInfo.objects.filter(status=True)
  10. for group in groups:
  11. # 群组照片
  12. set_group_photo_data(group.group_id)
  13. # 群组用户
  14. set_group_users_info(group)
  15. def retrieve_last_pk():
  16. """ 群组最后一张照片PK """
  17. groups = GroupInfo.objects.filter(status=True)
  18. for group in groups:
  19. group_photo = GroupPhotoInfo.objects.filter(group_id=group.group_id, status=True).last()
  20. r.set(GROUP_LAST_PHOTO_PK % group.group_id, group_photo and group_photo.pk or -1)
  21. def retrieve_group_user_status():
  22. """ 群组用户状态 """
  23. group_users = GroupUserInfo.objects.filter(status=True)
  24. for group_user in group_users:
  25. group_id, user_id = group_user.group_id, group_user.user_id
  26. if group_user.user_status == GroupUserInfo.PASSED:
  27. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  28. elif group_user.user_status == GroupUserInfo.DELETED:
  29. r.sadd(GROUP_USERS_DELETED_SET % group_id, user_id)
  30. elif group_user.user_status == GroupUserInfo.APPLYING:
  31. r.sadd(GROUP_USERS_APPLYING_SET % group_id, user_id)
  32. elif group_user.user_status == GroupUserInfo.REFUSED:
  33. r.sadd(GROUP_USERS_REFUSED_SET % group_id, user_id)
  34. def retrieve_group_photo_watchers():
  35. group_photos = GroupPhotoInfo.objects.filter(status=True)
  36. for group_photo in group_photos:
  37. photo_comments = PhotoCommentInfo.objects.filter(photo_id=group_photo.pk)
  38. for photo_comment in photo_comments:
  39. r.sadd(GROUP_PHOTO_WATCHER_SET % group_photo.pk, photo_comment.user_id)
  40. photo_thumbups = PhotoThumbUpInfo.objects.filter(photo_id=group_photo.pk)
  41. for photo_thumbup in photo_thumbups:
  42. r.sadd(GROUP_PHOTO_WATCHER_SET % group_photo.pk, photo_thumbup.user_id)
  43. def retrieve_redis_data():
  44. retrieve_group_data()
  45. retrieve_last_pk()
  46. retrieve_group_user_status()
  47. retrieve_group_photo_watchers()