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

from django.db.models import Q
from django.utils.decorators import method_decorator
from drf_yasg import openapi
from drf_yasg.utils import swagger_auto_schema
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 SupplierModel
from basic_info.serializer.supplier_serializer import SupplierSerializer, SupplierFindSerializer

param1 = openapi.Parameter(name='name', type=openapi.TYPE_STRING, description="供应商名称的关键字", in_=openapi.IN_QUERY)
param2 = openapi.Parameter(name='mobile', type=openapi.TYPE_STRING, description="供应商的手机号码", in_=openapi.IN_QUERY)
param3 = openapi.Parameter(name='phone', type=openapi.TYPE_STRING, description="供应商名称的联系电话", in_=openapi.IN_QUERY)


@method_decorator(name='list', decorator=swagger_auto_schema(
    manual_parameters=[param1, param2, param3],
    operation_description="供应商的搜索过滤")
                  )
class SupplierViewSet(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: 修改后的供应商信息信息

    list:
    供应商信息--搜索过滤的分页列表，可选json参数:name(名称)，mobile(手机号码)，phone(联系电话)

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

    retrieve:
    查询某一个供应商信息

    查询指定ID的供应商信息, status: 200(成功), return: 用户供应商信息
    """
    pagination_class = GlobalPagination
    queryset = SupplierModel.objects.all()
    serializer_class = SupplierSerializer

    def get_queryset(self):
        if self.action == 'list':  # 过滤查询
            # 获取请求中的数据
            name = self.request.query_params.get('name', None)
            phone = self.request.query_params.get('phone', None)
            mobile = self.request.query_params.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 SupplierModel.objects.filter(query).all()
        else:
            return SupplierModel.objects.all()