1. 什么是DrissionPage？
DrissionPage 是一个基于 Python 的网页自动化工具，主要用于网页数据的抓取、自动化操作和测试。它结合了 Selenium 和 Requests 的优点，提供了更简洁、高效的 API，适用于各种网页自动化任务。
2. 安装 DrissionPage
在开始使用 DrissionPage 之前，需要先安装它。可以通过 pip 进行安装：pip install DrissionPage。然后创建项目目录并初始化配置文件：
mkdir drission_project
cd drission_project
drission init
初始化后会生成以下文件：
config.ini：配置文件（存放浏览器驱动路径、测试环境等）pages/：页面对象目录tests/：测试用例目录utils/：工具类目录
3. 基本使用
3.1 初始化浏览器
DrissionPage 支持多种浏览器，包括 Chrome、Firefox 等。初始化浏览器的代码如下：
from DrissionPage import WebPage
# 初始化 Chrome 浏览器
page = WebPage('chrome')
3.2 打开网页
使用 get() 方法可以打开指定的网页：
page.get('https://www.example.com')
3.3 查找元素
DrissionPage 提供了多种查找元素的方法，常用的有 ele() 和 eles()：
ele()：查找单个元素。eles()：查找多个元素。
# 查找单个元素
element = page.ele('#element-id')
# 查找多个元素
elements = page.eles('.class-name')
3.4 操作元素
找到元素后，可以对其进行各种操作，如点击、输入文本等：
# 点击元素
element.click()
# 输入文本
element.input('Hello, DrissionPage!')
3.5 获取元素属性
可以通过 attr() 方法获取元素的属性：
# 获取元素的 href 属性
href = element.attr('href')
3.6 执行 JavaScript
DrissionPage 允许直接执行 JavaScript 代码：
# 执行 JavaScript
page.run_js('alert("Hello, DrissionPage!");')
3.7 页面对象（Page Object）
每个 Web 页面对应一个 Python 类，封装页面元素和操作方法。
# pages/login_page.py
from drissionpage import Page
class LoginPage(Page):
def __init__(self, driver):
super().__init__(driver)
self.username_input = self.by_id("username") # 定位元素
self.password_input = self.by_id("password")
self.login_button = self.by_id("login-btn")
def login(self, username, password):
self.username_input.send_keys(username)
self.password_input.send_keys(password)
self.login_button.click()
3.8 元素定位方式
支持多种定位策略：
| 定位方式 | 语法 |
|---|---|
ID |
self.by_id(“element_id”) |
Name |
self.by_name(“name”) |
XPath |
self.by_xpath(“//tag[@id]”) |
CSS Selector |
self.by_css(“div.class”) |
Link Text |
self.by_link_text(“Link Text”) |
3.9 常用操作方法
# 输入文本
self.username_input.send_keys("test_user")
# 点击元素
self.login_button.click()
# 获取文本
assert self.success_msg.text == "Login Success"
# 切换窗口/标签页
self.switch_to.window(handle)
# 执行 JavaScript
self.execute_script("window.scrollTo(0, document.body.scrollHeight);")
3.10 断言
集成 unittest 或 pytest 的断言方式：
from drissionpage import assert_
# 验证元素是否存在
assert_.is_present(self.username_input)
# 验证文本是否匹配
assert_.text_equals(self.login_button, "Login")
4. 高级功能
4.1 处理弹窗
DrissionPage 提供了处理弹窗的功能，可以轻松应对各种弹窗场景：
# 处理弹窗
page.handle_alert(accept=True)
4.2 切换 Frame
如果网页中有 iframe，可以使用 switch_to_frame() 方法切换到指定的 frame：
# 切换到 iframe
page.switch_to_frame('iframe-name')
4.3 等待元素
DrissionPage 提供了等待元素出现或消失的功能：
# 等待元素出现
page.wait_ele('#element-id')
# 等待元素消失
page.wait_ele_disappear('#element-id')
4.4 截图
可以轻松地对网页进行截图：
# 截图并保存
page.screenshot('screenshot.png')
4.5 数据驱动测试
结合 CSV 或 Excel 文件实现多组数据测试：
# tests/test_login_dataDriven.py
import csv
from drissionpage import TestCase
from pages.login_page import LoginPage
class LoginTestCase(TestCase):
@classmethod
def get_test_data(cls):
data = []
with open("testdata.csv") as f:
reader = csv.DictReader(f)
for row in reader:
data.append((row["username"], row["password"]))
return data
def test_login(self, username, password):
self.open("http://example.com/login")
login_page = LoginPage(self.driver)
login_page.login(username, password)
assert_.is_present(login_page.success_msg)
5. 实战示例
5.1 自动登录
以下是一个自动登录的示例：
from DrissionPage import WebPage
# 初始化浏览器
page = WebPage('chrome')
# 打开登录页面
page.get('https://www.example.com/login')
# 输入用户名和密码
page.ele('#username').input('your-username')
page.ele('#password').input('your-password')
# 点击登录按钮
page.ele('#login-button').click()
# 等待登录成功
page.wait_ele('#welcome-message')
# 截图保存
page.screenshot('login_success.png')
5.2 数据抓取
以下是一个简单的数据抓取示例：
from DrissionPage import WebPage
# 初始化浏览器
page = WebPage('chrome')
# 打开目标网页
page.get('https://www.example.com/data')
# 获取所有数据项
data_items = page.eles('.data-item')
# 遍历并打印数据
for item in data_items:
print(item.text)
# 关闭浏览器
page.quit()
若元素定位失败，则检查定位表达式是否正确。或者添加显式等待：
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
element = WebDriverWait(self.driver, 10).until(
EC.presence_of_element_located((By.ID, "username"))
)
5.3 编写测试用例
# tests/test_login.py
from drissionpage import TestCase
from pages.login_page import LoginPage
class LoginTestCase(TestCase):
def test_valid_login(self):
# 打开登录页面
self.open("http://example.com/login")
# 创建页面对象
login_page = LoginPage(self.driver)
# 执行登录操作
login_page.login("test_user", "test_password")
# 验证结果
assert_.is_present(login_page.success_msg)
assert_.text_equals(login_page.success_msg, "Login Success")
然后运行测试：drission run tests/test_login.py
6. 最佳实践
6.1 页面对象设计原则
- 每个页面对应一个类，封装页面元素和操作。
- 方法命名清晰（如 login()、submit_form()）。
6.2 元素定位优化：
- 避免使用易变动的属性（如 class 名称）。
- 优先使用 ID 或 name，其次使用相对稳定的 XPath 或 CSS Selector。
6.3 异常处理：
try:
self.login_button.click()
except NoSuchElementException:
self.log_error("Login button not found")
6.4 环境配置：
使用 config.ini 管理不同环境（开发/生产）的 URL 和驱动路径。
7. 总结
DrissionPage 是一个功能强大且易于使用的网页自动化工具，适用于各种网页自动化任务。希望本文的 DrissionPage 的基本使用方法和一些高级功能能够帮助到你。