林轩 1 dia atrás
pai
commit
5c4deddb57
1 arquivos alterados com 3 adições e 0 exclusões
  1. 3 0
      apps/PestAnalysis/views.py

+ 3 - 0
apps/PestAnalysis/views.py

@@ -221,6 +221,7 @@ class PestCountAPIView(APIView):
         try:
             # regionCode=city_410100&page=1&pageSize=20&startDate=2025-12-04&endDate=2025-12-19
             data = request.query_params
+            pest = data.get("pest")
             code = data.get("regionCode") 
             code = code.split("_")[-1]
             start = data.get("startDate")
@@ -247,6 +248,8 @@ class PestCountAPIView(APIView):
                     start = int(datetime.datetime.strptime(start, "%Y-%m-%d").timestamp())
                     end = int(datetime.datetime.strptime(end, "%Y-%m-%d").timestamp())
                     query = query.filter(date__range=[start, end])
+                if pest:
+                    query = query.filter(pest__icontains=pest)
                 count = query.count()
                 query = query[(page-1)*pageSize:page*pageSize]
                 data = CbdAddrPestSerializer(query, many=True).data