drf视图组件

from rest_framework.permissions import BasePermission
class UserPermission(BasePermission):
message = '不是超级用户，查看不了'
def has_permission(self, request, view):
# user_type = request.user.get_user_type_display()
# if user_type == '超级用户':
user_type = request.user.user_type
print(user_type)
if user_type == 1:
return True
else:
return False
class Course(APIView):
authentication_classes = [TokenAuth, ]
permission_classes = [UserPermission,]

def get(self, request):
return HttpResponse('get')

def post(self, request):
return HttpResponse('post')