From 0a1b068ea1f3c25a5ee30b39d4b3b3bf4dc5fa3c Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Thu, 27 Oct 2022 10:07:09 +0800
Subject: [PATCH] =?UTF-8?q?develop=20=E5=A2=9E=E5=8A=A0=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/develop/serializers.py |  1 +
 apps/develop/views.py       | 22 ++++++++++++++--------
 2 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/apps/develop/serializers.py b/apps/develop/serializers.py
index f8d2f4bc..98fae652 100755
--- a/apps/develop/serializers.py
+++ b/apps/develop/serializers.py
@@ -21,6 +21,7 @@ class CleanDataSerializer(serializers.Serializer):
 
 
 class TestAlgoSerializer(serializers.Serializer):
+    codes = serializers.ListField(child=serializers.CharField(), label='需要识别的算法')
     face_img = serializers.CharField()
     global_img = serializers.CharField()
 
diff --git a/apps/develop/views.py b/apps/develop/views.py
index 7990e50e..637d4ecb 100755
--- a/apps/develop/views.py
+++ b/apps/develop/views.py
@@ -174,19 +174,25 @@ class TestViewSet(CustomGenericViewSet):
         create_remind(event, params={'area': '生产车间', 'employee': '访客石非凡', 'event': '未带安全帽'})
         return Response()
 
+    @action(methods=['post'], detail=False, serializer_class=Serializer)
+    def test_snap_only(self, request, pk=None):
+        """通道抓图
+
+        通道抓图
+        """
+        return Response(dhClient.snap(request.data['code']))
+
     @action(methods=['post'], detail=False, serializer_class=TestAlgoSerializer)
     def test_algo(self, request, pk=None):
-        """测试算法1
+        """测试算法
 
-        测试算法1
+        测试算法
         """
         from apps.ai.main import ai_analyse
-        face_img = dhClient.get_full_pic(path=request.data['face_img'])
-        global_img = dhClient.get_full_pic(path=request.data['global_img'])
-        ret = ai_analyse(codes=['helmet'],
-                         face_img=face_img,
-                         global_img=global_img)
-        return Response({'result': ret.keys(), 'face_img': face_img, 'global_img': global_img})
+        ret = ai_analyse(codes=request.data['codes'],
+                         face_img=request.data.get('face_img', None),
+                         global_img=request.data.get('global_img', None))
+        return Response(ret)
 
     @action(methods=['post'], detail=False, serializer_class=Serializer)
     def test_algo_full(self, request, pk=None):