合同增加筛选条件

hb_server/apps/sam/filters.py

@@ -9,3 +9,10 @@ class OrderFilterSet(filters.FilterSet):
     class Meta:
         model = Order
         fields = ['product', 'contract', 'customer', 'create_time_start', 'create_time_end']
+
+
+class ContractFilterSet(filters.FilterSet):
+    create_time_start = filters.DateFilter(field_name="create_time", lookup_expr='gte')
+    create_time_end = filters.NumberFilter(field_name="create_time", lookup_expr='lte')
+    class Meta:
+        fields = ['customer', 'create_time_start', 'create_time_end']

hb_server/apps/sam/views.py

@@ -5,7 +5,7 @@ from rest_framework.mixins import CreateModelMixin, DestroyModelMixin, ListModel
 from apps.mtm.models import Material
 from apps.inm.models import FIFO, FIFOItem, FIFOItemProduct, IProduct, WareHouse
 from apps.inm.signals import update_inm
-from apps.sam.filters import OrderFilterSet
+from apps.sam.filters import ContractFilterSet, OrderFilterSet
 from apps.sam.serializers import ContractCreateUpdateSerializer, ContractSerializer, CustomerCreateUpdateSerializer, CustomerSerializer, OrderCreateUpdateSerializer, OrderSerializer, SaleCreateSerializer, SaleListSerializer, SaleProductCreateSerializer, SaleProductListSerializer
 from apps.sam.models import Contract, Customer, Order, Sale, SaleProduct
 from rest_framework.viewsets import GenericViewSet, ModelViewSet
@@ -44,7 +44,7 @@ class ContractViewSet(CreateUpdateCustomMixin, ModelViewSet):
     queryset = Contract.objects.select_related('customer').all()
     serializer_class = ContractSerializer
     search_fields = ['name']
-    filterset_fields = []
+    filterset_class = ContractFilterSet
     ordering_fields = ['create_time']
     ordering = ['-create_time']