From 705ed09307452a2f929de7010e1522d45ef582e6 Mon Sep 17 00:00:00 2001 From: xianfuxing Date: Mon, 20 Aug 2018 16:16:45 +0800 Subject: [PATCH] # add search for device_id --- apps/counter/api/views.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/apps/counter/api/views.py b/apps/counter/api/views.py index 6c572f5..f2ae239 100644 --- a/apps/counter/api/views.py +++ b/apps/counter/api/views.py @@ -28,7 +28,13 @@ class DeviceLogListAPIView(ListAPIView): filter_backends = [SearchFilter, OrderingFilter] pagination_class = PostPageNumberPagination search_fields = ['device_id'] - queryset = DeviceCount.objects.get_queryset().order_by('-data_time') + + def get_queryset(self, *args, **kwargs): + queryset_list = DeviceCount.objects.get_queryset().order_by('-data_time') + device_id = self.request.GET.get('device_id') + if device_id: + queryset_list = queryset_list.filter(device_id__icontains=device_id) + return queryset_list class DeviceInfoListAPIView(ListAPIView): @@ -37,7 +43,6 @@ class DeviceInfoListAPIView(ListAPIView): filter_backends = [SearchFilter, OrderingFilter] pagination_class = PostPageNumberPagination search_fields = ['device_id', 'online'] - queryset = DeviceInfo.objects.all() def get_queryset(self, *args, **kwargs): queryset_list = DeviceInfo.objects.get_queryset().order_by('device_id')