You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

17 lines
460 B

5 years ago
  1. from rest_framework.permissions import BasePermission
  2. class IsOwnComment(BasePermission):
  3. @classmethod
  4. def has_object_permission(cls, request, view, obj):
  5. if request.user.is_superuser:
  6. return True
  7. return obj.user.id == request.user.id
  8. class IsStaff(BasePermission):
  9. def has_permission(self, request, view):
  10. if request.user.is_superuser or request.user.is_staff:
  11. return True
  12. return False