Merge branch 'feature/pres-3034' into feature/uat-tmp

周伟奇
Showing 3 changed files with 113 additions and 66 deletions
src/apps/doc/mixins.py
src/apps/doc/views.py
src/pos/views.py
--- a/src/apps/doc/mixins.py
View file @b27910f
+++ b/src/apps/doc/mixins.py
View file @b27910f
@@ -6,6 +6,7 @@ from .models import HILDoc, AFCDoc, AFCSEOCRResult, AFCOCRResult, HILSEOCRResult
 from . import consts
 from prese.compare import pre_compare, get_empty_result
 from common.mixins import LoggerMixin
+from settings import conf
 class MPOSHandler:
@@ -179,11 +180,18 @@ class PosHandler:
    @staticmethod
    def de_mortgage_ocr_process(img_base64):
-        result_obj = {}
+        result_obj = {
-        url = 'http://file-classification.situdata.com/bs/all'
+            'customerName': '',
-        json_data = {"file": img_base64, "classify": consts.MVC_CLASSIFY}
+            'application': '',
+            'deMortgageDate': ''
+        }
+        json_data = {"file": img_base64, "classify": consts.MVC_CLASSIFY, "version": "green"}
        try:
+            url = conf.OCR_URL_FOLDER
            response = requests.post(url, json=json_data)
+            if response.status_code != 200:
+                return result_obj
+            response.encoding = 'unicode_escape'
            ocr_res = response.json()
            results = ocr_res.get('results')
            if ocr_res.get('page', '') == 'VehicleRegArea':
@@ -199,10 +207,46 @@ class PosHandler:
                info = results.get('1.机动车所有人/身份证名称/号码', {})
                result_obj['customerName'] = info.get('words', '').split('/')[0]
        except Exception as e:
-            LoggerMixin.running_log.error('[PosHandler de_mortgage_ocr_process] [error={0}]'.format(e.format_exc()))
+            LoggerMixin.running_log.error("[PosHandler de_mortgage_ocr_process] error", exc_info=1)
-            result_obj = {
+        return result_obj
-                'customerName': '',
-                'application': '',
+    @staticmethod
-                'deMortgageDate': ''
+    def de_mortgage_ocr_process1(img_base64):
-            }
+        result_obj = {
+            'customerName': '',
+            'application': '',
+            'deMortgageDate': ''
+        }
+        url = conf.OCR_URL_FOLDER
+        json_data = {"file": img_base64, "classify": consts.MVC_CLASSIFY, "version": "green"}
+        try:
+            response = requests.post(url, json=json_data)
+            # unicode转中文
+            response.encoding = 'unicode_escape'
+            if response.status_code != 200:
+                LoggerMixin.exception_log.error(
+                    "[PosHandler de_mortgage_ocr_process1] request error, url: %s, response: %s",
+                    url, response.text)
+                return result_obj
+            result = response.json()
+            data = result.get('data', [])
+            for item in data:
+                ocr_res = item.get('data', {})
+                if ocr_res.get('page', '') == 'page_34':
+                    words_result = ocr_res.get('words_result', {})
+                    registration_bar = words_result.get('registration_bar', [])
+                    for registration in registration_bar:
+                        if registration.get('register_type', '') == '抵押登记':
+                            register_info = registration.get('register_info', {})
+                            result_obj['application'] = register_info.get('抵押权人姓名/名称', '')
+                        elif registration.get('register_type', '') == '解除抵押':
+                            register_info = registration.get('register_info', {})
+                            result_obj['deMortgageDate'] = register_info.get('解除抵押日期', '')
+                elif ocr_res.get('page', '') == 'page_12':
+                    words_result = ocr_res.get('words_result', {})
+                    for _, word_result in words_result.items():
+                        if word_result.get('chinese_key', '') == '1.机动车所有人/身份证名称/号码':
+                            result_obj['customerName'] = word_result.get('words', '').split('/')[0]
+        except Exception as e:
+            LoggerMixin.exception_log.error("[PosHandler de_mortgage_ocr_process1] error", exc_info=1)
        return result_obj
--- a/src/apps/doc/views.py
View file @b27910f
+++ b/src/apps/doc/views.py
View file @b27910f
@@ -804,8 +804,6 @@ class SECompareView(GenericView, PreSEHandler):
    # SE preSettlement
    @use_args(se_compare_args, location='data')
    def post(self, request, args):  # interface_report pos to ocr
-        # 模拟耗时操作
-        time.sleep(25)
        start_time = time.time()
        log_base = '[prese]'
        # 存库
--- a/src/pos/views.py
View file @b27910f
+++ b/src/pos/views.py
View file @b27910f
@@ -5,6 +5,7 @@ from apps.doc.views import CustomDecimal, CustomDate
 from common import response
 from apps.doc.mixins import PosHandler
 from common.tools.comparison import cp
 from rest_framework.permissions import IsAuthenticated
 from apps.account.authentication import OAuth2AuthenticationWithUser
@@ -36,18 +37,18 @@ class NSCInvoiceView(GenericView):
    @use_args(input_args, location='data')
    def post(self, request, args):  # interface_report mpos to ocr
        content = args.get('content', {})
-        invoice_code = content['invoiceCode']
+        invoice_code = content.get('invoiceCode', '')
-        invoice_number = content['invoiceNumber']
+        invoice_number = content.get('invoiceNumber', '')
-        issue_date = content['issueDate']
+        issue_date = content.get('issueDate', None)
-        buyer_name = content['buyerName']
+        buyer_name = content.get('buyerName', '')
-        buyer_id = content['buyerId']
+        buyer_id = content.get('buyerId', 0)
-        vin = content['vin']
+        vin = content.get('vin', '')
-        dealer = content['dealer']
+        dealer = content.get('dealer', '')
-        price_with_vat = content['priceWithVat']
+        price_with_vat = content.get('priceWithVat', 0.0)
-        price_no_vat = content['priceNoVat']
+        price_no_vat = content.get('priceNoVat', 0.0)
-        price_in_capitals = content['priceInCapitals']
+        price_in_capitals = content.get('priceInCapitals', '')
-        vat = content['vat']
+        vat = content.get('vat', 0.0)
-        vat_rate = content['vatRate']
+        vat_rate = content.get('vatRate', 0.0)
        return response.ok()
@@ -68,46 +69,50 @@ de_mortgage_comments = {
 # 解除抵押识别处理
-# class DeMortgageView(GenericView):
+class DeMortgageView(GenericView):
-#     @use_args(de_mortgage_args, location='data')
+    permission_classes = [IsAuthenticated]
-#     def post(self, request, args):  # interface_report mpos to ocr
+    authentication_classes = [OAuth2AuthenticationWithUser]
-#         img_files = args.get('file_base64', [])
-#         customer_name = args.get('customerName', '')
+    @use_args(de_mortgage_args, location='data')
-#         application = args.get('application', '')
+    def post(self, request, args):  # interface_report mpos to ocr
-#         de_mortgage_date = args.get('deMortgageDate')
+        img_files = args.get('file_base64', [])
-#
+        customer_name = args.get('customerName', '')
-#         fields_input = {
+        application = args.get('application', '')
-#             'customerName': customer_name,
+        de_mortgage_date = args.get('deMortgageDate')
-#             'application': application,
-#             'deMortgageDate': de_mortgage_date
+        fields_input = {
-#         }
+            'customerName': customer_name,
-#         de_mortgage_info = {}
+            'application': application,
-#         # 绿本必须分开ocr
+            'deMortgageDate': de_mortgage_date
-#         for img_file in img_files:
+        }
-#             info = PosHandler.de_mortgage_ocr_process(img_file)
+        de_mortgage_info = {}
-#             de_mortgage_info.update(info)
+        # 绿本必须分开ocr
-#
+        for img_file in img_files:
-#         request_pass = True
+            info = PosHandler.de_mortgage_ocr_process1(img_file)
-#         fields_result = []
+            de_mortgage_info.update(info)
-#         for field_name, input_val in enumerate(fields_input):
-#             field_result = {
+        request_pass = True
-#                 "name": field_name,
+        fields_result = []
-#                 "input": input_val,
+        for field_name, input_val in fields_input.items():
-#                 "ocr": de_mortgage_info.get(field_name, ''),
+            field_result = {
-#                 "field_is_pass": False,
+                "name": field_name,
-#                 "comments": ''
+                "input": input_val,
-#             }
+                "ocr": de_mortgage_info.get(field_name, ''),
-#             result, _ = cp.common_compare(field_result['input'], field_result['ocr'])
+                "field_is_pass": False,
-#             if result == cp.RESULT_Y:
+                "comments": ''
-#                 fields_result['field_is_pass'] = result
+            }
-#             else:
+            result, _ = cp.common_compare(field_result['input'], field_result['ocr'])
-#                 request_pass = False
+            if result == cp.RESULT_Y:
-#                 fields_result['comments'] = de_mortgage_comments.get(field_name, '')
+                fields_result['field_is_pass'] = result
-#
+            else:
-#             fields_result.append(field_result)
+                request_pass = False
-#
+                fields_result['comments'] = de_mortgage_comments.get(field_name, '')
-#         result = {
-#             "is_pass": request_pass,
+            fields_result.append(field_result)
-#             "fields": fields_result
-#         }
+        result = {
-#         return response.ok(data=result)
+            "is_pass": request_pass,
+            "fields": fields_result
+        }
+        return response.ok(data=result)