|
|
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()
|
