develop 增加测试接口

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()
 

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):