MOD:保存ocr结果遇到报错重试一次，希望解决两个文件首次入库都插入的问题

冯轩
Showing 1 changed file with 20 additions and 0 deletions
src/apps/doc/management/commands/ocr_process.py
--- a/src/apps/doc/management/commands/ocr_process.py
View file @12dbbe5
+++ b/src/apps/doc/management/commands/ocr_process.py
View file @12dbbe5
@@ -2130,9 +2130,19 @@ class Command(BaseCommand, LoggerMixin):
                                        # 更新OCR累计识别结果表
                                        if business_type == consts.HIL_PREFIX:
                                            result_class = HILOCRResult if is_ca else HILSEOCRResult
+                                            try:
+                                                res_obj = atomicSaveDBHIL(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
+                                            except Exception as e:
+                                            # 遇到报错重试一次，希望解决两个文件首次入库都插入的问题
+                                                self.online_log.error('{0} [process error (ocr result save) retry] [task={1}] [error={2}]'.format(self.log_base, task_str, traceback.format_exc()))
                                                res_obj = atomicSaveDBHIL(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
                                        else:
                                            result_class = AFCOCRResult if is_ca else AFCSEOCRResult
+                                            try:
+                                                res_obj = atomicSaveDBAFC(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
+                                            except Exception as e:
+                                            # 遇到报错重试一次，希望解决两个文件首次入库都插入的问题
+                                                self.online_log.error('{0} [process error (ocr result save) retry] [task={1}] [error={2}]'.format(self.log_base, task_str, traceback.format_exc()))
                                                res_obj = atomicSaveDBAFC(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
                                    except Exception as e:
@@ -2172,9 +2182,19 @@ class Command(BaseCommand, LoggerMixin):
                                    # 更新OCR累计识别结果表
                                    if business_type == consts.HIL_PREFIX:
                                        result_class = HILOCRResult if is_ca else HILSEOCRResult
+                                        try:
+                                            res_obj = atomicSaveDBHIL(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
+                                        except Exception as e:
+                                            # 遇到报错重试一次，希望解决两个文件首次入库都插入的问题
+                                            self.online_log.error('{0} [process error (ocr result save) retry] [task={1}] [error={2}]'.format(self.log_base, task_str, traceback.format_exc()))
                                            res_obj = atomicSaveDBHIL(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
                                    else:
                                        result_class = AFCOCRResult if is_ca else AFCSEOCRResult
+                                        try:
+                                            res_obj = atomicSaveDBAFC(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
+                                        except Exception as e:
+                                            # 遇到报错重试一次，希望解决两个文件首次入库都插入的问题
+                                            self.online_log.error('{0} [process error (ocr result save) retry] [task={1}] [error={2}]'.format(self.log_base, task_str, traceback.format_exc()))
                                            res_obj = atomicSaveDBAFC(self,result_class,doc,license_summary,ic_merge,rp_merge, task_str, financial_statement_dict, financial_explanation_dict)
                                except Exception as e:
                                    self.online_log.error(