import cv2

from db import MysqlSession
from db.mysql import ZxIeOcrerror
from photo_mask_error_check import check_error

if __name__ == '__main__':
    session = MysqlSession()
    ocr_errors = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress)
                  .filter(ZxIeOcrerror.checktime.is_(None)).order_by(ZxIeOcrerror.pk_phrec.desc()).limit(50).all())
    session.close()

    for ocr_error in ocr_errors:
        image = check_error(ocr_error)
        cv2.imwrite(f"./mask_optimization_result/{ocr_error.cfjaddress}.jpg", image)