Selenium 保存会话信息避免重复登录实战!

前言

  • • 在一些实际开发场景中,我们在使用 Selenium 做自动化测试时需要保留用户的会话信息,从而避免重复登录,今天这篇文章就带大家实战如何使用 Selenium 保存会话信息。

版本

  • • Python 3.x

整体思路

  • • 当我们打开页面时,检测会话信息文件是否存在,如果存在直接加载会话信息,如果不存在,则进行登录并保存会话信息到本地文件中。

实践

  • • 当我们第一次打开页面时,登录并保存会话信息。

    from selenium import webdriver
    from selenium.webdriver.common.keys import Keys
    import pickle

    第一次登录

    driver = webdriver.Chrome()

    driver.get("https://example.com/login")

    使用自动填充或其它方式登录

    username_input = driver.find_element_by_name("username")
    password_input = driver.find_element_by_name("password")
    username_input.send_keys("your_username")
    password_input.send_keys("your_password")
    password_input.send_keys(Keys.RETURN)

    登录成功后将driver.get_cookies()返回的cookies保存到名为"cookies.pkl"的文件中

    pickle.dump(driver.get_cookies(), open("cookies.pkl", "wb"))

    关闭 driver

    driver.close()

    第二次访问

    driver = webdriver.Chrome()

    这里需要提前访问一下再加载 cookies 否则会出现 InvalidCookieDomainException: Message: invalid cookie domain

    driver.get("https://example.com/login")

    加载 "cookies.pkl" 中的会话信息

    cookies = pickle.load(open("cookies.pkl", "rb"))
    for cookie in cookies:
    driver.add_cookie(cookie)

    携带会话信息访问

    driver.get("https://example.com/xxx")

问题

  • • 谷歌默认加载使用的是 data domain,所以需要先访问目标页面切换到对应的 domain:

    driver.get("https://example.com/login")

    加载 "cookies.pkl" 中的会话信息

    cookies = pickle.load(open("cookies.pkl", "rb"))
    for cookie in cookies:
    driver.add_cookie(cookie)

  • • 除此之外,如果你访问的目标页面为一级域名,而保存的 cookies 中存在二级域名的 domain,那么也会出现 invalid cookie domain 的问题,你可以将 cookiedomain 二级域名 domain 转换为 一级域名 domain 再进行加载,例如:

    cookies domain:
    .example.com
    .example.com
    xxx.example.com
    xxx.example.com

    那么需要全部调整为:
    .example.com
    .example.com
    .example.com
    .example.com

最后感谢每一个认真阅读我文章的人,看着粉丝一路的上涨和关注,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走!

软件测试面试文档

我们学习必然是为了找到高薪的工作,下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料,并且有字节大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。

相关推荐
Angel Q.4 小时前
特征提取 | DINO 到 DINOv3
图像处理·python
不如语冰5 小时前
AI大模型入门-pytorch-张量2 tensor创建
python
NPE~5 小时前
[AI]Agent开发——ADK框架使用
人工智能·python·ai·教程·adk·agent开发
矮个史蒂芬5 小时前
统计verilog .v文件中output 的bit数
python
gwf2165 小时前
磨损均衡算法(Wear Leveling)——SSD如何让每块闪存“公平退休“?
运维·数据库·人工智能·python·嵌入式硬件·算法·智能硬件
爱吃提升5 小时前
python 分布式爬虫、爬虫合规与综合实战项目
分布式·爬虫·python
审小匠OpenCPAi5 小时前
银行流水核查怎么自动化?单边匹配、双向勾稽与图聚类异常检测的工程对比
java·前端·人工智能·python·审计
元Y亨H7 小时前
Pandas 解析 Excel 导致的内存溢出(MemoryError)
python·excel
金銀銅鐵7 小时前
[Python] 用 turtle 来绘制瑞典国旗
python
Albert Edison7 小时前
【GUI 自动化测试】Pywinauto 常见操作
自动化测试·python·pywinauto