Merge branch 'CHINARPA-4562'

@@ -4,5 +4,6 @@ from . import views
 
 
 urlpatterns = [
 urlpatterns = [
     path(r'', views.DocView.as_view()),
     path(r'', views.DocView.as_view()),
+    path(r'query/employee', views.EmployeeView.as_view()),
     path(r'contract/v1', views.SEContractView.as_view()),
     path(r'contract/v1', views.SEContractView.as_view()),
 ]
 ]

@@ -90,6 +90,10 @@ class CustomDecimal(fields.Decimal):
 def load_data(request, schema):
 def load_data(request, schema):
     return request.data
     return request.data
 
 
+employee_args = {
+    'application_id': fields.Str(required=True, validate=validate.Length(max=64)),
+    'business_type': fields.Str(required=True, validate=validate.Length(max=64)),
+}
 
 
 go_args = {
 go_args = {
     'image': fields.Raw(required=True),
     'image': fields.Raw(required=True),
@@ -1845,3 +1849,29 @@ class GoView(GenericView):
                 return response.ok(data=result)
                 return response.ok(data=result)
             else:
             else:
                 return response.error_msg(msg='识别错误')
                 return response.error_msg(msg='识别错误')
+
+class EmployeeView(GenericView):
+    permission_classes = [IsAuthenticated]
+    authentication_classes = [OAuth2AuthenticationWithUser]
+
+    @use_args(employee_args, location='data')
+    def post(self, request, args): 
+        
+        application_id = args.get('application_id')
+        business_type = args.get('business_type')
+        
+        ocr_result_class = HILOCRResult if business_type in consts.HIL_SET else AFCOCRResult
+        ocr_result_info = ocr_result_class.objects.filter(application_id=application_id).first()
+        self.running_log.info('[query Employee] [application_id={0}] [business_type={1}] [ocr_result_info exist={2}]'.format(application_id, business_type, ocr_result_info is not None))
+        if not ocr_result_info:
+            self.running_log.info('[query Employee] [application_id={0}] ocr_result none'.format(application_id))
+            return response.ok(data=False)
+        bss_ocr_str = ocr_result_info.bss_ocr
+        bss_ocr = json.loads(bss_ocr_str)
+        self.running_log.info('[query Employee] [application_id={0}] [bss_ocr={1}]'.format(application_id, bss_ocr))
+        for one_bss in bss_ocr:
+            income_keywords = one_bss.get('income_keywords')
+            self.running_log.info('[query Employee] [application_id={0}] [income_keywords={1}]'.format(application_id, income_keywords))
+            if income_keywords is not None and len(income_keywords) > 0:
+                return response.ok(data=True)
+        return response.ok(data=False)
\ No newline at end of file