优化错误涂抹分析

photo_mask_error_report.py

@@ -1,10 +1,11 @@
 from collections import defaultdict
 from datetime import date, timedelta
 
-from sqlalchemy import update, exists, select, and_
+from sqlalchemy import update, and_
 
 from db import MysqlSession
-from db.mysql import ZxIeOcrerror, ZxPhrec
+from db.mysql import ZxIeOcrerror
+from photo_mask_error_check import auto_check_error
 from util import util
 
 if __name__ == '__main__':
@@ -13,8 +14,7 @@ if __name__ == '__main__':
     ocr_errors = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress,
                                 ZxIeOcrerror.pk_phhd, ZxIeOcrerror.cfjaddress2)
                   .filter(and_(ZxIeOcrerror.creationtime >= today,
-                               ZxIeOcrerror.creationtime < today + timedelta(days=1)))
-                  .order_by(ZxIeOcrerror.pk_phrec.desc()).all())
+                               ZxIeOcrerror.creationtime < today + timedelta(days=1))).all())
     session.close()
 
     result = defaultdict(int)
@@ -24,35 +24,17 @@ if __name__ == '__main__':
             result[ocr_error.cfjaddress2] += 1
             continue
 
-        if ocr_error.cfjaddress[17] == '1':
-            session = MysqlSession()
-            update_flag = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
-                checktime=util.get_default_datetime(), cfjaddress2="图片类型错误"))
-            session.execute(update_flag)
-            session.commit()
-            session.close()
-            result["图片类型错误"] += 1
-            continue
+        error_descript = auto_check_error(ocr_error)
+        if error_descript == "未知错误":
+            check_time = None
+        else:
+            check_time = util.get_default_datetime()
 
         session = MysqlSession()
-        query = select(exists().where(ZxPhrec.cfjaddress == ocr_error.cfjaddress))
-        record_exists = session.execute(query).scalar()
-        session.close()
-        if not record_exists:
-            session = MysqlSession()
-            update_flag = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
-                checktime=util.get_default_datetime(), cfjaddress2="未同步"))
-            session.execute(update_flag)
-            session.commit()
-            session.close()
-            result["未同步"] += 1
-            continue
-
-        session = MysqlSession()
-        update_flag = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
-            cfjaddress2="未知错误"))
-        session.execute(update_flag)
+        update_error = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
+            checktime=check_time, cfjaddress2=error_descript))
+        session.execute(update_error)
         session.commit()
         session.close()
-        result["未知错误"] += 1
+        result[error_descript] += 1
     print(result)