85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
from datetime import date
|
||
|
||
import cv2
|
||
from sqlalchemy import update, and_, select, exists
|
||
|
||
from db import MysqlSession
|
||
from db.mysql import ZxIeOcrerror, ZxPhrec
|
||
from photo_mask.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
|
||
|
||
if error_ocr.cfjaddress[17] == '1':
|
||
return "图片类型错误"
|
||
|
||
session = MysqlSession()
|
||
query = select(exists().where(ZxPhrec.cfjaddress == error_ocr.cfjaddress))
|
||
record_exists = session.execute(query).scalar()
|
||
session.close()
|
||
if not record_exists:
|
||
return "未同步"
|
||
|
||
return "未知错误"
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 默认
|
||
session = MysqlSession()
|
||
ocr_error = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress,
|
||
ZxIeOcrerror.cfjaddress2)
|
||
.filter(and_(ZxIeOcrerror.checktime.is_(None), ZxIeOcrerror.paint_date >= date.today()))
|
||
.limit(1).one())
|
||
session.close()
|
||
# 手动填充
|
||
# ocr_error = ZxIeOcrerror()
|
||
# ocr_error.pk_phrec = 0
|
||
# ocr_error.cXm = ""
|
||
# ocr_error.cSfzh = ""
|
||
# ocr_error.cfjaddress = ""
|
||
# 手动选择pk
|
||
# session = MysqlSession()
|
||
# ocr_error = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress)
|
||
# .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 == "未知错误":
|
||
check_finish = input("是否完成(是/否)：")
|
||
if check_finish == "是":
|
||
error_descript = input("错误描述：")
|
||
|
||
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}")
|