switched my_company and my_products from method to class views

This commit is contained in:
Sam
2021-03-08 12:24:44 +00:00
parent ebd1b6744c
commit ec27937b85
7 changed files with 48 additions and 48 deletions

View File

@@ -35,6 +35,19 @@ class IsStaff(permissions.BasePermission):
return request.user.is_staff
class IsSiteAdmin(permissions.BasePermission):
"""
Grant permission if request.user.role == 'SITE_ADMIN'
"""
admin_role = 'SITE_ADMIN'
def has_object_permission(self, request, view, obj):
return request.user.role == self.admin_role
def has_permission(self, request, view):
return request.user.role == self.admin_role
class ReadOnly(permissions.BasePermission):
def has_permission(self, request, view):