selenium自动化识别注册验证码

base.py：
import pyautogui
import pytesseract
import time
from PIL import Image
class Base():

def __init__(self, driver:webdriver.Chrome, base_url='',  timeout=10, t=0.5):
    self.driver = driver
    self.timeout = timeout
    self.t = t
    self.base_url = base_url

#获取验证码

def getId(self, ele):
    # 通过id获取元素
    return self.driver.find_element_by_id(ele)
   
      def verification_code(self, img_ele, img_value):
    # 获取全屏图片，并截取验证码图片的位置
    self.driver.get_screenshot_as_file('verification_code_image.png')
    if img_ele == "ID":
        location = self.getId(img_value).location
        size = self.getId(img_value).size
        print(location)
    elif img_ele == "CLASS_NAME":
        location = self.getClass(img_value).location
        size = self.getClass(img_value).size
    elif img_ele ==