diff --git a/src/apps/doc/management/commands/ocr_process.py b/src/apps/doc/management/commands/ocr_process.py index ab9b74f..d3ea91b 100644 --- a/src/apps/doc/management/commands/ocr_process.py +++ b/src/apps/doc/management/commands/ocr_process.py @@ -628,6 +628,13 @@ class Command(BaseCommand, LoggerMixin): except Exception as e: self.online_log.error('{0} [process error (report db save)] [error={1}]'.format( self.log_base, traceback.format_exc())) + + try: + doc.status = DocStatus.PROCESS_FAILED.value + doc.save() + except Exception as e: + self.online_log.error('{0} [process error (db save)] [error={1}]'.format( + self.log_base, traceback.format_exc())) else: with lock: todo_count_dict[task_str] = pdf_handler.img_count