auth.py 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. from rest_framework import serializers
  2. from django.contrib.auth import get_user_model
  3. from django.urls import reverse
  4. from misago.acl import serialize_acl
  5. from .user import UserSerializer
  6. UserModel = get_user_model()
  7. __all__ = [
  8. 'AuthenticatedUserSerializer',
  9. 'AnonymousUserSerializer',
  10. ]
  11. class AuthFlags(object):
  12. def get_is_authenticated(self, obj):
  13. return bool(obj.is_authenticated)
  14. def get_is_anonymous(self, obj):
  15. return bool(obj.is_anonymous)
  16. class AuthenticatedUserSerializer(UserSerializer, AuthFlags):
  17. email = serializers.SerializerMethodField()
  18. is_authenticated = serializers.SerializerMethodField()
  19. is_anonymous = serializers.SerializerMethodField()
  20. class Meta:
  21. model = UserModel
  22. fields = UserSerializer.Meta.fields + (
  23. 'is_hiding_presence',
  24. 'limits_private_thread_invites_to',
  25. 'subscribe_to_started_threads',
  26. 'subscribe_to_replied_threads',
  27. 'is_authenticated',
  28. 'is_anonymous',
  29. )
  30. def get_acl(self, obj):
  31. return serialize_acl(obj)
  32. def get_email(self, obj):
  33. return obj.email
  34. def get_api_url(self, obj):
  35. return {
  36. 'avatar': reverse(
  37. 'misago:api:user-avatar', kwargs={'pk': obj.pk}),
  38. 'options': reverse(
  39. 'misago:api:user-forum-options', kwargs={'pk': obj.pk}),
  40. 'username': reverse(
  41. 'misago:api:user-username', kwargs={'pk': obj.pk}),
  42. 'change_email': reverse(
  43. 'misago:api:user-change-email', kwargs={'pk': obj.pk}),
  44. 'change_password': reverse(
  45. 'misago:api:user-change-password', kwargs={'pk': obj.pk}),
  46. }
  47. AuthenticatedUserSerializer = AuthenticatedUserSerializer.exclude_fields(
  48. 'is_avatar_locked', 'is_blocked', 'is_followed', 'is_signature_locked',
  49. 'meta', 'signature', 'status',
  50. )
  51. class AnonymousUserSerializer(serializers.Serializer, AuthFlags):
  52. id = serializers.ReadOnlyField()
  53. acl = serializers.SerializerMethodField()
  54. is_authenticated = serializers.SerializerMethodField()
  55. is_anonymous = serializers.SerializerMethodField()
  56. def get_acl(self, obj):
  57. if hasattr(obj, 'acl_cache'):
  58. return serialize_acl(obj)
  59. else:
  60. return {}