PyQt5 內(nèi)嵌瀏覽器注入 Javascript 腳本實(shí)現(xiàn)自動(dòng)化操作

xiguadada 發(fā)布于2019-07-30 17:40 / 2108人閱讀

摘要：概要應(yīng)同學(xué)邀請(qǐng)，演示如何使用內(nèi)嵌瀏覽器瀏覽網(wǎng)頁，并注入腳本實(shí)現(xiàn)自動(dòng)化操作。在默認(rèn)配置中植入內(nèi)容，這樣腳本會(huì)在所有打開的網(wǎng)頁中執(zhí)行，不論跳轉(zhuǎn)到哪個(gè)網(wǎng)址。腳本使用網(wǎng)址中的路徑名，判斷當(dāng)前網(wǎng)頁位置，從而決定執(zhí)行哪種操作。

概要

應(yīng)同學(xué)邀請(qǐng)，演示如何使用 PyQt5 內(nèi)嵌瀏覽器瀏覽網(wǎng)頁，并注入 Javascript 腳本實(shí)現(xiàn)自動(dòng)化操作。

sg 原貼地址：如何在Python利用runJavaScript模擬鼠標(biāo)移動(dòng)頁面的某個(gè)元素
https://segmentfault.com/q/10...

下面測(cè)試的是一個(gè)廉價(jià)機(jī)票預(yù)訂網(wǎng)站（http://www.flyscoot.com/），關(guān)鍵點(diǎn)如下

使用 QWebEngineView 加載網(wǎng)頁，并顯示進(jìn)度。

在默認(rèn)配置（QWebEngineProfile）中植入 Javascript 內(nèi)容，這樣腳本會(huì)在所有打開的網(wǎng)頁中執(zhí)行，不論跳轉(zhuǎn)到哪個(gè)網(wǎng)址。

Javascript 腳本使用網(wǎng)址中的路徑名，判斷當(dāng)前網(wǎng)頁位置，從而決定執(zhí)行哪種操作。

python 代碼示例

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""使用 PyQt5 內(nèi)嵌瀏覽器瀏覽網(wǎng)頁，并注入 Javascript 腳本實(shí)現(xiàn)自動(dòng)化操作。"""
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)

        # 導(dǎo)航/工具
        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("網(wǎng)址"))
        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 腳本示例

// 簡(jiǎn)單起見，這里只演示部分頁面，腳本內(nèi)容摘自 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);
}