from datetime import datetime
from base import Base
from PIL import Image
import pytesseract
import platform


class ImageTextOcr(Base):

    def __init__(self):
        super(ImageTextOcr, self).__init__()
        current_os = platform.system()
        if current_os == 'Windows':
            pytesseract.pytesseract.tesseract_cmd = r'E:\pc\tesseract-ocr\tesseract.exe'
            self.log("当前操作系统是 Windows")
        elif current_os == 'Linux':
            self.log("当前操作系统是 Ubuntu")
        else:
            self.log(f"当前操作系统是 {current_os}")

    def is_valid_time(self, input_str):
        try:
            valid_time = datetime.strptime(input_str, "%Y-%m-%d")  # 根据实际时间格式调整
            return valid_time
        except ValueError:
            return False

    def image_text_ocr(self, text_dict, path):
        valid_time_list = []
        # 加载图像
        image = Image.open(path)
        result = pytesseract.image_to_string(image, config=r'--oem 3 --psm 6 -l chi_sim+eng')
        lines = result.split()
        for line in lines:
            if 'S$T' in line or 'SST' in line:
                text_dict['方案编号'] = line.replace('S$T', 'SST').replace('试验方案编号:', '')

            if 'CNAS' in line:
                text_dict['标志'] = 'cnas中文,'

            if '200015344424' in line:
                text_dict['标志'] = '国cma,'

            valid_time = self.is_valid_time(line)
            if valid_time:
                valid_time_list.append(valid_time)

        if valid_time_list:
            text_dict['签发日期'] = max(valid_time_list).strftime("%Y-%m-%d")
        return text_dict

    def run(self, text_dict, path):
        res_list = self.image_text_ocr(text_dict, path)
        return res_list


if __name__ == '__main__':
    image_text_ocr = ImageTextOcr()
    text_dict = {
        '方案编号': '',
        '签发日期': '',
        '标志': ''
    }
    res = image_text_ocr.run(text_dict, '../target_img/image_2.jpg')
    print(res)