增加查询条件

This commit is contained in:
曹前明 2022-07-13 14:26:22 +08:00
parent c89a5b9023
commit b2780d5a3d
3 changed files with 4 additions and 2 deletions

View File

@ -24,6 +24,7 @@ class OperationViewSet(CustomModelViewSet):
create_serializer_class = OperationCreateUpdateSerializer
update_serializer_class = OperationCreateUpdateSerializer
serializer_class = OperationSerializer
filterset_fields = ['state', 'opl_operation__cate']
def update(self, request, *args, **kwargs):
obj = self.get_object()
@ -45,7 +46,7 @@ class OplViewSet(CustomModelViewSet):
update_serializer_class = OplCreateUpdateSerializer
serializer_class = OplSerializer
select_related_fields = ['cate']
filterset_fields = ['operation']
filterset_fields = ['operation', 'cate']
def destroy(self, request, *args, **kwargs):
obj = self.get_object()

View File

@ -111,7 +111,7 @@ class RpjViewSet(CustomModelViewSet):
update_serializer_class = RpjCreateUpdateSerializer
serializer_class = RpjListSerializer
select_related_fields = ['rparty', 'belong_dept']
filterset_fields = ['rparty', 'belong_dept']
filterset_fields = ['rparty', 'belong_dept', 'state']
def get_queryset(self):
queryset = super().get_queryset()

View File

@ -21,6 +21,7 @@ class VisitViewSet(CustomModelViewSet):
update_serializer_class = VisitCreateUpdateSerializer
serializer_class = VisitSerializer
retrieve_serializer_class = VisitDetailSerializer
filterset_fields = ['state']
def get_queryset(self):
user = self.request.user