Python+Selenium實現web自動化跳過登錄

背景

自動化模擬登錄，遇到圖形驗證碼或者動態驗證碼時，代碼操作複雜。該怎麼跳過登錄？

思路

研究文檔發現，selenium有add_cookie()方法

方法

driver.add_cookie({'name': k, 'value': v})

其中k,v對應一條cookie的鍵值

代碼

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time, sys, io

driver = webdriver.Chrome(executable_path="C:/Program Files (x86)/Google/Chrome/Application/chromedriver.exe")
driver.get("http://example.com/index?argv=1")
assert "頁面標題" in driver.title
driver.maximize_window()

# __IDENTIFY__=1951aca300a50f64a26a8ed5e970f4c3
# ttdd_session_id=040f57dbd50492b70c0a5322f82ff9f9f8212cc969367b366f9ecffc3fe5999b
cookies = dict(__IDENTIFY__='1951aca300a50f64a26a8ed5e970f4c3', ttdd_session_id='040f57dbd50492b70c0a5322f82ff9f9f8212cc969367b366f9ecffc3fe5999b')
for k, v in cookies.items():
    driver.add_cookie({'name': k, 'value': v})

time.sleep(1)
driver.refresh()
# driver.get("http://example.com/index?argv=1")  # why not work?
time.sleep(5)
# 點擊 元素
e = driver.find_element_by_xpath('/html[1]/body[1]/div[1]/div[3]/div[1]/div[1]/ul[1]/li[9]/a[1]/span[1]')  # use ChroPath to locate element
e.click()
time.sleep(2)
# 點擊 元素
e = driver.find_element_by_xpath('/html[1]/body[1]/div[1]/div[3]/div[1]/div[1]/ul[1]/li[9]/ul[1]/li[7]/a[1]/span[1]')  # use ChroPath to locate element
e.click()
time.sleep(2)
# 輸入文本
# e= driver.find_element_by_xpath('/html[1]/body[1]/div[1]/div[3]/div[2]/div[1]/div[2]/form[1]/div[1]/input[1]')
e = driver.find_element_by_xpath("//input[contains(@placeholder,'只支持item_id搜索')]")
e.click()
e.send_keys(538)
time.sleep(2)
# 點擊 元素
e = driver.find_element_by_xpath("//button[contains(@class,'btn btn-sm green table-group-action-submit')]")
e.click()
time.sleep(2)   
# 查找是否有 元素
try:
    e = driver.find_element_by_xpath("//tr[1]//td[14]//a[2]")
    print('找到元素')
except:
    print('無此元素')


# 遍歷表格
cells = []
trlist = driver.find_elements_by_tag_name('tr')
for row in trlist:
    tdlist = row.find_elements_by_tag_name('td')
    print('列表行數', tdlist)
    row_list = []
    for col in tdlist:
        row_list.append(col.text)
        print(col.text + '\t', end= '')
    print('\n')
    cells.append(row_list)
print('*' * 200)
print(cells)

time.sleep(10)
driver.close() # 關閉當前tab
time.sleep(5)
driver.quit() # 關閉瀏覽器所有tab

總結

1、添加多條cookie，可用for循環
2、使用by_tag_name方法對錶格內容進行遍歷
3、cookie抓包可用fiddler，或者瀏覽器F12功能
4、元素定位可用Chrome插件ChroPath
5、WebDriver下載地址： https://selenium.dev/documentation/en/webdriver/driver_requirements/#quick-reference
6、WebDriver路徑添加到系統Path