代码收藏家技术教程 2024-08-27

【Python】Selenium自动化脚本编写入门教程：源码案例详解（零基础保姆篇）

文章目录

准备工作

安装selenium

配置浏览器驱动

脚本测试

什么是Selenium？

Selenium功能示例(可直接拷贝执行)

编写一个简单的Python脚本来使用Selenium打开一个网页，并获取网页的标题

下面是一个示例，自动打开csdn并进行搜索

Selenium使用

Selenium元素定位

文本输入、清除与提交

获取页面内容

调整浏览器窗口尺寸

前进一页后退一页

页面刷新

窗口切换

frame/iframe内元素操作

获取标签元素的属性值（复选框）

下拉列表操作

弹窗操作

鼠标操作

键盘操作

JS代码执行

窗口截图

三种等待方法

更多相关内容可查看

准备工作

首先有python环境以及pycharm工具若需要下载：请参考【Python/PyCharm】卸载、安装、配置环境、及错误解决（保姆篇）

安装selenium

在pycharm终端运行如下命令,下图为安装成功

pip install selenium==4.3.0

pip list

配置浏览器驱动

下载浏览器驱动这里给出各浏览器（Chrome、Firefox、Edge等）的驱动下载地址，自行下载所需浏览器驱动

查看自己的浏览器版本，下载对应版本的驱动

我这里是126的Edge

将下载好的浏览器驱动放置在Python安装目录下（即Python环境变量配置目录）

脚本测试

这里是一个简单的脚本可以打开浏览器，三秒后关闭

import time
# 导入selenium包
from selenium import webdriver
# 打开Firefox浏览器
browser = webdriver.Edge()
# 停留三秒
time.sleep(3)
# 关闭浏览器
browser.quit()

到这里就可以打开页面了，例如一些自动抢票、自动帮你打开页面、或者自动帮助你日常需要干的事情、后面都会给出案例

什么是Selenium？

附官方文档：https://www.selenium.dev/zh-cn/documentation/webdriver/getting_started/first_script/

Selenium 是一个用于自动化浏览器操作的开源工具套件。它最初是为Web应用程序测试而创建的，但后来也被广泛用于网页数据抓取等任务。Selenium支持多种浏览器（如Chrome、Firefox、Safari等）和操作系统（Windows、Mac、Linux等），并提供了灵活的API来模拟用户在浏览器中的操作

Selenium功能示例(可直接拷贝执行)

编写一个简单的Python脚本来使用Selenium打开一个网页，并获取网页的标题

from selenium import webdriver

# 使用Chrome浏览器驱动程序
driver = webdriver.Edge()

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

# 获取网页标题并输出
print("Page title is:", driver.title)

# 关闭浏览器
driver.quit()

下面是一个示例，自动打开csdn并进行搜索

import time

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


def open_csdn():
    driver = webdriver.Edge()
    driver.get("https://blog.csdn.net/Aaaaaaatwl/article/details/140791747")
    input_text = WebDriverWait(driver, 30).until(
        EC.element_to_be_clickable((By.ID, "search-blog-words"))
    )
    input_text.send_keys("Python")

    password_login_tab = WebDriverWait(driver, 30).until(
        EC.element_to_be_clickable((By.CLASS_NAME, "btn-search-blog"))
    )
    password_login_tab.click()
    time.sleep(20)
    driver.quit()


if __name__ == "__main__":
    open_csdn()

上述代码是监听了一个输入框跟一个按钮

成功示例

Selenium使用

Selenium元素定位

我个人觉得定位是基本，就是如何找到以上案例中的输入框，按钮，剩下的其他就是最基本的输入输出api了

当使用Selenium通过元素的ID来定位元素时，可以使用By.ID选择器。以下是一个简单的示例代码，演示了如何使用Selenium定位并操作一个具有特定ID的元素：

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

# 创建一个.Edge WebDriver实例
driver = webdriver.Edge()

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过ID定位元素，并操作该元素
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.ID, "element-id"))
    )
    
    # 在找到的元素上执行操作，例如点击
    element.click()

finally:
    # 关闭浏览器
    driver.quit()

通过元素的名称（name属性）来定位元素时，可以使用By.NAME选择器。以下是一个示例代码

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

# 创建一个Edge WebDriver实例
driver = webdriver.Edge(executable_path='/path/to/your/msedgedriver.exe')

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过名称定位元素，并操作该元素
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.NAME, "element-name"))
    )
    
    # 在找到的元素上执行操作，例如点击
    element.click()

finally:
    # 关闭浏览器
    driver.quit()

3.通过类名（class name）定位元素可以使用By.CLASS_NAME选择器。以下是一个示例代码

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

# 创建一个Edge WebDriver实例
driver = webdriver.Edge(executable_path='/path/to/your/msedgedriver.exe')

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过类名定位元素，并操作该元素
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.CLASS_NAME, "element-class"))
    )
    
    # 在找到的元素上执行操作，例如点击
    element.click()

finally:
    # 关闭浏览器
    driver.quit()

通过标签名定位：如果你想定位页面上的第一个<a>标签，那么可以这样定位它：

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

# 创建一个Edge WebDriver实例
driver = webdriver.Edge(executable_path='/path/to/your/msedgedriver.exe')

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过标签名定位元素，并操作该元素
    elements = driver.find_elements(By.TAG_NAME, "a")
    
    # 输出所有链接的文本内容
    for element in elements:
        print(element.text)

finally:
    # 关闭浏览器
    driver.quit()

通过链接文本定位：如果页面上有一个链接，其文本是"About Us"，那么可以这样定位它：

# 创建一个Edge WebDriver实例（省略前文相同部分）

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过链接文本定位元素，并点击该链接
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.LINK_TEXT, "About Us"))
    )
    
    # 点击链接
    element.click()

finally:
    # 关闭浏览器
    driver.quit()

通过部分链接文本定位：如果页面上有一个链接，其文本是"Welcome to our website"，你可以使用部分文本来定位它：

# 创建一个Edge WebDriver实例（省略前文相同部分）

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过部分链接文本定位元素，并点击该链接
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.PARTIAL_LINK_TEXT, "Welcome"))
    )
    
    # 点击链接
    element.click()

finally:
    # 关闭浏览器
    driver.quit()

通过CSS选择器定位

# 创建一个Edge WebDriver实例（省略前文相同部分）

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过CSS选择器定位元素，并操作该元素
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.CSS_SELECTOR, "input#username"))
    )
    
    # 在找到的输入框中输入文本
    element.send_keys("username123")

finally:
    # 关闭浏览器
    driver.quit()

通过XPath定位-F12选中框右键即可复制XPath路径

# 创建一个Edge WebDriver实例（省略前文相同部分）

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

    # 等待页面加载完毕
    WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

    # 通过XPath定位元素，并操作该元素
    element = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.XPATH, "//input[@id='username']"))
    )
    
    # 在找到的输入框中输入文本
    element.send_keys("username123")

finally:
    # 关闭浏览器
    driver.quit()

By对象实例对象

class By:
    """
    Set of supported locator strategies.
    """

    ID = "id"
    XPATH = "xpath"
    LINK_TEXT = "link text"
    PARTIAL_LINK_TEXT = "partial link text"
    NAME = "name"
    TAG_NAME = "tag name"
    CLASS_NAME = "class name"
    CSS_SELECTOR = "css selector"

文本输入、清除与提交

send_keys(‘XXX’)文本输入。

clear()文本清空。

submit()提交：submit()只能用于包含属性type='submit’的标签，并且嵌套在form表单中。

import time
# 导入selenium包
from selenium import webdriver
from selenium.webdriver.common.by import By
# 启动并打开指定页面
browser = webdriver.Firefox()
browser.get("http://www.csdn.net")
# 通过css选择器定位输入框
input_text = browser.find_element(By.CSS_SELECTOR,'#toolbar-search-input')
# 输入文本
input_text.send_keys("selenium")
# 停留2秒
time.sleep(2)
# 清空文本
input_text.clear()
# 通过xpath选择器定位输入框,并输入CSDN
input_text = browser.find_element(By.XPATH,'//*[@id="sb_form_q"]').send_keys("CSDN")
time.sleep(2)
# 定位提交按钮后提交
browser.find_element(By.XPATH,'//*[@id="sb_form_go"]').submit()
# 停留三秒后关闭浏览器
time.sleep(3)
browser.quit()