import cv2

from db import MysqlSession
from db.mysql import ZxIeOcrerror
from photo_mask_error_check import check_error
from ucloud import ufile
from util import image_util

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:
        final_img_url = ufile.get_private_url(ocr_error.cfjaddress, "drg100")
        final_image = image_util.read(final_img_url)
        cv2.imwrite(f"./mask_optimization_result/answer/{ocr_error.cfjaddress}.jpg", final_image)
        image = check_error(ocr_error)
        cv2.imwrite(f"./mask_optimization_result/{ocr_error.cfjaddress}.jpg", image)