import cv2
from sqlalchemy import update

import util.util
from db import MysqlSession
from db.mysql import ZxIeOcrerror
from photo_mask.photo_mask import mask_photo
from ucloud import ufile
from util import image_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")
    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)


if __name__ == '__main__':
    # 默认
    session = MysqlSession()
    ocr_error = (session.query(ZxIeOcrerror.pk_phrec, ZxIeOcrerror.cXm, ZxIeOcrerror.cSfzh, ZxIeOcrerror.cfjaddress)
                 .filter(ZxIeOcrerror.checktime.is_(None)).order_by(ZxIeOcrerror.pk_phrec.desc()).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()

    img = check_error(ocr_error)
    cv2.imwrite(f"./mask_error_check/{ocr_error.cfjaddress}.jpg", img)

    check_finish = input("是否完成(是/否)：")
    if check_finish == "是":
        remarks = input("备注：")
        session = MysqlSession()
        update_flag = (update(ZxIeOcrerror).where(ZxIeOcrerror.pk_phrec == ocr_error.pk_phrec).values(
            checktime=util.util.get_default_datetime(), cfjaddress2=remarks))
        session.execute(update_flag)
        session.commit()
        session.close()