switched my_company and my_products from method to class views
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user