百度api識別驗證碼登錄

import time
from selenium import webdriver
from aip import AipOcr

def initial():
    """ 初始化連接 """
    APP_ID = '16611607'
    API_KEY = 'wAIXfXOUS8ztLa4FrK3rZex1'
    SECRET_KEY = '3b8nvjSGUZq0LPC18VVAizKYRBbny6Mq'
    return AipOcr(APP_ID, API_KEY, SECRET_KEY)
    
def get_file_content(filePath):
    """ 讀取圖片 """
    with open(filePath, 'rb') as f:
        return f.read()
        
def selenium_msg():
    driver = webdriver.Chrome()
    try:
        driver.get('https://pythonav.com/login/')
        time.sleep(1)

        # 輸入用戶密碼
        driver.find_element_by_id('id_username').send_keys('python')
        time.sleep(1)
        driver.find_element_by_id('id_password').send_keys('dsb123')
        time.sleep(2)

        # 驗證碼識別
        file_path = 'b.png' 
        image = driver.find_element_by_id('image_code')  #找到驗證碼 
        image.screenshot(file_path)   #保存驗證碼圖片

        # 百度ai相關
        client = initial()
        image = get_file_content(file_path)   #讀取圖片文件
        res1 = client.basicGeneral(image)  # 調用通用文字識別, 圖片參數為本地圖片
        # res2 = client.basicAccurate(image)  # 調用通用文字識別（高精度版）
        a = ''
        for text in res1['words_result']:
            a += text['words']
        time.sleep(2)
        driver.find_element_by_id('id_code').send_keys(a)
        time.sleep(2)
        driver.find_element_by_class_name('btn-primary').click()  #登錄按鈕進行點擊
    finally:
        time.sleep(5)
        driver.quit()
if __name__ == '__main__':
    selenium_msg()