report fix

@@ -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,20 @@ 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'
 
 

+import pyodbc
+
+hil_sql = """
+    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()