forked from PyQt5/PyQt
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BlockRequest.py
62 lines (50 loc) · 1.92 KB
/
BlockRequest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#!/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: QWebView.BlockAds
@description: 拦截请求
"""
from PyQt5.QtCore import QUrl, QBuffer, QByteArray
from PyQt5.QtNetwork import QNetworkAccessManager
from PyQt5.QtWebKitWidgets import QWebView
__Author__ = 'Irony'
__Copyright__ = 'Copyright (c) 2019'
__Version__ = 'Version 1.0'
class RequestInterceptor(QNetworkAccessManager):
def createRequest(self, op, originalReq, outgoingData):
"""创建请求
:param op: 操作类型见http://doc.qt.io/qt-5/qnetworkaccessmanager.html#Operation-enum
:param originalReq: 原始请求
:param outgoingData: 输出数据
"""
url = originalReq.url().toString()
if url.find('pos.baidu.com') > -1 and url.find('ltu=') > -1:
# 拦截百度联盟的广告
print('block:', url)
originalReq.setUrl(QUrl())
if op == self.PostOperation and outgoingData:
# 拦截或者修改post数据
# 读取后要重新设置,不然网站接收不到请求
data = outgoingData.readAll().data()
print('post data:', data)
# 修改data后重新设置
outgoingData = QBuffer(self)
outgoingData.setData(data)
return super(RequestInterceptor, self).createRequest(op, originalReq, outgoingData)
class Window(QWebView):
def __init__(self, *args, **kwargs):
super(Window, self).__init__(*args, **kwargs)
self.resize(800, 600)
self.page().setNetworkAccessManager(RequestInterceptor(self))
if __name__ == '__main__':
import sys
from PyQt5.QtWidgets import QApplication
app = QApplication(sys.argv)
w = Window()
w.show()
w.load(QUrl('https://so.csdn.net/so/search/s.do?q=Qt&t=blog'))
sys.exit(app.exec_())