rest_permissions.py 986 B

12345678910111213141516171819202122232425262728293031
  1. from rest_framework.permissions import BasePermission
  2. from django.core.exceptions import PermissionDenied
  3. from django.utils.translation import ugettext as _
  4. from misago.core.exceptions import Banned
  5. from misago.users.bans import get_request_ip_ban
  6. from misago.users.models import Ban
  7. class UnbannedOnly(BasePermission):
  8. def is_request_banned(self, request):
  9. ban = get_request_ip_ban(request)
  10. if ban:
  11. hydrated_ban = Ban(
  12. check_type=Ban.IP, user_message=ban['message'], expires_on=ban['expires_on']
  13. )
  14. raise Banned(hydrated_ban)
  15. def has_permission(self, request, view):
  16. self.is_request_banned(request)
  17. return True
  18. class UnbannedAnonOnly(UnbannedOnly):
  19. def has_permission(self, request, view):
  20. if request.user.is_authenticated:
  21. raise PermissionDenied(_("This action is not available to signed in users."))
  22. self.is_request_banned(request)
  23. return True