drf ModelViewSet之匹配路由參數


urls.py

    url(r'employees/(?P<employee_id>\d+)/detail/', EmployeesViewSet.as_view({'get': 'retrieve'})),

views.py

class EmployeesViewSet(DestroyModelMixin, CreateModelMixin, ListModelMixin, UpdateModelMixin, GenericViewSet):
    serializer_class = EmployeesSerializer
    # queryset = models.Employees.objects.filter(tags=0).all().order_by("-create_time")
    permission_classes = [IsAuthenticated]
    ordering_fields = ('name', 'department', 'add_people')
    lookup_url_kwarg = 'employee_id'
    filterset_class = EmployeesManagerFilter
    # filter_backends = [filters.DjangoFilterBackend]
    # search_fields = ['$name', '$email']

    def get_queryset(self, *args, **kwargs):
        # department_id = self.request.query_params.get("department_id", "")
        #
        # if department_id:
        #     return models.Department.objects.get(id=department_id).linkdepartments.filter(tags=0).all()

        return models.Employees.objects.filter(tags=0).order_by("-create_time")

    def get_serializer_class(self):
        # print(self.action)
        if self.action == 'add':
            return EmployeesCreateSerializer
        elif self.action == 'partial_update':
            return EmployeesUpdateSerializer
        elif self.action in ['add_invite', 'invite']:
            return EmployeesInviteSerializer

        return EmployeesSerializer

其中:

    views.py 中添加 lookup_url_kwarg = 'employee_id' 自動獲取路由中的參數 注意lookup_url_kwarg后面的參數名需要和路有中的無名分組的參數名一致才能
獲取到值。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM