2
2
from rest_framework .viewsets import ModelViewSet
3
3
4
4
from .models import Class , Student , Teacher
5
- from .permissions import IsAdminOrReadOnly , IsSuperUserOrReadOnly , IsOwnerOrReadOnly
5
+ from .permissions import IsSuperUserOrReadOnly , IsOwnerOrReadOnly
6
6
from .serializers import StudentSerializer , ClassSerializer , TeacherSerializer
7
7
8
8
9
- class IsAdminViewSet (ModelViewSet ):
9
+ class ProtectedViewSet (ModelViewSet ):
10
10
permission_classes = (
11
- IsAdminOrReadOnly ,
11
+ # IsAdminOrReadOnly,
12
+
12
13
)
13
14
pagination_class = LimitOffsetPagination
15
+ page_size_query_param = 'page'
16
+ page_size = 1
14
17
15
18
16
19
class IsSuperUserViewSet (ModelViewSet ):
@@ -27,7 +30,7 @@ class IsOwnerViewSet(ModelViewSet):
27
30
pagination_class = LimitOffsetPagination
28
31
29
32
30
- class StudentsViewSet (ModelViewSet ):
33
+ class StudentsViewSet (ProtectedViewSet ):
31
34
serializer_class = StudentSerializer
32
35
33
36
def get_queryset (self ):
@@ -37,7 +40,7 @@ def get_queryset(self):
37
40
return Student .objects .select_related ('_class' ).get (pk = pk )
38
41
39
42
40
- class ClassViewSet (ModelViewSet ):
43
+ class ClassViewSet (ProtectedViewSet ):
41
44
serializer_class = ClassSerializer
42
45
43
46
def get_queryset (self ):
@@ -47,7 +50,7 @@ def get_queryset(self):
47
50
return Class .objects .filter (pk = pk )
48
51
49
52
50
- class TeacherViewSet (ModelViewSet ):
53
+ class TeacherViewSet (ProtectedViewSet ):
51
54
serializer_class = TeacherSerializer
52
55
53
56
def get_queryset (self ):
0 commit comments