C++ WebDriver扩展-平芜编程栈

概述

WebDriver协议基于HTTP，使用JSON进行数据传输，定义了client与driver之间的通信标准。无论client的实现语言（如Java或C#），都能通过协议中的endpoints准确指示driver执行各种操作，覆盖了Selenium的所有功能。这些endpoints详细列出了所有可用的功能点。

目标

msedgedriver下载

启动

msedgedriver.exe --port=8080 --remote-allow-origins=* - --verbose -allow-insecure-localhost --allowed-origins=* --allowed-ips=["127.0.0.1"] --headless --disable-gpu

示例

#include "webdriver.h" int main() { WebDriver m_driver; m_driver.NavigateTo("https://www.hao123.com/"); qDebug() << m_driver.getPageSource(); }

效果

webdriver.h

#ifndef WEBDRIVER_H #define WEBDRIVER_H #include <QNetworkAccessManager> #include <QNetworkReply> class WebDriver : public QObject { Q_OBJECT public: WebDriver(); ~WebDriver(); void getStatus(); void getTimeout(); void setTimeOut(int implicit, long long pageLoad, long long script); void NavigateTo(const QString& url); QString getCurrentUrl(); void historyBack(); void historyForward(); void refresh(); QString getTitle(); QString showWindow(); void closeWindow(); void getWindowHandles(); void createWindow(); QString getPageSource(); signals: void response(const QByteArray& bytes); private: void connectToServer(const QUrl& url); private: void post(const QByteArray& bytes); void get(); void deleteRes(); private: void waitFinish(); private: QString m_host; QNetworkAccessManager m_networkManger; QNetworkRequest m_client; std::string m_sessionID; std::string m_elementID; QByteArray m_responseCache; }; #endif // WEBDRIVER_H

webdriver.cpp

#include "webdriver.h" #include <QElapsedTimer> #include <QApplication> #include <QJsonDocument> #include <QJsonObject> #include "webDriverResponseJson.h" #include "webDriverRequestJson.h" WebDriver::WebDriver() : m_host("http://127.0.0.1:8080"){ connectToServer(QUrl(m_host + "/session")); } WebDriver::~WebDriver() { if (m_sessionID.empty()) return; m_client.setUrl(QUrl(m_host + QString("/session/") + QString::fromStdString(m_sessionID))); deleteRes(); waitFinish(); } void WebDriver::connectToServer(const QUrl& url) { m_client.setUrl(url); m_client.setHeader(QNetworkRequest::ContentTypeHeader, "application/json; charset=utf-8"); m_client.setHeader(QNetworkRequest::UserAgentHeader, "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0"); m_client.setRawHeader("Cache-Control", "no-cache"); REQ::Options options; options.capabilities.browserName = "edge"; std::string json = JS::serializeStruct(options, JS::SerializerOptions(JS::SerializerOptions::Compact)); post(QString::fromStdString(json).toLocal8Bit()); waitFinish(); // 反序列 RES::MSEdgeRoot response; JS::ParseContext parseContext(m_responseCache.data()); auto error = parseContext.parseTo(response); if (JS::Error::NoError != error) { std::string errorStr = parseContext.makeErrorString(); fprintf(stderr, "Error parsing struct %s\n", errorStr.c_str()); } m_sessionID = response.value.sessionId; } void WebDriver::NavigateTo(const QString& url) { m_client.setUrl(QUrl(m_host + QString("/session/") + QString::fromStdString(m_sessionID) + "/url")); REQ::NavigateTo nav; nav.url = url.toStdString(); std::string json = JS::serializeStruct(nav, JS::SerializerOptions(JS::SerializerOptions::Compact)); post(QString::fromStdString(json).toLocal8Bit()); waitFinish(); QJsonParseError jsonError; QJsonDocument doucment = QJsonDocument::fromJson(m_responseCache, &jsonError); // 转化为 JSON 文档 if (!doucment.isNull() && (jsonError.error == QJsonParseError::NoError)) { // 解析未发生错误 if (doucment.isObject()) { // JSON 文档为对象 QJsonObject object = doucment.object(); // 转化为对象 if (object.contains("value")) { // 包含指定的 key QJsonValue value = object.value("value"); // 获取指定 key 对应的 value qDebug() << "NavigateTo " << (value.isNull() ? "Success" : "Fail"); } } } } QString WebDriver::getPageSource() { m_client.setUrl(QUrl(m_host + QString("/session/") + QString::fromStdString(m_sessionID) + "/source")); get(); waitFinish(); QJsonParseError jsonError; QJsonDocument doucment = QJsonDocument::fromJson(m_responseCache, &jsonError); // 转化为 JSON 文档 if (!doucment.isNull() && (jsonError.error == QJsonParseError::NoError)) { // 解析未发生错误 if (doucment.isObject()) { // JSON 文档为对象 QJsonObject object = doucment.object(); // 转化为对象 if (object.contains("value")) { // 包含指定的 key QJsonValue value = object.value("value"); // 获取指定 key 对应的 value if (value.isString()) return value.toString(); } } } return QString::fromStdString(m_responseCache.toStdString()); } void WebDriver::waitFinish() { QEventLoop loop; connect(this, &WebDriver::response, &loop, &QEventLoop::quit); loop.exec(); } void WebDriver::post(const QByteArray& bytes) { auto reply = m_networkManger.post(m_client, bytes); connect(reply, &QNetworkReply::finished, [this, reply]() { m_responseCache = reply->readAll(); qDebug() << "finish:" << m_responseCache; emit response(m_responseCache); }); } void WebDriver::get() { auto reply = m_networkManger.get(m_client); connect(reply, &QNetworkReply::finished, [this, reply]() { m_responseCache = reply->readAll(); qDebug() << "finish:" << m_responseCache; emit response(m_responseCache); }); } void WebDriver::deleteRes() { auto reply = m_networkManger.deleteResource(m_client); connect(reply, &QNetworkReply::finished, [this, reply]() { emit response(m_responseCache); }); }

example.py

import requests import json session_url = 'http://localhost:8080/session' session_pars ={"capabilities": {"browserName": "edge"}} r_session = requests.post(session_url,json=session_pars) print(json.dumps(r_session.json(),indent=2))