获取请求信息

QWebEngineView/GetRequestInfo.py

@@ -0,0 +1,122 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Created on 2019年9月24日
+@author: Irony
+@site: https://pyqt5.com https://github.com/892768447
+@email: [email protected]
+@file: QWebEngineView.BlockAds
+@description: 拦截请求
+"""
+from PyQt5.QtCore import QUrl
+from PyQt5.QtNetwork import QNetworkAccessManager, QNetworkRequest
+from PyQt5.QtWebEngineCore import QWebEngineUrlSchemeHandler, QWebEngineUrlScheme, \
+    QWebEngineUrlRequestInterceptor
+from PyQt5.QtWebEngineWidgets import QWebEngineView, QWebEngineProfile
+
+__Author__ = 'Irony'
+__Copyright__ = 'Copyright (c) 2019'
+__Version__ = 'Version 1.0'
+
+
+class UrlSchemeHandler(QWebEngineUrlSchemeHandler):
+    AttrType = QNetworkRequest.User + 1
+
+    def __init__(self, *args, **kwargs):
+        super(UrlSchemeHandler, self).__init__(*args, **kwargs)
+        self._manager = QNetworkAccessManager(self)
+        self._manager.finished.connect(self.onFinished)
+
+    def requestStarted(self, request):
+        # 拦截
+        # request.fail(QWebEngineUrlRequestJob.RequestDenied)
+        # print('initiator:', request.initiator())
+        print('requestMethod:', request.requestMethod())
+        print('requestHeaders:', request.requestHeaders())
+        url = request.requestUrl()
+        if url.scheme().startswith('myurl'):
+            url.setScheme(url.scheme().replace('myurl', 'http'))
+        print('requestUrl:', url)
+
+        # 构造真实请求
+        req = QNetworkRequest(url)
+        req.setAttribute(self.AttrType, request)  # 记录
+        for headerName, headerValue in request.requestHeaders().items():
+            req.setRawHeader(headerName, headerValue)
+        method = request.requestMethod()
+        if method == b'GET':
+            self._manager.get(req)
+        elif method == b'POST':
+            self._manager.post(req)
+
+    def onFinished(self, reply):
+        req = reply.request()  # 获取请求
+        o_req = req.attribute(self.AttrType, None)
+        if o_req:
+            o_req.reply(req.header(QNetworkRequest.ContentTypeHeader) or b'text/html', reply)
+            o_req.destroyed.connect(reply.deleteLater)
+
+
+# 把所有请求重定向到myurl
+class RequestInterceptor(QWebEngineUrlRequestInterceptor):
+
+    def interceptRequest(self, info):
+        url = info.requestUrl()
+        if url.scheme() == 'http':
+            # 重定向
+            url.setScheme('myurl')
+            info.redirect(url)
+        elif url.scheme() == 'https':
+            # 重定向
+            url.setScheme('myurls')
+            info.redirect(url)
+
+
+class Window(QWebEngineView):
+
+    def __init__(self, *args, **kwargs):
+        super(Window, self).__init__(*args, **kwargs)
+        self.resize(800, 600)
+        profile = QWebEngineProfile.defaultProfile()
+
+        # 首先获取默认的url协议
+        o_http = QWebEngineUrlScheme.schemeByName(b'http')
+        o_https = QWebEngineUrlScheme.schemeByName(b'https')
+        print('scheme:', o_http, o_https)
+
+        # 这里需要修改增加本地文件和跨域支持
+        CorsEnabled = 0x80  # 5.14才增加
+        o_http.setFlags(
+            o_http.flags() | QWebEngineUrlScheme.SecureScheme | QWebEngineUrlScheme.LocalScheme | QWebEngineUrlScheme.LocalAccessAllowed | CorsEnabled)
+        o_https.setFlags(
+            o_https.flags() | QWebEngineUrlScheme.SecureScheme | QWebEngineUrlScheme.LocalScheme | QWebEngineUrlScheme.LocalAccessAllowed | CorsEnabled)
+
+        # 安装url拦截器和自定义url协议处理
+        de = QWebEngineProfile.defaultProfile()  # @UndefinedVariable
+        de.setRequestInterceptor(RequestInterceptor(self))
+        self.urlSchemeHandler = UrlSchemeHandler(self)
+        de.installUrlSchemeHandler(b'myurl', self.urlSchemeHandler)  # for http
+        de.installUrlSchemeHandler(b'myurls', self.urlSchemeHandler)  # for https
+
+
+if __name__ == '__main__':
+    import sys
+    import os
+    import webbrowser
+    import cgitb
+
+    cgitb.enable(format='text')
+    from PyQt5.QtWidgets import QApplication
+
+    app = QApplication(sys.argv)
+    # 开启F12 控制台功能，需要单独通过浏览器打开这个页面
+    # 这里可以做个保护, 发布软件,启动时把这个环境变量删掉。防止他人通过环境变量开启
+    os.environ['QTWEBENGINE_REMOTE_DEBUGGING'] = '9966'
+    # 打开调试页面
+    webbrowser.open_new_tab('http://127.0.0.1:9966')
+
+    w = Window()
+    w.show()
+    w.load(QUrl('https://pyqt.site'))
+    sys.exit(app.exec_())