djnago ModelViewSet

class StaffViewSet(viewsets.ModelViewSet):
    '''
    	由于ModelViewSet 本身就封装了 GET PUT POST PATCH 如无特殊处理可以直接使用
    	因为下面使用了一对一 和 多对多 所以重写了 create update destroy
    '''
    serializer_class = StaffSerializer
    queryset = Staff.objects.all()
    # filter_backends 封装了djnago filter 根据字段精确查找和根据字段模糊搜索
    filter_backends = (DjangoFilterBackend,filters.SearchFilter)
  
    filterset_fields = ['user__realname','employee_number','user__mobile','job_title']
    search_fields = ['user__realname','employee_number','user__mobile','job_title']
		
    '''
    	post
    '''
    def create(self, request, *args, **kwargs):
        department_id = request.data.pop('department_id')
        manage = request.data.pop('manage')
        user_data = request.data.pop('user')
        u_ser = UserSerializer(data=user_data)
        u_ser.is_valid(raise_exception=True)
        user = User.objects.create(**u_ser.validated_data)
        serializer = StaffSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.instance = Staff.objects.create(user=user, **serializer.validated_data)
        department = Department.objects.filter(pk=department_id).first()
        StaffDepartment.objects.create(staff=serializer.instance,department=department,manage=manage)
        headers = super().get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
		'''put'''
    def update(self, request, *args, **kwargs):
        partial = kwargs.get('partial', False)
        instance = self.get_object()
        user_data = request.data.pop('user', None)
        department_id = request.data.pop('department_id')
        manage = request.data.pop('manage')
        StaffDepartment.objects.filter(staff=instance).update(department=Department.objects.filter(pk=department_id).first(),manage=manage)
        if user_data:
            user_ser = UserSerializer(instance.user, data=user_data, partial=partial)
            user_ser.is_valid(raise_exception=True)
            user_ser.save()
        return super().update(request, args, kwargs)    

		'''
			delete
		'''
    def destroy(self, request, *args, **kwargs):
        instance = self.get_object()
        user = instance.user
        user.delete()
        return Response(status=status.HTTP_204_NO_CONTENT)

urls.py

from rest_framework import routers


router = routers.DefaultRouter()
router.register('profile', ProfilesView)


path('api-auth/', include('rest_framework.urls')),
path('', include(router.urls)),

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值