views.py 580 B

12345678910111213141516
  1. # coding=utf-8
  2. from utils.permission import IsAdministrator
  3. from utils.custom_modelviewset import CustomModelViewSet
  4. from apps.staff.models import Department
  5. from apps.staff.serializers import DepartmentSerializer
  6. from apps.staff.filters import DepartmentFilter
  7. class DepartmentViewSet(CustomModelViewSet):
  8. permission_classes = [IsAdministrator, ]
  9. queryset = Department.objects.filter(delete=False)
  10. serializer_class = DepartmentSerializer
  11. def filter_queryset(self, queryset):
  12. f = DepartmentFilter(self.request.GET, queryset=queryset)
  13. return f.qs