import re
from PIL import Image
from aip import AipOcr
from selenium import webdriver


def orc_test(browser):
    browser.save_screenshot(r'D:\JYD_project\本科审核评估\support_login\save_screenshot.png')
    right = 150 + 1350  # 上面右边点的坐标
    down = 45 + 375  # 下面右边点的坐标
    image = Image.open(r'D:\JYD_project\本科审核评估\support_login\save_screenshot.png')
    code_image = image.crop((1350, 375, right, down))
    code_image.save(r'D:\JYD_project\本科审核评估\support_login\code_image.png')
    APP_ID = ''
    API_KEY = ''
    SECRET_KEY = ''
    client = AipOcr(APP_ID, API_KEY, SECRET_KEY)

    # 读取图片
    def get_file_content(file_path):
        with open(file_path, 'rb') as fp:
            return fp.read()

    image = get_file_content(r'D:\JYD_project\本科审核评估\support_login\code_image.png')

    # 调用通用文字识别（高精度版） """
    code_ocr_original_result = client.basicAccurate(image)
    # print(code_ocr_original_result)
    code_ocr_final_result = ''
    for text in code_ocr_original_result['words_result']:
        for i in text.values():
            code_ocr_final_result = code_ocr_final_result + i

    code_ocr_final_result_with_out = code_ocr_final_result.replace(' ', '')
    result = re.sub('[\W_]', '', code_ocr_final_result_with_out)
    return result


if __name__ == '__main__':
    browser = webdriver.Edge()
    browser.maximize_window()
    browser.get('http://188.18.18.227:8091/')
    print(orc_test(browser))