Merge branch 'master' into feature/pres

@@ -20,7 +20,7 @@ from common.tools.pdf_to_img import PDFHandler
 from apps.doc import consts
 from apps.doc import consts
 from apps.doc.exceptions import OCR1Exception, OCR2Exception, LTGTException
 from apps.doc.exceptions import OCR1Exception, OCR2Exception, LTGTException
 from apps.doc.ocr.wb import BSWorkbook
 from apps.doc.ocr.wb import BSWorkbook
-from apps.doc.models import OfflineReport
+from apps.doc.models import OfflineReport, AFCOfflineReport
 from apps.doc.named_enum import OfflineFailureReason
 from apps.doc.named_enum import OfflineFailureReason
 
 
 
 
@@ -447,7 +447,8 @@ class Command(BaseCommand, LoggerMixin):
                 finally:
                 finally:
                     end_time = time.time()
                     end_time = time.time()
                     try:
                     try:
-                        OfflineReport.objects.create(
+                        report_table = OfflineReport if input_dir.find(consts.HIL_PREFIX) != -1 else AFCOfflineReport
+                        report_table.objects.create(
                             input_folder=input_dir,
                             input_folder=input_dir,
                             doc_type=consts.DDA_CLASSIFY,
                             doc_type=consts.DDA_CLASSIFY,
                             file_name=name,
                             file_name=name,

@@ -21,7 +21,7 @@ from common.electronic_afc_contract.afc_contract_ocr import predict as afc_predi
 from apps.doc import consts
 from apps.doc import consts
 from apps.doc.exceptions import OCR1Exception, OCR2Exception, LTGTException
 from apps.doc.exceptions import OCR1Exception, OCR2Exception, LTGTException
 from apps.doc.ocr.wb import BSWorkbook
 from apps.doc.ocr.wb import BSWorkbook
-from apps.doc.models import OfflineReport
+from apps.doc.models import OfflineReport, AFCOfflineReport
 from apps.doc.named_enum import OfflineFailureReason
 from apps.doc.named_enum import OfflineFailureReason
 
 
 
 
@@ -769,7 +769,8 @@ class Command(BaseCommand, LoggerMixin):
                 finally:
                 finally:
                     end_time = time.time()
                     end_time = time.time()
                     try:
                     try:
-                        OfflineReport.objects.create(
+                        report_table = OfflineReport if input_dir.find(consts.HIL_PREFIX) != -1 else AFCOfflineReport
+                        report_table.objects.create(
                             input_folder=input_dir,
                             input_folder=input_dir,
                             doc_type=classify,
                             doc_type=classify,
                             file_name=name,
                             file_name=name,

@@ -18,7 +18,7 @@ from common.tools.pdf_to_img import PDFHandler
 from apps.doc import consts
 from apps.doc import consts
 from apps.doc.exceptions import OCR1Exception, OCR4Exception
 from apps.doc.exceptions import OCR1Exception, OCR4Exception
 from apps.doc.ocr.wb import BSWorkbook
 from apps.doc.ocr.wb import BSWorkbook
-from apps.doc.models import OfflineReport
+from apps.doc.models import OfflineReport, AFCOfflineReport
 from apps.doc.named_enum import OfflineFailureReason
 from apps.doc.named_enum import OfflineFailureReason
 
 
 
 
@@ -385,7 +385,8 @@ class Command(BaseCommand, LoggerMixin):
                 finally:
                 finally:
                     end_time = time.time()
                     end_time = time.time()
                     try:
                     try:
-                        OfflineReport.objects.create(
+                        report_table = OfflineReport if input_dir.find(consts.HIL_PREFIX) != -1 else AFCOfflineReport
+                        report_table.objects.create(
                             input_folder=input_dir,
                             input_folder=input_dir,
                             doc_type=classify,
                             doc_type=classify,
                             file_name=name,
                             file_name=name,

@@ -21,7 +21,7 @@ from common.tools.pdf_to_img import PDFHandler
 from apps.doc import consts
 from apps.doc import consts
 from apps.doc.exceptions import OCR1Exception, OCR4Exception
 from apps.doc.exceptions import OCR1Exception, OCR4Exception
 from apps.doc.ocr.wb import BSWorkbook, PatternFill
 from apps.doc.ocr.wb import BSWorkbook, PatternFill
-from apps.doc.models import OfflineReport
+from apps.doc.models import OfflineReport, AFCOfflineReport
 from apps.doc.named_enum import OfflineFailureReason
 from apps.doc.named_enum import OfflineFailureReason
 
 
 
 
@@ -632,7 +632,8 @@ class Command(BaseCommand, LoggerMixin):
                 finally:
                 finally:
                     end_time = time.time()
                     end_time = time.time()
                     try:
                     try:
-                        OfflineReport.objects.create(
+                        report_table = OfflineReport if input_dir.find(consts.HIL_PREFIX) != -1 else AFCOfflineReport
+                        report_table.objects.create(
                             input_folder=input_dir,
                             input_folder=input_dir,
                             doc_type=consts.FOLDER_WSC_CLASSIFY,
                             doc_type=consts.FOLDER_WSC_CLASSIFY,
                             file_name=name,
                             file_name=name,

@@ -923,6 +923,7 @@ class MposReport(models.Model):
     image_count = models.SmallIntegerField(default=0, verbose_name="图片数目")
     image_count = models.SmallIntegerField(default=0, verbose_name="图片数目")
     # status = models.SmallIntegerField(default=, verbose_name="请求状态")
     # status = models.SmallIntegerField(default=, verbose_name="请求状态")
     status = models.BooleanField(default=True, verbose_name="是否成功")
     status = models.BooleanField(default=True, verbose_name="是否成功")
+    source = models.SmallIntegerField(default=0, verbose_name="来源")  # 默认0，代表MPOS
 
 
     duration = models.IntegerField(null=True, verbose_name='处理时长')
     duration = models.IntegerField(null=True, verbose_name='处理时长')
 
 
@@ -950,20 +951,30 @@ class OfflineReport(models.Model):
         db_table = 'offline_report'
         db_table = 'offline_report'
 
 
 
 
-# class AFCOfflineReport(models.Model):
-#     id = models.AutoField(primary_key=True, verbose_name="id")  # 主键
-#
-#     input_folder = models.CharField(max_length=512, verbose_name="文件夹路径")
-#     doc_type = models.SmallIntegerField(default=0, verbose_name="文件类型")
-#     file_name = models.CharField(max_length=1024, verbose_name="文件名")
-#     status = models.BooleanField(default=True, verbose_name="是否成功")
-#     failure_reason = models.SmallIntegerField(default=0, verbose_name="失败原因")
-#     duration = models.IntegerField(verbose_name='处理时长')
-#
-#     create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
-#
-#     class Meta:
-#         managed = False
-#         db_table = 'afc_offline_report'
-#         situ_db_label = 'afc'
+class AFCOfflineReport(models.Model):
+    id = models.AutoField(primary_key=True, verbose_name="id")  # 主键
+
+    input_folder = models.CharField(max_length=512, verbose_name="文件夹路径")
+    doc_type = models.SmallIntegerField(default=0, verbose_name="文件类型")
+    file_name = models.CharField(max_length=1024, verbose_name="文件名")
+    status = models.BooleanField(default=True, verbose_name="是否成功")
+    failure_reason = models.SmallIntegerField(default=0, verbose_name="失败原因")
+    duration = models.IntegerField(verbose_name='处理时长')
+
+    create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
+
+    class Meta:
+        managed = False
+        db_table = 'afc_offline_report'
+        situ_db_label = 'afc'
 
 
+
+class GenericOCRReport(models.Model):
+    id = models.AutoField(primary_key=True, verbose_name="id")  # 主键
+    status = models.BooleanField(default=True, verbose_name="是否成功")
+    duration = models.IntegerField(null=True, verbose_name='处理时长')
+    create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
+
+    class Meta:
+        managed = False
+        db_table = 'generic_ocr_report'

@@ -46,6 +46,7 @@ from .models import (
     HILbankVerification,
     HILbankVerification,
     AFCbankVerification,
     AFCbankVerification,
     MposReport,
     MposReport,
+    GenericOCRReport,
 )
 )
 from .named_enum import ErrorType, AutoResult, WholeResult, RPAResult
 from .named_enum import ErrorType, AutoResult, WholeResult, RPAResult
 from .mixins import DocHandler, MPOSHandler, PreSEHandler
 from .mixins import DocHandler, MPOSHandler, PreSEHandler
@@ -1538,14 +1539,34 @@ class GoView(GenericView):
 
 
     @use_args(go_args, location='files')
     @use_args(go_args, location='files')
     def post(self, request, args):
     def post(self, request, args):
+        result = None
+        is_success = False
+        start_time = time.time()
+        try:
             files = [
             files = [
                 ('img', ('file', args.get('image'), 'application/octet-stream'))
                 ('img', ('file', args.get('image'), 'application/octet-stream'))
             ]
             ]
 
 
             go_result = requests.post(url=conf.GO_OCR_URL, files=files)
             go_result = requests.post(url=conf.GO_OCR_URL, files=files)
-
+        except Exception as e:
+            pass
+        else:
             if go_result.status_code == 200:
             if go_result.status_code == 200:
+                is_success = True
                 result = go_result.json().get('data', '')
                 result = go_result.json().get('data', '')
+        finally:
+            end_time = time.time()
+            duration_second = int(end_time - start_time)
+
+            try:
+                GenericOCRReport.objects.create(
+                    status=is_success,
+                    duration=duration_second,
+                )
+            except Exception as e:
+                self.exception_log.exception('[go view] [db save failed] [error={0}]'.format(traceback.format_exc()))
+
+            if is_success:
                 return response.ok(data=result)
                 return response.ok(data=result)
             else:
             else:
-            return response.res_content(3, '识别错误', data=None)
+                return response.error_msg(msg='识别错误')

@@ -35,6 +35,10 @@ def ok(**kwargs):
     return APIResponse(MetaStatus.SUCCESS.value, msg=MetaStatus.SUCCESS.verbose_name, **kwargs)
     return APIResponse(MetaStatus.SUCCESS.value, msg=MetaStatus.SUCCESS.verbose_name, **kwargs)
 
 
 
 
+def error_msg(msg='internal error', **kwargs):
+    return APIResponse(MetaStatus.INTERNAL_ERROR.value, msg=msg, **kwargs)
+
+
 def need_update(**kwargs):
 def need_update(**kwargs):
     return APIResponse(MetaStatus.NEED_UPDATE.value, msg=MetaStatus.NEED_UPDATE.verbose_name, **kwargs)
     return APIResponse(MetaStatus.NEED_UPDATE.value, msg=MetaStatus.NEED_UPDATE.verbose_name, **kwargs)
 
 

+import pyodbc
+
+hil_sql = """
+    create table generic_ocr_report
+    (
+       id bigint identity primary key,
+       status bit default 1 not null,
+       duration smallint,
+       create_time datetime not null
+    );
+
+    ALTER TABLE mpos_report ADD source tinyint default 0 not null;
+"""
+
+afc_sql = """
+    create table afc_offline_report
+    (
+       id bigint identity primary key,
+       input_folder nvarchar(512) not null,
+       doc_type tinyint default 0 not null,
+       file_name nvarchar(1024) not null,
+       status bit default 1 not null,
+       failure_reason tinyint default 0 not null,
+       duration smallint not null,
+       create_time datetime not null
+    );
+"""
+
+hil_cnxn = pyodbc.connect('DRIVER={ODBC Driver 17 for SQL Server};', autocommit=True)
+
+hil_cursor = hil_cnxn.cursor()
+hil_cursor.execute(hil_sql)
+
+hil_cursor.close()
+hil_cnxn.close()
+
+afc_cnxn = pyodbc.connect('DRIVER={ODBC Driver 17 for SQL Server};', autocommit=True)
+
+afc_cursor = afc_cnxn.cursor()
+afc_cursor.execute(afc_sql)
+
+afc_cursor.close()
+afc_cnxn.close()

@@ -12,6 +12,6 @@ EDMS_DOWNLOAD_URL = http://sccn0637.bmwgroup.net/FH/FileHold/DocumentRepository/
 EDMS_UPLOAD_URL = http://sccn0637.bmwgroup.net/FH/FileHold/DocumentRepository/UploadHandler.ashx
 EDMS_UPLOAD_URL = http://sccn0637.bmwgroup.net/FH/FileHold/DocumentRepository/UploadHandler.ashx
 DEALER_CODE = ocr_situ_group
 DEALER_CODE = ocr_situ_group
 
 
-BASE_URL = https://li19dkocruat01vm.bmwgroup.net
+BASE_URL = https://sfocr-uat.bmwgroup.net
 
 
 DELAY_SECONDS = 60
 DELAY_SECONDS = 60
\ No newline at end of file