selenium實現登錄百度（自動識別簡單驗證碼）

需要做的工作

0、工程結構

 

1、代碼：

 ①baidu_login.py

import re
import os
import sys
import time
import random
from selenium import webdriver
from PIL import Image, ImageEnhance
import pytesseract
from func import base642str, str2base64


def input_account(account='請傳入賬號參數', xpath_rule="//input[@id='TANGRAM__PSP_3__userName']"):
    '''模擬輸入賬號
    :param account: 賬號
    :param xpath_rule: 賬號輸入框的xpath定位規則
    :return:
    '''
    input_box1 = driver.find_element_by_xpath(xpath_rule)
    input_box1.send_keys(account)
    time.sleep(0.5)


def input_pwd(pwd, xpath_rule="//input[@id='TANGRAM__PSP_3__password']"):
    '''模擬輸入密碼
    :param account: base64后的密碼
    :param xpath_rule: 密碼輸入框的xpath定位規則
    :return:
    '''
    input_box2 = driver.find_element_by_xpath(xpath_rule)
    input_box2.clear()  # 清空密碼
    input_box2.send_keys(base642str(pwd))
    time.sleep(0.5)


def input_verify_code(verify_code, xpath_rule="TANGRAM__PSP_3__verifyCode"):
    '''模擬輸入驗證碼
    :param account: base64后的密碼
    :param xpath_rule: 密碼輸入框的xpath定位規則
    :return:
    '''
    driver.find_element_by_id(xpath_rule).send_keys(verify_code.strip())
    time.sleep(0.5)


def identify_verify_code(rootpath, pic_name="screenImg.png"):
    '''tesseract識別百度驗證碼
    :param rootpath: 驗證碼圖片保存的文件夾路徑
    :param pic_name: 驗證碼圖片保存的文件名
    :return: 識別后的驗證碼文本
    '''
    # 截圖或驗證碼圖片保存地址
    screenImg = os.path.join(rootpath, pic_name)
    # 瀏覽器頁面截屏
    time.sleep(3)
    driver.get_screenshot_as_file(screenImg)
    # 定位驗證碼位置及大小
    location = driver.find_element_by_id('TANGRAM__PSP_3__verifyCodeImg').location
    size = driver.find_element_by_id('TANGRAM__PSP_3__verifyCodeImg').size
    left = location['x']
    top = location['y']
    right = location['x'] + size['width']
    bottom = location['y'] + size['height']
    # 從文件讀取截圖，截取驗證碼位置再次保存
    img = Image.open(screenImg).crop((left, top, right, bottom))
    img = img.convert('L')  # 轉換模式：L | RGB
    img = ImageEnhance.Contrast(img)  # 增強對比度
    img = img.enhance(2.0)  # 增加飽和度
    img.save(screenImg)
    print("圖片驗證碼以保存：%s" % screenImg)
    # 再次讀取識別驗證碼
    print("開始讀取識別圖片驗證碼：%s" % screenImg)
    img = Image.open(screenImg)
    verifycode = pytesseract.image_to_string(img)
    print("識別結果：%s" % verifycode)
    return verifycode


def click_a_link(xpath_rule="//p[@id='TANGRAM__PSP_3__footerULoginBtn']"):
    '''點擊一個鏈接
    :param xpath_rule: 被點擊鏈接的xpath定位規則
    :return:
    '''
    input_box0 = driver.find_element_by_xpath(xpath_rule)
    input_box0.click()


def click_a_id_link(id_rule="TANGRAM__PSP_3__verifyCodeChange"):
    '''點擊一個鏈接
    :param id_rule: 被點擊鏈接的id定位規則
    :return:
    '''
    input_box0 = driver.find_element_by_id(id_rule)
    input_box0.click()


def is_need_verify_code():
    '''判斷是否需要驗證碼
    :return: 需要驗證碼返回True，否則False
    '''
    imgsrc = driver.find_element_by_id("TANGRAM__PSP_3__verifyCodeImg").get_attribute('src')
    if re.match(r'https://passport.baidu.com/cgi-bin/genimage.*', imgsrc):
        return True
    else:
        return False


def get_id_node_text(id_rule="TANGRAM__PSP_3__error"):
    '''獲取id節點提示信息
    :param id_rule:id節點的id匹配規則（id屬性的值）
    :return:該id節點中的文本信息
    '''
    one_node = driver.find_element_by_id(id_rule)
    text_info = one_node.text
    return text_info


def is_login_success():
    '''判斷登錄是否成功
    :return: 登錄成功返回True，否則False
    '''
    current_title = driver.title.strip()
    if current_title.startswith("登錄"):
        return False
    else:
        return True


def deal_much_pop_up_window():
    '''處理手機驗證碼認證反復彈窗
    :return:
    '''
    i = 0
    while True:  # 處理手機驗證碼認證反復彈窗
        try:
            one3_click = driver.find_element_by_id("TANGRAM__%s__header_a" % (22 + i))  # 22+i在應對彈窗的關閉按鈕id名稱發生變化。
        except Exception as e:
            print("無需手機驗證碼")
            break
        else:
            print("第 %s 次彈出安全驗證，要求獲取手機驗證碼" % (i + 1))
            time.sleep(0.5)
            print("1s后自動選擇無需手機驗證碼")
            time.sleep(1)
            one3_click.click()
            print("1s后自動點擊登陸")
            time.sleep(1)
            click_a_link(xpath_rule="//input[@id='TANGRAM__PSP_3__submit']")  # 點擊登錄，提交表單
            time.sleep(2)
            # 判斷是否成功登陸
            current_title = driver.title.strip()
            if current_title.startswith("登錄"):
                print('333-登陸失敗...')
                time.sleep(0.5)
                print('333-2秒后自動重試...')
                i = i + 1
                time.sleep(2)
                continue
            else:
                print("打印標題")
                print(driver.title)
                print('333-登錄成功...')
                sys.exit()  # 程序終止


def deal_a_pop_up_window(xpath_rule="//input[@id='TANGRAM__PSP_27__rebindGuideCancel']"):
    '''處理一次彈窗
    :param xpath_rule:處理彈窗的按鈕/鏈接的xpath匹配規則
    :return:
    '''
    # 判斷是否需要手機號綁定確認
    try:
        # 綁定手機號確認
        one_click = driver.find_element_by_xpath(xpath_rule)
    except Exception as e:
        print("無需綁定手機號確認")
    else:
        print("彈出了綁定手機號確認，1s后自動選擇不需要")
        time.sleep(1)
        one_click.click()


if __name__ == '__main__':
    # 將mm隱匿化
    # print(str2base64("123456"))
    # str1_base64="MTIzNDU2"
    # print(base642str(str1_base64))
    # sys.exit()

    # 賬號和密碼准備
    account = random.randint(0, 1000000)  # 隨機數字作為賬號
    pwd = "MTIzNDU2"
    print("account: %s" % account)
    # 最大登錄次數
    max_login = 16
    # 當前目錄設置為根路徑
    ROOT_PATH = os.getcwd()
    print('000-正在啟用selenium...')
    # 調用環境變量指定的PhantomJS瀏覽器創建瀏覽器對象
    chromedriver_exe_path = os.path.join(ROOT_PATH, "chromedriver.exe")
    driver = webdriver.Chrome(chromedriver_exe_path)
    print('000-啟用OK')

    # 請求登錄頁面
    url = 'https://passport.baidu.com/v2/?login'
    print('111-selenium正在請求頁面：%s' % url)
    driver.get(url)  # get方法請求頁面，獲取響應
    print('111-請求OK')

    print("打印標題")
    print(driver.title)

    # 點擊賬號和密碼登錄
    click_a_link()

    print('222-selenium正在填寫表單...')
    time.sleep(1)
    # 第一次嘗試登錄

    # 模擬填寫賬號
    input_account(account)

    # 模擬填寫密碼
    input_pwd(pwd)

    # 判斷是否需要驗證碼
    is_need = is_need_verify_code()
    if is_need:  # 需要驗證碼
        print("需要驗證碼")
        # 自動識別和模擬填寫驗證碼
        code = identify_verify_code(rootpath=ROOT_PATH, pic_name="screenImg.png")  # 自動識別驗證碼
        input_verify_code(code)  # 模擬填寫驗證碼
    else:  # 不需要驗證碼
        print("不需要驗證碼")
        print('222-填寫表單OK')
        time.sleep(1)

    print('333-selenium提交表單...')
    click_a_link(xpath_rule="//input[@id='TANGRAM__PSP_3__submit']")  # 點擊登錄，提交表單
    print("第 %s 次嘗試登錄" % 1)
    time.sleep(3)

    # 處理反復彈窗（手機驗證碼）：點擊關閉按鈕
    deal_much_pop_up_window()
    # 處理一次彈窗（綁定手機號確認）：點擊不需要修改
    deal_a_pop_up_window()

    for login_i in range(max_login - 1):
        # 判斷是否登錄成
        is_logined = is_login_success()
        if is_logined:
            print("登錄成功")
            break
        else:
            print("第 %s 次登錄失敗，正在嘗試重新登錄..." % (login_i + 1))
            # 第二次嘗試登錄
            print("=>第 %s 次嘗試登錄" % (login_i + 2))
            error_info = get_id_node_text()
            if ("帳號或密碼錯誤" in error_info) or ("用戶名或密碼有誤"):  # 第二次嘗試登錄2.1
                print("正在嘗試重新輸入密碼...")
                # 模擬填寫密碼
                input_pwd(pwd)
                # 判斷是否需要驗證碼
                is_need = is_need_verify_code()
                if is_need:
                    print("需要驗證碼")
                    # 點擊更新驗證碼
                    click_a_id_link()
                    # 自動識別和模擬填寫驗證碼
                    code = identify_verify_code(rootpath=ROOT_PATH, pic_name="screenImg.png")  # 自動識別驗證碼
                    input_verify_code(code)  # 模擬填寫驗證碼
                    print("提示：驗證碼錯誤，js會使得提交無效")
                else:
                    print("不需要驗證碼")

            elif ("驗證碼" in error_info):  # 第二次嘗試登錄2.2
                print("正在嘗試重新輸入了驗證碼和密碼...")
                time.sleep(1)
                # 模擬填寫密碼
                input_pwd(pwd)
                # 點擊更新驗證碼
                click_a_id_link()
                # 自動識別和模擬填寫驗證碼
                code = identify_verify_code(rootpath=ROOT_PATH, pic_name="screenImg.png")  # 自動識別驗證碼
                input_verify_code(code)  # 模擬填寫驗證碼
                print("提示：驗證碼錯誤，js會使得提交無效")
                time.sleep(2)

            else:
                print("其他未知異常：登錄失敗")
                sys.exit()  # 程序終止

            print('222-填寫表單OK')
            time.sleep(1)

            print('333-selenium提交表單...')
            click_a_link(xpath_rule="//input[@id='TANGRAM__PSP_3__submit']")  # 點擊登錄，提交表單
            time.sleep(3)

    # 打印標題
    print("打印標題")
    print(driver.title)
    time.sleep(1)

    # 判斷是否成功登陸
    is_logined = is_login_success()
    if is_logined:
        print('333-登錄成功...')
    else:
        print('333-登陸失敗...')

②func.py

import base64
import time


def timestamp2datems(timestamp):
    '''
    時間戳轉為日期字串，精確到ms。單位s
    :param timestamp:時間戳
    :return:日期字串
    '''
    local_time = time.localtime(timestamp)
    # data_head = time.strftime("%Y-%m-%d %H:%M:%S", local_time)
    data_head = time.strftime("%Y-%m-%d_%H-%M-%S", local_time)
    data_secs = (timestamp - int(timestamp)) * 1000
    dt_ms = "%s.%03d" % (data_head, data_secs)
    # print(dt_ms)
    return dt_ms


def bit2humanView(bit_val):
    '''
    文件大小bit轉為人類易讀大小bit、KB、MB
    :param bit_val:字節數值
    :return:人類易讀大小和單位
    '''
    is2kb = int(bit_val / 1042)  # 轉換為kb取整
    is2mb = int(bit_val / 1024 / 1024)  # 轉為mb取整
    is2gb = int(bit_val / 1024 / 1024 / 1024)  # 轉為gb取整
    if is2gb is not 0:
        gb_val = bit_val / 1024 / 1024 / 1024
        return "%.2f GB" % gb_val
    if is2mb is not 0:
        mb_val = bit_val / 1024 / 1024
        return "%.2f MB" % mb_val
    if is2kb is not 0:
        kb_val = bit_val / 1024
        return "%.2f KB" % kb_val
    return "%s bit" % bit_val


def str2base64(pwd_decode_str):
    '''
    明文str轉為base64密文
    :param pwd_decode_str: 明文str
    :return: base64密文
    '''
    base64_encrypt = base64.b64encode(pwd_decode_str.encode('utf-8'))
    pwd_encode_str = str(base64_encrypt, 'utf-8')
    return pwd_encode_str


def base642str(pwd_encode_str):
    '''
    base64密文轉為明文str
    :param pwd_encode_str: base64密文
    :return: 明文str
    '''
    base64_decrypt = base64.b64decode(pwd_encode_str.encode('utf-8'))
    pwd_decode_str = str(base64_decrypt, 'utf-8')
    return pwd_decode_str

　　

③依賴包：requirements.txt

Pillow==6.0.0
pytesseract==0.2.6
selenium==3.141.0

　　

2、selenium的相關支持：

①google瀏覽器；

②相應版本的瀏覽器驅動chromedriver.exe

 http://npm.taobao.org/mirrors/chromedriver/

 

3、PIL安裝：pip  install  pillow

 

4、圖像識別工具：

①windows安裝tesseract.exe 

②python安裝pytesseract