零操作,高效下载:利用Playwright和Python完成文件下载
前言
Playwright 是一个由 Microsoft 开发的跨浏览器自动化测试工具,它提供了强大的功能来模拟用户在浏览器中的操作。Playwright 还可以应用于诸如文件下载等实际场景的测试。本文将介绍如何利用 Playwright 在 Python 中实现文件下载功能。
文件下载
文件下载也是我们进行web自动化测试时经常需要面对的场景,如果我们使用的是selenium,会出现无法处理Windows系统弹出的对话框的情况,需要借助autoit等第三方工具来实现下载功能,playwright则可以不借助其他工具来实现文件自动下载。
playwright提供了expect_download()
操作来实现文件的下载操作,当浏览器上下文关闭时,所有属于浏览器上下文的下载文件都会被删除。
下载开始后会发出下载事件。下载完成后,下载路径可用:
with page.expect_download() as download_info:
page.get_by_text("Download file").click()
download = download_info.value
# wait for download to complete
path = download.path()
相关操作
- 取消下载
如果下载已经完成或取消,则不会失败。成功取消后,download.failure()将解析为'canceled'
download.cancel()
- .删除下载的文件
download.delete()
- 返回下载错误(如果有)
download.failure()
- 获取下载所属的页面
download.page
- 下载路径
下载成功,则返回下载文件的路径。如有必要,该方法将等待下载完成。该方法在远程连接时抛出。
download.path()
注:下载的文件名是随机 GUID,使用download.suggested_filename获取建议的文件名
- 将下载复制到用户指定的路径。在下载仍在进行时调用此方法是安全的。如有必要,将等待下载完成。
download.save_as(path)
- 返回此下载的建议文件名
download.suggested_filename
注:通常由浏览器根据Content-Disposition响应标头或download属性计算得出
- 返回下载的 url
download.url
实例
我们以下载我们常用的Python单元测试框架pytest为例,使用playwright的代码如下:
from playwright.sync_api import sync_playwright
def run(playwright):
browser = playwright.chromium.launch(headless=False)
context = browser.new_context(accept_downloads=True)
# Open new page
page = context.new_page()
page.goto("https://pypi.org/project/pytest/#files")
# 点击要下载的文件按钮
with page.expect_download() as download_info:
page.click("text=pytest-7.3.1.tar.gz")
download = download_info.value
path = download.path()
download.save_as(path)
# 打印保存路径
print(path)
# ---------------------
context.close()
browser.close()
with sync_playwright() as playwright:
run(playwright)
运行脚本,结果如下,我们可以看到,我们正在下载pytest的压缩文件。
总结
使用 Playwright 和 Python,我们可以轻松实现文件下载功能,这对于需要大量下载文件的应用程序来说非常实用。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 记一次.NET内存居高不下排查解决与启示