python本地浏览器注入js_PyQt5内嵌浏览器注入JavaScript脚本实现自动化操作的代码实例...

概要

应同学邀请，演示如何使用 PyQt5 内嵌浏览器浏览网页，并注入 Javascript 脚本实现自动化操作。

下面测试的是一个廉价机票预订网站(http://www.flyscoot.com/)，关键点如下

使用 QWebEngineView 加载网页，并显示进度。

在默认配置(QWebEngineProfile)中植入 Javascript 内容，这样脚本会在所有打开的网页中执行，不论跳转到哪个网址。

Javascript 脚本使用网址中的路径名，判断当前网页位置，从而决定执行哪种操作。

python 代码示例

#!/usr/bin/env python3

# -*- coding: utf-8 -*-

'''使用 PyQt5 内嵌浏览器浏览网页，并注入 Javascript 脚本实现自动化操作。'''

import os

import sys

from datetime import datetime

from PyQt5.QtWidgets import (

QWidget, QApplication, QVBoxLayout, QHBoxLayout,

QDesktopWidget, QTextEdit, QLabel, QLineEdit, QPushButton,

QFileDialog, QProgressBar,

)

from PyQt5.QtCore import QUrl, pyqtSlot

from PyQt5.QtWebEngineWidgets import QWebEngineView, QWebEngineProfile, QWebEngineScript, QWebEnginePage

class Browser(QWidget):

def __init__(self):

super().__init__()

self.init_ui()

# 脚本

self.profile = QWebEngineProfile.defaultProfile()

self.script = QWebEngineScript()

self.prepare_script()

def init_ui(self):

self.webView = QWebEngineView()

self.logEdit = QTextEdit()

self.logEdit.setFixedHeight(100)

self.addrEdit = QLineEdit()

self.addrEdit.returnPressed.connect(self.load_url)

self.webView.urlChanged.connect(

lambda i: self.addrEdit.setText(i.toDisplayString()))

self.jsEdit = QLineEdit()

self.jsEdit.setText('inject.js')

loadUrlBtn = QPushButton('加载')

loadUrlBtn.clicked.connect(self.load_url)

chooseJsBtn = QPushButton('选择脚本文件')

chooseJsBtn.clicked.connect(self.choose_js_file)

# 导航/工具

top = QWidget()

top.setFixedHeight(80)

topBox = QVBoxLayout(top)

topBox.setSpacing(0)

topBox.setContentsMargins(5, 0, 0, 5)

progBar = QProgressBar()

progBox = QHBoxLayout()

progBox.addWidget(progBar)

topBox.addLayout(progBox)

naviBox = QHBoxLayout()

naviBox.addWidget(QLabel('网址'))

naviBox.addWidget(self.addrEdit)

naviBox.addWidget(loadUrlBtn)

topBox.addLayout(naviBox)

naviBox = QHBoxLayout()

naviBox.addWidget(QLabel('注入脚本文件'))

naviBox.addWidget(self.jsEdit)

naviBox.addWidget(chooseJsBtn)

topBox.addLayout(naviBox)

self.webView.loadProgress.connect(progBar.setValue)

# 主界面

layout = QVBoxLayout(self)

layout.addWidget(self.webView)

layout.addWidget(top)

layout.addWidget(self.logEdit)

self.show()

self.resize(1024, 900)

self.center()

def center(self):

qr = self.frameGeometry()

cp = QDesktopWidget().availableGeometry().center()

qr.moveCenter(cp)

self.move(qr.topLeft())

@pyqtSlot()

def load_url(self):

url = self.addrEdit.text().strip()

if not url.lower().startswith('http://')

and not url.lower().startswith('https://'):

url = 'http://{}'.format(url)

self.load(url)

@pyqtSlot()

def choose_js_file(self):

f, _ = QFileDialog.getOpenFileName(filter="Javascript files(*.js)")

if os.path.isfile(f):

self.jsEdit.setText(f)

self.prepare_script()

def prepare_script(self):

path = self.jsEdit.text().strip()

if not os.path.isfile(path):

self.log('invalid js path')

return

self.profile.scripts().remove(self.script)

with open(path, 'r') as f:

self.script.setSourceCode(f.read())

self.profile.scripts().insert(self.script)

self.log('injected js ready')

def log(self, msg, *args, **kwargs):

m = msg.format(*args, **kwargs)

self.logEdit.append('{} {}'.format(

datetime.now().strftime('%H:%M:%S'), m))

def load(self, url):

self.log(f'loading {url}')

self.addrEdit.setText(url)

self.webView.load(QUrl(url))

if __name__ == '__main__':

app = QApplication(sys.argv)

b = Browser()

b.load('http://www.flyscoot.com/')

sys.exit(app.exec_())

Javascript 脚本示例

// 简单起见，这里只演示部分页面，脚本内容摘自 Heng丶原贴文。

function handle(path) {

// 首页

if (path == '/zh') {

document.getElementsByClassName('radio-inline')[1].click();

document.getElementById('oneway_from').value='广州 (CAN)';

document.getElementById('oneway_to').value='新加坡 (SIN)';

document.getElementById('oneway_departuredate').value='2018年9月10日';

document.getElementsByClassName('btn--booking')[1].click();

return;

}

// 选择航班

if (path == '/Book/Flight') {

document.getElementsByClassName('price--sale')[0].click();

document.getElementsByClassName('heading-4')[0].click();

document.getElementsByClassName('btn-submit')[0].click();

return;

}

// 乘客信息

if (path == '/BookFlight/Passengers') {

document.getElementsByClassName('fname1')[0].value = "匿名";

}

}

let host = document.location.hostname;

if (host.endsWith('.flyscoot.com')) {

handle(document.location.pathname);

}

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对脚本之家的支持。如果你想了解更多相关内容请查看下面相关链接