软件测试学习笔记丨 Selenium 复用 cookie

2024-05-30
北京
本文字数：663 字
阅读完需：约 2 分钟

本文转自测试人社区，原文链接：https://ceshiren.com/t/topic/22521
说明：本篇博客基于 selenium 4.1.0

说明

cookie 复用，跳过登录步骤

# 1. 首次运行，手动登录进网站
# 2. 记录cookie信息到本地cookie = driver.get_cookies()
# 3. 后续再次登录时，传入本地cookie信息，登录进网站for i in cookie:    driver.add_cookie(i)

复制代码

代码示例

import osimport yamlfrom selenium import webdriverfrom selenium.webdriver.support import expected_conditionsfrom selenium.webdriver.support.wait import WebDriverWait

driver = webdriver.Chrome()driver.implicitly_wait(10)driver.get('https://work.weixin.qq.com/wework_admin/loginpage_wx')

if not os.path.exists("resource"):    os.mkdir("resource")
# 判断cookie是否存在，有则使用cookie跳过登录，无则手动登录if os.path.exists("resource/cookie.yaml"):    # 读取cookie，传入浏览器中    cookie = yaml.safe_load(open("resource/cookie.yaml"))    for i in cookie:        driver.add_cookie(i)    # 进入网站    driver.get("https://work.weixin.qq.com/wework_admin/frame#index")else:    # 扫码登录，等待页面加载    WebDriverWait(driver, 60).until(expected_conditions.url_contains("wework_admin/frame"))
    # 存入cookie信息到本地    cookie = driver.get_cookies()    with open("resource/cookie.yaml", "w") as f:        yaml.safe_dump(cookie, f)

time.sleep(3)driver.quit()