纯cdp自动化框架sbcdp-优快云博客

前言

闲来无事，想学习一下CDP协议，于是便看搜罗了一下cdp的开源项目，从playwright看到DrissionPage，最后无意中发现SeleniumBase这个自动化测试框架中也支持了CDP并且使用异步websockets实现的（并且提供了过5s盾的案例），于是萌生了将CDP从SeleniumBase剥离出来实现一个支持异步的python自动化框架的想法。

便做了个对比：
playwright：同时支持同步和异步，但是需要node，底层也是对node的封装。
DrissionPage：不支持异步
SeleniumBase：支持WebDrive和CDP，丰富易用的api（但是多是方便自动化测试）

可行，开干

正文

这么繁杂的工作怎么能自己动手呢？直接将SeleniumBase拉下来，丢给ai，开始和ai掰扯指挥ai干活，让其帮我将CDP部分的代码剥离出来，很快啊，ai就将cdp剥离了出来，并跑通了ai自己编写的测试用例。看看ai干的咋样，基本没啥问题，于是调整了一下目录结构，手动删除一些没删除干净的无用代码。这样一个同步的cdp自动化框架基本就完成了。

SeleniumBase关于CDP部分的代码是使用异步的websockets实现的，但是SeleniumBase却使用loop.run_until_complete去跑异步代码，硬生生将异步改成了同步。这与我的初衷相悖，于是继续让ai将同步改回异步。ai三下五除二就搞定了，一个异步的cdp协议的自动化框架就初具雏形了。

为了同时支持异步同步，我便手动增加了一些代码，让其支持同步，调整项目结构，修复bug。

很快剥离工作便完成了，因为是从SeleniumBase剥离出来的CDP协议，我便将这个项目命名为sbcdp。

项目地址：https://github.com/ConlinH/sbcdp

使用

安装

pip install sbcdp

示例

import asyncio
from sbcdp import AsyncChrome as Chrome


async def main():
    async with Chrome() as chrome:
        await chrome.get("https://httpbin.org/forms/post")
        await chrome.type('input[name="custname"]', "sbcdp 用户")
        await chrome.type('input[name="custtel"]', "123-456-7890")
        await chrome.type('input[name="custemail"]', "test@cdp-base.com")
        await chrome.type('textarea[name="comments"]', "这是使用sbcdp框架的测试")

        # 选择单选按钮
        await chrome.click('input[value="large"]')
        # 等待元素
        element = await chrome.find_element("button")
        await element.click()
        await chrome.sleep(2)


if __name__ == '__main__':
    asyncio.new_event_loop().run_until_complete(main())
    # asyncio.run(main())

5s盾示例

import asyncio
from contextlib import suppress

from sbcdp import AsyncChrome as Chrome


async def main():
    # url = "https://fractal-testnet.unisat.io/explorer"
    url = "https://steamdb.info/"
    # url = "https://cn.airbusan.com/content/individual"
    # url = "https://pastebin.com/login"
    # url = "https://simple.ripley.com.pe/"
    # url = "https://www.e-food.gr/"
    async with Chrome() as chrome:
        await chrome.get(url)
        await chrome.sleep(5)
        with suppress(Exception):
            await chrome.mouse_click('input[type=checkbox]')
        assert 'cf_clearance' in {c.name: c.value for c in await chrome.get_all_cookies()}
        print({c.name: c.value for c in await chrome.get_all_cookies()})


if __name__ == "__main__":
    asyncio.new_event_loop().run_until_complete(main())
    # asyncio.run(main())