调整项目结构，模块命名最好不要和包名相同，容易导致导包问题

2024-08-23 15:22:20 +08:00
parent 9d1b4abb89
commit f367a521fb
7 changed files with 8 additions and 8 deletions
--- a/photo_mask/photo_mask_error_check.py
+++ b/photo_mask/photo_mask_error_check.py
@@ -0,0 +1,98 @@
+import sys
+from datetime import date, timedelta
+
+import cv2
+from sqlalchemy import update, and_, select, exists
+
+from db import MysqlSession
+from db.mysql import ZxIeOcrerror, ZxPhrec
+from photo_mask.auto_photo_mask import mask_photo
+from ucloud import ufile
+from util import image_util, util
+
+
+def check_error(error_ocr):
+    img_url = ufile.get_private_url(error_ocr.cfjaddress, "drg2015")
+    if not img_url:
+        # 没有自动涂抹的图片
+        img_url = ufile.get_private_url(error_ocr.cfjaddress, "drg103")
+        if not img_url:
+            raise Exception("云空间已删除")
+    name = error_ocr.cXm
+    id_card_num = error_ocr.cSfzh
+
+    image = mask_photo(img_url, name, id_card_num, (0, 0, 0))[1]
+    final_img_url = ufile.get_private_url(error_ocr.cfjaddress, "drg100")
+    final_image = image_util.read(final_img_url)
+    return image_util.combined(final_image, image)
+
+
+def auto_check_error(error_ocr):
+    if error_ocr.cfjaddress2:
+        return error_ocr.cfjaddress2
+
+    cfjaddress = error_ocr.cfjaddress
+    if cfjaddress[17] == "1":
+        return "图片类型错误"
+
+    session = MysqlSession()
+    cfjaddress = cfjaddress.rsplit(".", 1)[0] + ".%"
+    query = select(exists().where(ZxPhrec.cfjaddress.like(cfjaddress)))
+    record_exists = session.execute(query).scalar()
+    session.close()
+    if not record_exists:
+        return "未同步"
+
+    return "未知错误"
+
+
+if __name__ == '__main__':
+    # 默认
+    today = date.today()
+    session = MysqlSession()
+    ocr_error = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress,
+                               ZxIeOcrerror.cfjaddress2)
+                 .filter(and_(ZxIeOcrerror.checktime.is_(None),
+                              # ZxIeOcrerror.creationtime >= today,
+                              ZxIeOcrerror.creationtime < today + timedelta(days=1)))
+                 .limit(1).one())
+    session.close()
+    # 手动填充
+    # ocr_error = ZxIeOcrerror()
+    # ocr_error.pk_phrec = 0
+    # ocr_error.cXm = ""
+    # ocr_error.cSfzh = ""
+    # ocr_error.cfjaddress = ""
+    # ocr_error.cfjaddress2 = ""
+    # 手动选择pk
+    # session = MysqlSession()
+    # ocr_error = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress,
+    #                            ZxIeOcrerror.cfjaddress2)
+    #              .filter(ZxIeOcrerror.pk_phrec == 0).one())
+    # session.close()
+
+    error_descript = auto_check_error(ocr_error)
+    try:
+        img = check_error(ocr_error)
+        cv2.imwrite(f"./mask_error_check/{ocr_error.cfjaddress}.jpg", img)
+    except Exception as e:
+        error_descript = str(e)
+
+    if error_descript == "未知错误":
+        while True:
+            check_finish = input("是否完成(是/否)：")
+            if check_finish == "是":
+                error_descript = input("错误描述：")
+                break
+            elif check_finish == "否":
+                sys.exit()
+            else:
+                print("无效输入！请输入“是”或者“否”")
+
+    session = MysqlSession()
+    update_error = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
+        checktime=util.get_default_datetime(), cfjaddress2=error_descript))
+    session.execute(update_error)
+    session.commit()
+    session.close()
+    print(f"【{ocr_error.pk_phrec}】错误分析已完成。错误描述：{error_descript}")