百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT技术 > 正文

五分钟轻松掌握 Python 自动化测试 Selenium

wptr33 2025-03-12 21:10 24 浏览

一、Selenium

selenium 是一个老牌的测试工具和自动化工具,它的用途非常多,掌握这么技能绝对是值得,并且如果有需要切换其他的技术,大致应该是相似的,也就是说从它迁移到别的工具是轻松的。

二、为什么 Selenium 需要驱动器呢?

有些网站渲染的内容是在页面渲染之后,通过 js 加载数据然后更新内容。在此之前我们拿不到页面的内容。我们可以通过浏览器驱动器来操作浏览器获取到页面的内容。再来看看 Selenium 的架构:

三、浏览器驱动器

Selenium 在运行之前需要驱动器,下面是不同浏览器的驱动器:

  • chrome 驱动器下载地址
  • firefox 驱动器下载地址
  • edge 驱动器下载地址
  • safari 官方文档

下载的需要注意版本:不同的浏览器对应不同的, 考虑大概率会用 chrome 但是对于高于 Chrome 115 版本的, chrome 会给你一个 json 端点,自己找需要的版本下载。

下载后,记录文件位置 例如 windows 下: "D:\web-driver\chrome-win64\chrome.exe", chrome 的驱动器在 11.7M 左右,复制一份放在自己的目录下也不大。

四、在 Chrome 中使用 WebDriver 进行测试

selenium 核心就是要得到驱动器 driver,所有的操作都是在驱动器上使用的:

from selenium import webdriver

# 指定驱动程序的位置
driver_path = 'C:\Users\Administrator\Desktop\chromedriver'  # 修改为你的驱动程序路径

driver = webdriver.Chrome(executable_path=driver_path)

driver.get("http://example.com") # 打开网页

print(driver.title) # 打印页面标题

driver.quit() # 关闭浏览器

这段代码也很好解释,webdriver 使用 Chrome 驱动。驱动使用 get 方法获取页面内容,然后打印驱动对象上页面的 title, 然后驱动关闭六浏览器。

Chrome 驱动的参数 Opiton

前面我们使用 Chrome 驱动,其实 Chrome 驱动还可以传递众多参数,驱动时 chrome opiton 的一些配置配置参数:

chrome_options.add_argument("--headless") # 无头模式
chrome_options.add_argument("--disable-gpu") # 禁用 gpu 加速
chrome_options.add_argument("--window-size=1920,1080") # 指定窗口大小
chrome_options.add_argument("--disable-extensions") # 禁用扩展
chrome_options.add_argument("--no-sandbox") # 禁用沙盒
chrome_options.add_argument("--ignore-certificate-errors") # 忽略证书错误
chrome_options.add_argument("--proxy-server=http://proxyserver:port") # 设置代理
chrome_options.add_argument("--disable-popup-blocking") #禁用弹出窗口阻止
chrome_options.add_argument("--start-maximized") # 启动时最大化窗口
chrome_options.add_argument("--disable-notifications") # 禁用通知
chrome_options.add_argument("--disable-infobars") #禁用信息栏
chrome_options.add_argument("--user-data-dir=/path/to/profile") # 指定用户数据目录
chrome_options.add_argument("--disable-restore-session-state") # 禁用崩溃恢复提示
chrome_options.add_argument("--mute-audio") # 静音音频

五、页面动态内容的加载

Selenium 处理动态内容是常见的需求,等待 JavaScript 动态加载。如果没有等待内容加载完毕,那么查找元素可能会导致错误。原因也很简单,就是没有加载完成。

等待也可以分为不同的层级,web deriver API 级别的。Python 等待以及页面级的等待。

WebDriverWait 等待

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 初始化 WebDriver
driver = webdriver.Chrome()

# 打开网页
driver.get('http://example.com')

# 显式等待,等待某个元素可见
element = WebDriverWait(driver, 10).until(
    EC.visibility_of_element_located((By.ID, 'dynamicElementId'))
)

# 对元素执行操作
element.click()

driver.find_element 显示等待

driver = webdriver.Chrome()
driver.implicitly_wait(10)  # 全局等待 10 秒

driver.get('http://example.com')

# 如果元素还没准备好,WebDriver 将等待,直到找到元素或超时
element = driver.find_element(By.ID, 'dynamicElementId')
element.click()

time.sleep() Python 等待(不推荐)

import time

driver = webdriver.Chrome()
driver.get('http://example.com')

time.sleep(5)  # 强制等待 5 秒
element = driver.find_element(By.ID, 'dynamicElementId')
element.click()

页面级等待

from selenium.webdriver.support.ui import WebDriverWait

# 等待页面完全加载,通过检查 JavaScript 变量
WebDriverWait(driver, 10).until(
    lambda driver: driver.execute_script('return document.readyState') == 'complete'
)

有了对 selenium 的基础使用和认识,我们就需要较为全面的了解 selenium 的 api 了,当我们对其 api 有较为全面的了解,我们很快就能入手了。

六、selenium 浏览器操作

  • driver.get(url) 打开指定的 URL。
  • driver.quit() 关闭浏览器并结束 WebDriver 会话。
  • driver.close() 关闭当前窗口。
  • driver.maximize_window() 最大化浏览器窗口。
  • driver.execute_script(script) 执行 JavaScript 脚本。

七、元素定位

  • driver.find_element(By.ID, id) 通过 ID 定位元素。
  • driver.find_element(By.NAME, name) 通过 NAME 定位元素。
  • driver.find_element(By.XPATH, xpath) 通过 XPath 定位元素。
  • driver.find_element(By.CSS_SELECTOR, selector) 通过 CSS 选择器定位元素。
  • driver.find_elements(By.CLASS_NAME, class_name) 定位所有符合类名的元素,返回列表。

八、元素操作

  • element.click() 点击元素。
  • element.send_keys(keys) 向元素输入文本。
  • element.clear() 清空输入框的内容。
  • element.get_attribute(attribute) 获取元素的属性值。
  • element.get_text() 获取元素的文本内容。
  • element.is_displayed() 检查元素是否显示。

九、等待操作

  • driver.implicitly_wait(time) 设置全局隐式等待时间。
  • WebDriverWait(driver, time).until(condition) 显式等待指定的条件。

十、窗口与导航

  • driver.back() 导航到前一个页面。
  • driver.forward() 导航到下一个页面。
  • driver.refresh() 刷新当前页面。
  • driver.switch_to.frame(frame) 切换到指定的 iframe。
  • driver.switch_to.alert() 切换到 alert 对话框。
  • driver.switch_to.window(window_name) 切换到指定的窗口。

十一、常用功能

处理 Cookie

cookies = driver.get_cookies()
driver.add_cookie({'name': 'cookie_name', 'value': 'cookie_value'})
driver.delete_all_cookies()

截图

driver.save_screenshot('screenshot.png')

上传图片

file_input = driver.find_element(By.NAME, 'file_upload') file_input.send_keys('/path/to/file.txt')

执行 JS

# 执行 js 脚本
driver.execute_script("alert('Hello, World!');")

# 从 localstorage 中获取 token
token = driver.execute_script("return localStorage.getItem('auth_token');") print(f"Token: {token}")

# 获取元素 css 属性
element = driver.find_element(By.ID, 'element_id')
background_color = driver.execute_script("return window.getComputedStyle(arguments[0]).backgroundColor;", element)

十二、Selenium 与 XPATH

虽然页面里面使用 css 选择器更加靠近 web 端,但是 xpath 在查找元素和定位时更加直接,简单高效。前面已经介绍 api 了,下面是一个实例:

from selenium import webdriver
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get("https://example.com")

# 使用 XPath 查找元素
element = driver.find_element(By.XPATH, "//div[@id='main-content']//h1")
print(element.text)

find_element 第一个参数指定 By.XPATH,后面传入 xpath 字符串即可,也是特别方便。

十三、selenium 与爬虫

得益于 selenium 能够直接访问实际页面,动态加载数据爬虫也有的重要作用。它可以弥补传统爬虫工具的不足,抓取由 JavaScript 渲染的内容。但要注意的是,它的性能相对较低,不适合大规模数据抓取。

十四、小结

本文主要介绍了 Selenium 相关内容,主要涉及 Selenium 知识面,从开始的 Python 小案例,到后面的 API 全面了解,以及 Selenium 的常用功能,到最后的 XPATH 以及爬虫的认知。这些内容已经能够全面,且具有实践性。

作者:编程杂货铺
链接:
https://juejin.cn/post/7402328037390123071

相关推荐

oracle数据导入导出_oracle数据导入导出工具

关于oracle的数据导入导出,这个功能的使用场景,一般是换服务环境,把原先的oracle数据导入到另外一台oracle数据库,或者导出备份使用。只不过oracle的导入导出命令不好记忆,稍稍有点复杂...

继续学习Python中的while true/break语句

上次讲到if语句的用法,大家在微信公众号问了小编很多问题,那么小编在这几种解决一下,1.else和elif是子模块,不能单独使用2.一个if语句中可以包括很多个elif语句,但结尾只能有一个else解...

python continue和break的区别_python中break语句和continue语句的区别

python中循环语句经常会使用continue和break,那么这2者的区别是?continue是跳出本次循环,进行下一次循环;break是跳出整个循环;例如:...

简单学Python——关键字6——break和continue

Python退出循环,有break语句和continue语句两种实现方式。break语句和continue语句的区别:break语句作用是终止循环。continue语句作用是跳出本轮循环,继续下一次循...

2-1,0基础学Python之 break退出循环、 continue继续循环 多重循

用for循环或者while循环时,如果要在循环体内直接退出循环,可以使用break语句。比如计算1至100的整数和,我们用while来实现:sum=0x=1whileTrue...

Python 中 break 和 continue 傻傻分不清

大家好啊,我是大田。今天分享一下break和continue在代码中的执行效果是什么,进一步区分出二者的区别。一、continue例1:当小明3岁时不打印年龄,其余年龄正常循环打印。可以看...

python中的流程控制语句:continue、break 和 return使用方法

Python中,continue、break和return是控制流程的关键语句,用于在循环或函数中提前退出或跳过某些操作。它们的用途和区别如下:1.continue(跳过当前循环的剩余部分,进...

L017:continue和break - 教程文案

continue和break在Python中,continue和break是用于控制循环(如for和while)执行流程的关键字,它们的作用如下:1.continue:跳过当前迭代,...

作为前端开发者,你都经历过怎样的面试?

已经裸辞1个月了,最近开始投简历找工作,遇到各种各样的面试,今天分享一下。其实在职的时候也做过面试官,面试官时,感觉自己问的问题很难区分候选人的能力,最好的办法就是看看候选人的github上的代码仓库...

面试被问 const 是否不可变?这样回答才显功底

作为前端开发者,我在学习ES6特性时,总被const的"善变"搞得一头雾水——为什么用const声明的数组还能push元素?为什么基本类型赋值就会报错?直到翻遍MDN文档、对着内存图反...

2023金九银十必看前端面试题!2w字精品!

导文2023金九银十必看前端面试题!金九银十黄金期来了想要跳槽的小伙伴快来看啊CSS1.请解释CSS的盒模型是什么,并描述其组成部分。答案:CSS的盒模型是用于布局和定位元素的概念。它由内容区域...

前端面试总结_前端面试题整理

记得当时大二的时候,看到实验室的学长学姐忙于各种春招,有些收获了大厂offer,有些还在苦苦面试,其实那时候的心里还蛮忐忑的,不知道自己大三的时候会是什么样的一个水平,所以从19年的寒假放完,大二下学...

由浅入深,66条JavaScript面试知识点(七)

作者:JakeZhang转发链接:https://juejin.im/post/5ef8377f6fb9a07e693a6061目录由浅入深,66条JavaScript面试知识点(一)由浅入深,66...

2024前端面试真题之—VUE篇_前端面试题vue2020及答案

添加图片注释,不超过140字(可选)1.vue的生命周期有哪些及每个生命周期做了什么?beforeCreate是newVue()之后触发的第一个钩子,在当前阶段data、methods、com...

今年最常见的前端面试题,你会做几道?

在面试或招聘前端开发人员时,期望、现实和需求之间总是存在着巨大差距。面试其实是一个交流想法的地方,挑战人们的思考方式,并客观地分析给定的问题。可以通过面试了解人们如何做出决策,了解一个人对技术和解决问...