添加判断tmp_res是否为空的逻辑

chenyao
Showing 1 changed file with 2 additions and 2 deletions
src/apps/doc/management/commands/ocr_process.py
--- a/src/apps/doc/management/commands/ocr_process.py
View file @5958e83
+++ b/src/apps/doc/management/commands/ocr_process.py
View file @5958e83
@@ -1018,9 +1018,9 @@ class Command(BaseCommand, LoggerMixin):
                        # 添加处理, 
                        # [售后回租合同] - 如果 key 是 "承租人签字", 且内容中包含 签署日期:XXXX, 则将签署日期去除
                        # [车辆租赁抵押合同] - 如果 key 是 ""
-                        if key == '承租人签字' and '签署日期' in tmp_res:
+                        if key == '承租人签字' and tmp_res is not None and '签署日期' in tmp_res:
                            res[key] = tmp_res.split('签署日期')[0]
-                        if key == "抵押人签字" and "签署日期" in tmp_res:
+                        if key == "抵押人签字" and tmp_res is not None and "签署日期" in tmp_res:
                            res[key] = tmp_res.split("签署日期")[0]
                        res.setdefault(consts.IMG_PATH_KEY, dict())[key] = page_info_dict.get(str(img_pno), {}).get(
                            consts.IMG_PATH_KEY, '')