Hotfix before deadline

This commit is contained in:
ITQ
2024-03-04 23:21:51 +03:00
parent 9a4b18b4c1
commit 79f0aaaaa6
9 changed files with 40 additions and 29 deletions
+7 -4
View File
@@ -1,11 +1,14 @@
from rest_framework import status
from rest_framework.exceptions import APIException
from rest_framework.permissions import BasePermission
class CanAccessPost(BasePermission):
message = "You do not have permission to access this post."
class CustomForbidden(APIException):
status_code = status.HTTP_404_NOT_FOUND
default_detail = "You dont have access to view this post."
class CanAccessPost(BasePermission):
def has_object_permission(self, request, view, obj):
if (
obj.author.isPublic
@@ -14,7 +17,7 @@ class CanAccessPost(BasePermission):
):
return True
return False
raise CustomForbidden
class CanAccessFeed(BasePermission):
@@ -29,4 +32,4 @@ class CanAccessFeed(BasePermission):
):
return True
return False
raise CustomForbidden