views.py
1.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# _*_ coding: utf-8 _*_
# @Time : 2020/8/31 20:17
# @Author vanwhebin
from .models import User
from rest_framework import viewsets
from usercenter.serializers import UserSerializer, GroupSerializer, PermissionSerializer, MyTokenObtainPairSerializer
from rest_framework.views import APIView
from django.contrib.auth import authenticate
from utils.util import response
from rest_framework.response import Response
from rest_framework import status
from rest_framework.permissions import IsAdminUser
from django.contrib.auth.models import Group, Permission
from rest_framework_simplejwt.views import TokenObtainPairView
class MyTokenObtainPairView(TokenObtainPairView):
serializer_class = MyTokenObtainPairSerializer
class UserViewSet(viewsets.ModelViewSet):
queryset = User.objects.all()
serializer_class = UserSerializer
permission_classes = (IsAdminUser, )
class GroupViewSet(viewsets.ModelViewSet):
queryset = Group.objects.all()
serializer_class = GroupSerializer
permission_classes = (IsAdminUser,)
class PermissionViewSet(viewsets.ModelViewSet):
queryset = Permission.objects.all()
serializer_class = PermissionSerializer
permission_classes = (IsAdminUser,)