ERP5/ERP_5/apps/basic_info/views/customer_views.py

from django.db.models import Q
from drf_yasg.utils import swagger_auto_schema
from rest_framework.decorators import action
from rest_framework.viewsets import ModelViewSet

from ERP_5.utils.base_views import MultipleDestroyMixin, MultipleOpenMixin
from ERP_5.utils.paginations import GlobalPagination
from basic_info.models import CustomerModel
from basic_info.serializer.customer_serializer import CustomerSerializer, CustomerFindSerializer


class CustomerViewSet(ModelViewSet, MultipleDestroyMixin, MultipleOpenMixin):
    """
     create:
     客户信息--新增

     客户信息新增, status: 201(成功), return: 新增客户信息信息

     destroy:
     客户信息--删除

     客户信息删除, status: 204(成功), return: None

     multiple_delete:
     客户信息--批量删除,必传参数：ids=[1,2,3,4...]

     客户信息批量删除, status: 204(成功), return: None

     multiple_open:
     客户信息--批量启用或者禁用,必传(json)参数：ids=[1,2,3,4...](列表中可以只有一个)，is_open=1/0

     {
         "ids":[1,2],
         "is_open":"0"
     }
     is_open=1表示禁用，is_open=0表示启用，
     客户信息批量启用或者禁用, status: 204(成功), return: None

     update:
     客户信息--修改,

     客户信息修改, status: 200(成功), return: 修改后的客户信息信息

     partial_update:
     客户信息--局部修改,可以传参任意属性的值，服务器会修改指定的属性值

     客户信息局部修改, status: 200(成功), return: 修改后的客户信息信息

     find:
     客户信息--获取分页列表，可选json参数:name(名称)，mobile(手机号码)，phone(联系电话)

     {
         "name":"长沙",
         "mobile":"186","phone":xxx
     }
     客户信息列表信息, status: 200(成功), return: 客户信息信息列表

     retrieve:
     查询某一个客户信息

     查询指定ID的客户信息, status: 200(成功), return: 用户客户信息
     """

    queryset = CustomerModel.objects.all()
    serializer_class = CustomerSerializer
    pagination_class = GlobalPagination

    def get_queryset(self):
        if self.action == 'find':  # 过滤查询
            # 获取请求中的数据
            name = self.request.data.get('name', None)
            phone = self.request.data.get('phone', None)
            mobile = self.request.data.get('mobile', None)
            query = Q()
            if name:
                query.add(Q(name__contains=name), 'AND')  # 多条件组合
            if phone:
                query.add(Q(phone__contains=phone), 'AND')
            if mobile:
                query.add(Q(mobile__contains=mobile), 'AND')
            return CustomerModel.objects.filter(query).all()
        else:
            return CustomerModel.objects.all()

    @swagger_auto_schema(method='POST', request_body=CustomerFindSerializer, operation_description="查询过滤")
    @action(methods=['POST'], detail=False)
    def find(self, request, *args, **kwargs):
        return super(CustomerViewSet, self).list(request, args, kwargs)