yf_yzl пре 2 година
родитељ
комит
63f0f8c0d7
1 измењених фајлова са 9 додато и 6 уклоњено
  1. 9 6
      smartfarming/views/counts_views.py

+ 9 - 6
smartfarming/views/counts_views.py

@@ -190,13 +190,16 @@ class RecentPestCountAPIView(APIView):
                 tp_lst = i.split("#")
                 for k in tp_lst:
                     pest = k.split(",")
-                    p_0 = pest[0]
-                    p_1 = pest[1]
-                    p_name = insect_dict.get(p_0)
-                    if p_name not in pest_count.keys():
-                        pest_count[p_name] = int(p_1)
+                    if len(pest) == 2:
+                        p_0 = pest[0]
+                        p_1 = pest[1]
+                        p_name = insect_dict.get(p_0)
+                        if p_name not in pest_count.keys():
+                            pest_count[p_name] = int(p_1)
+                        else:
+                            pest_count[p_name] += int(p_1)
                     else:
-                        pest_count[p_name] += int(p_1)
+                        logger.error(f"害虫识别格式有误:{k}")
             result = sorted(pest_count.items(), key=itemgetter(1), reverse=False)
             result = {m[0]:m[1] for m in result[-10:]}
             return Response({"code": 0, "msg": "success", "data": result})