DrissionPage自动化框架学习
DrissionPage学习
官网：http://drissionpage.cn/features/intimate/
1.DrissionPage事件
1.1元素定位查找
# 根据属性查找，@ 后面可跟任意属性
page.ele('@id:ele_id', timeout=2) # 查找 id 为 ele_id 的元素，设置等待时间2秒
page.eles('@class') # 查找所有拥有 class 属性的元素
page.eles('@class:class_name') # 查找所有 class 含有 ele_class 的元素
page.eles('@class=class_name') # 查找所有 class 等于 ele_class 的元素
# 根据 class 或 id 查找
page.ele('#ele_id') # 等价于 page.ele('@id=ele_id')
page.ele('#:ele_id') # 等价于 page.ele('@id:ele_id')
page.ele('.ele_class') # 等价于 page.ele('@class=ele_class')
page.ele('.:ele_class') # 等价于 page.ele('@class:ele_class')
# 根据 tag name 查找
page.ele('tag:li') # 查找第一个 li 元素
page.eles('tag:li') # 查找所有 li 元素
# 根据 tag name 及属性查找
page.ele('tag:div@class=div_class') # 查找 class 为 div_class 的 div 元素
page.ele('tag:div@class:ele_class') # 查找 class 含有 ele_class 的 div 元素
page.ele('tag:div@class=ele_class') # 查找 class 等于 ele_class 的 div 元素
page.ele('tag:div@text():search_text') # 查找文本含有 search_text 的 div 元素
page.ele('tag:div@text()=search_text') # 查找文本等于 search_text 的 div 元素
# 根据文本内容查找
page.ele('search text') # 查找包含传入文本的元素
page.eles('text:search text') # 如文本以 @、tag:、css:、xpath:、text: 开头，则应在前加上 text: 避免冲突
page.eles('text=search text') # 文本等于 search_text 的元素
# 根据 xpath 或 css selector 查找
page.eles('xpath://div[@class="ele_class"]')
page.eles('css:div.ele_class')
# 根据 loc 查找
loc1 = By.ID, 'ele_id'
loc2 = By.XPATH, '//div[@class="ele_class"]'
page.ele(loc1)
page.ele(loc2)
# 查找下级元素
element = page.ele('@id:ele_id')
element.ele('@class:class_name') # 在 element 下级查找第一个 class 为 ele_class 的元素
element.eles('tag:li') # 在 ele_id 下级查找所有li元素
# 根据位置查找
element.parent # 父元素
element.next # 下一个兄弟元素
element.prev # 上一个兄弟元素
# 获取 shadow-root，把它作为元素对待。只支持 open 的 shadow-root
ele1 = element.shadow_root.ele('tag:div')
# 串连查找
page.ele('@id:ele_id').ele('tag:div').next.ele('some text').eles('tag:a')
# 简化写法
eles = page('@id:ele_id')('tag:div').next('some text').eles('tag:a')
ele2 = ele1('tag:li').next('some text')
1.2元素操作
element.click(by_js) # 点击元素，可选择是否用 js 方式点击
element.input(value) # 输入文本
element.run_script(js) # 对元素运行 JavaScript 脚本
element.submit() # 提交
element.clear() # 清空元素
element.screenshot(path, filename) # 对元素截图
element.select(text) # 根据文本选择下拉列表
element.set_attr(attr, value) # 设置元素属性值
element.remove_attr(attr) # 删除属性
element.drag(x, y, speed, shake) # 拖动元素相对距离，可设置速度和是否随机抖动
element.drag_to(ele_or_loc, speed, shake) # 拖动元素到另一个元素或某个坐标，可设置速度和是否随机抖动
element.hover() # 在元素上悬停鼠标
1.3获取元素属性
element.html # 返回元素 outerHTML
element.inner_html # 返回元素 innerHTML
element.tag # 返回元素 tag name
element.text # 返回元素 innerText 值
element.comments # 返回元素内注释列表
element.link # 返回元素 href 或 src 绝对 url
element.texts() # 返回元素内所有直接子节点的文本，包括元素和文本节点，可指定只返回文本节点
element.attrs # 返回元素所有属性的字典
element.attr(attr) # 返回元素指定属性的值
element.css_path # 返回元素绝对 css 路径
element.xpath # 返回元素绝对 xpath 路径
element.parent # 返回元素父元素
element.next # 返回元素后一个兄弟元素
element.prev # 返回元素前一个兄弟元素
element.parents(num) # 返回第 num 级父元素
element.nexts(num, mode) # 返回后面第几个元素或节点
element.prevs(num, mode) # 返回前面第几个元素或节点
element.ele(loc_or_str, timeout) # 返回当前元素下级第一个符合条件的子元素、属性或节点文本
element.eles(loc_or_str, timeout) # 返回当前元素下级所有符合条件的子元素、属性或节点文本
2.DrissionPage基本使用
2.1控制浏览器，导入ChromiumPage：
from DrissionPage import ChromiumPage
2.1.1 示例代码1：
from DrissionPage import ChromiumPage
# 创建页面对象，并启动或接管浏览器
page = ChromiumPage()
# 跳转到登录页面
page.get('https://gitee.com/login') # get()方法用于访问参数中的网址。它会等待页面完全加载，再继续执行后面的代码。
# 定位到账号文本框
ele = page.ele('#user_login') # ele()方法用于查找元素,'#user_login'是定位符文本，#意思是按id属性查找元素。ele()内置了等待，如果元素未加载，它会执行等待，直到元素出现或到达时限。默认超时时间 10 秒。
# 输入对文本框输入账号
ele.input('账号')
# 定位到密码文本框并输入密码
page.ele('#user_password').input('密码')
# 点击登录按钮
page.ele('@value=登 录').click() # @表示按属性名查找
2.1.2示例代码2(Recorder是追加写入)：
from DrissionPage import ChromiumPage
from DataRecorder import Recorder # 这个是追加写入
from time import sleep
# 创建页面对象
page = ChromiumPage()
# 创建记录器对象
recorder = Recorder('data.csv')
recorder.add_data(("标题", "导演", "上映时间", "评分"))
# 访问网页
page.get('https://www.maoyan.com/board/4')
while True:
# 遍历页面上所有 dd 元素
for mov in page.eles('t:dd'):
# 获取须要的信息
score = mov('.score').text
title = mov('@data-act=boarditem-click').attr('title')
star = mov('.star').text
time = mov('.releasetime').text
# 写入到记录器
recorder.add_data((title, star, time, score))
sleep(1)
# 获取下一页按钮，有就点击
btn = page('下一页', timeout=2)
if btn:
btn.click()
page.wait.load_start()
# 没有则退出程序
else:
break
recorder.record()
2.1.3 示例代码3
#抓取震坤行商品
from DrissionPage import ChromiumPage
import time
page = ChromiumPage()
# page = ChromiumPage()这句意思是打开一个浏览器，这个浏览器是安装DrissionPage时自动安装的
# 跳转到震坤行页面
page.get('https://www.zkh.com/?utm_source=baidu&utm_medium=CPT')
time.sleep(1)
page.ele('xpath://*[@id="app"]/div/div/div[3]/div/div[1]/div[1]/input').input("口罩")
time.sleep(1)
page.ele('xpath://*[@id="app"]/div/div/div[3]/div/div[1]/button').click()
# page.to_tab(page.latest_tab) goods-title
while True:
page.wait.load_start()
page.scroll.to_location(0,10000) #滚动到指定位置
time.sleep(0.5)
page.scroll.to_location(0, 10000)
time.sleep(0.5)
page.scroll.to_location(0, 10000)
time.sleep(0.5)
divs=page.eles('xpath://div[@class="goods-name clamp2"]')
for i in divs:
title = i.attr('title')
print(title)
2.2收发数据包，导入SessionPage：
from DrissionPage import SessionPage
2.2.1示例代码：
from DrissionPage import SessionPage
# 创建页面对象
page = SessionPage()
# 爬取前5页
for i in range(1, 6):
page.get(f'https://gitee.com/explore/all?page={i}')
# 获取所有开源库<a>元素列表
links = page.eles('.title project-namespace-path') # 页面对象的eles()获取页面中所有class属性为'title project-namespace-path'的元素对象，eles()方法用于查找多个符合条件的元素，返回由它们组成的list
# 遍历所有<a>元素
for link in links:
# 打印链接信息
print(f"{link.text}:{link.link}") # .text获取元素的文本，.link获取元素的href或src属性
2.2.2下载网页
from DrissionPage import SessionPage
url = 'https://www.baidu.com/img/flexible/logo/pc/result.png'
save_path = r'images' # 保存的路径，当前的项目路径，如为空则保存在与py同目录
page = SessionPage()
page.download(url, save_path, 'img') # 支持重命名，处理文件名冲突，img是图片名字
3.获取新窗口句柄
3.1.无需切入切出，逻辑清晰
使用过 selenium 的人都知道，selenium 同一时间只能操作一个标签页或<iframe>元素，要操作其它标签页，或<iframe>元素，需要用switch_to() 方法切换，操作后还要切换回来。如果是多级<iframe>，还要逐级切入，相当麻烦。
DrissionPage 则无需这些麻烦的操作，它把每个标签页和<iframe>都看作独立的对象，可以同时并发操作，而且可以直接跨多层<iframe>获取里面的元素，然后直接处理，非常方便。
对比一下，获取 2 层<iframe>内一个 id 为'div1'的元素：
# 使用 selenium
driver.switch_to.frame(0)
driver.switch_to.frame(0)
ele = driver.find_element(By.ID, 'div1')
driver.switch_to.default_content()
# 使用 DrissionPage
ele = page('#div1')
多标签页同时操作，selenium 无此功能：
tab1 = page.get_tab(1)
tab2 = page.get_tab(2)
tab1.get('https://www.baidu.com')
tab2.get('https://www.163.com')
3.2 示例代码1
#登录百度爬取
from DrissionPage import ChromiumPage
import time
page = ChromiumPage()这句意思是打开一个浏览器，这个浏览器是安装DrissionPage时自动安装的
# 跳转百度页面
page.get('https://www.baidu.com')
time.sleep(1)
page.ele('xpath://*[@id="kw"]').input("宿舍")
time.sleep(1)
page.ele('xpath://*[@id="su"]').click()
time.sleep(1)
page.ele('xpath://*[@id="3"]/div/div[1]/h3/a').click()
#切换到新页面
ab=page.get_tab(page.latest_tab) #获取新窗口页面句柄
# 等待页面跳转
page.wait.load_start()
# ab = page.geqwt_tab(0) #这样也可以获取新框框页面句柄
print(ab.title) #获取页面标题
# exit()
#切换到新页面
# ab=page.get_tab(1)
# page.get_tab(page.latest_tab)
# print(ab.title)
# page.get_tab(page.latest_tab)
# print(page.title)
# exit()
time.sleep(1)
#通过用新窗口的句柄来进行操作
ab.ele('xpath:/html/body/div[1]/div[3]/form/div/input[4]').input("家")
ab.ele('xpath:/html/body/div[1]/div[3]/form/div/span').click()
lis = ab.ele('xpath:/html/body/div[2]/div[1]/div/div[2]/ul')
# lis = page.ele('xpath://ul[@class="w4]')
# print(lis)
lis1 = lis.eles('tag:li')
for i in lis1:
lis2 = i.eles('tag:a')
for j in lis2:
href=j.attr('href') #获取href属性值
# title=i.text
title=j.attr('title') #获取title属性值
print(href,title)
time.sleep(4)
# ab.close() #关闭当前页面 （这样也可以）
page.close_tabs(ab) #关闭当前页面
# ac=page.get_tab(-1)
ac=page.get_tab(page.latest_tab) #将句柄调回原来的页面中
print(ac.title)
time.sleep(1)
ac.ele('xpath://*[@id="6"]/div/div[1]/h3/a').click()
time.sleep(2)
page.quit() #退出浏览器
4.与selenium对比
4.1跳转到一个标签页
# 使用 selenium：
driver.switch_to.window(driver.window_handles[0])
# 使用 DrissionPage：
tab = page.get_tab(1)
4.2按文本选择下拉列表
# 使用 selenium：
from selenium.webdriver.support.select import Select
select_element = Select(element)
select_element.select_by_visible_text('text')
# 使用 DrissionPage：
element.select('text')
4.3拖拽一个元素
# 使用 selenium：
ActionChains(driver).drag_and_drop(ele1, ele2).perform()
# 使用 DrissionPage：
ele1.drag_to(ele2)
4.4滚动窗口到底部（保持水平滚动条不变）
# 使用 selenium：
driver.execute_script("window.scrollTo(document.documentElement.scrollLeft, document.body.scrollHeight);")
# 使用 DrissionPage：
page.scroll.to_bottom()
4.5设置 headless 模式
# 使用 selenium：
options = webdriver.ChromeOptions()
options.add_argument("--headless")
# 使用 DrissionPage：
set_headless(True)
4.6获取伪元素内容
# 使用 selenium：
text = webdriver.execute_script('return window.getComputedStyle(arguments[0], "::after").getPropertyValue("content");', element)
# 使用 DrissionPage：
text = element.pseudo.after
4.7用 xpath 直接获取属性或文本节点（返回文本）
# 使用 selenium：
相当复杂
# 使用 DrissionPage：
class_name = element('xpath://div[@id="div_id"]/@class')
text = element('xpath://div[@id="div_id"]/text()[2]')
5.模式切换
http://drissionpage.cn/features/features_demos/change_mode/
用浏览器登录网站，然后切换到 requests 读取网页。两者会共享登录信息。
from DrissionPage import WebPage
from time import sleep
# 创建页面对象，默认 d 模式
page = WebPage()
# 访问个人中心页面（未登录，重定向到登录页面）
page.get('https://gitee.com/profile')
# 输入账号密码登录
page.ele('@id:user_login').input('your_user_name')
page.ele('@id:user_password').input('your_password\n')
page.wait.load_start()
# 切换到 s 模式
page.change_mode()
# 登录后 session 模式的输出
print('登录后title：', page.title, '\n')
#输出：
登录后title： 个人资料 - 码云 Gitee.com
获取登录后的元素
# 接上段代码
foot = page.ele('#footer-left') # 用 id 查找元素
first_col = foot.ele('css:>div') # 使用 css selector 在元素的下级中查找元素（第一个）
lnk = first_col.ele('text:命令学') # 使用文本内容查找元素
text = lnk.text # 获取元素文本
href = lnk.attr('href') # 获取元素属性值
print(text, href, '\n')
# 简洁模式串联查找
text = page('@id:footer-left')('css:>div')('text:命令学').text
print(text)
#输出
Git 命令学习 https://oschina.gitee.io/learn-git-branching/
Git 命令学习