Как обнаружить нажатие кнопки внутри Qwebengine в pyside2

Question

Я написал приложение в pyside2, которое открывало веб-страницу в QWebEngine.

Эта веб-страница имеет 2 кнопки, я не понимаю, как я могу обнаружить нажатие кнопки в модуле приложения pyside2, мне нужно выполнить другую операцию с нажатием этой кнопки.

Пример

ниже мой код

from PySide2.QtWidgets import QApplication
from PySide2.QtWebEngineWidgets import QWebEngineView, QWebEnginePage
from PySide2.QtCore import QUrl

class WebEnginePage(QWebEnginePage):
    def __init__(self, *args, **kwargs):
        QWebEnginePage.__init__(self, *args, **kwargs)
        self.featurePermissionRequested.connect(self.onFeaturePermissionRequested)

    def onFeaturePermissionRequested(self, url, feature):
        if feature in (QWebEnginePage.MediaAudioCapture, 
            QWebEnginePage.MediaVideoCapture, 
            QWebEnginePage.MediaAudioVideoCapture):
            self.setFeaturePermission(url, feature, QWebEnginePage.PermissionGrantedByUser)
        else:
            self.setFeaturePermission(url, feature, QWebEnginePage.PermissionDeniedByUser)

app = QApplication([])

view = QWebEngineView()
page = WebEnginePage()
page.profile().clearHttpCache()
view.setPage(page)
view.load(QUrl("https://test.webrtc.org/"))

view.show()
app.exec_()

ниже выводится:

Теперь я просто хочу закрыть свое приложение, когда пользователь нажимает кнопку «Пуск».

Answer 1

QWebEnginePage позволяет вам выполнить js, чтобы вы могли найти кнопку , например, идентификатор, и связать ее с обратным вызовом, но это позволит вам выполнить ее только в js, и вы захотите сделайте это на python, поэтому решение состоит в том, чтобы использовать QWebChannel для экспорта QObject и вызова функции, которая закроет окно в обратном вызове. Экспорт объекта должен быть выполнен после загрузки страницы, поэтому должен использоваться сигнал loadFinished.

В моем примере я использую кнопку, чтобы указать, правильно ли загружено окно, поэтому вы должны нажать кнопку, если кнопка красная.

from PySide2 import QtCore, QtWidgets, QtWebEngineWidgets, QtWebChannel


class WebEnginePage(QtWebEngineWidgets.QWebEnginePage):
    def __init__(self, *args, **kwargs):
        super(WebEnginePage, self).__init__(*args, **kwargs)
        self.loadFinished.connect(self.onLoadFinished)

    @QtCore.Slot(bool)
    def onLoadFinished(self, ok):
        print("Finished loading: ", ok)
        if ok:
            self.load_qwebchannel()
            self.run_scripts_on_load()

    def load_qwebchannel(self):
        file = QtCore.QFile(":/qtwebchannel/qwebchannel.js")
        if file.open(QtCore.QIODevice.ReadOnly):
            content = file.readAll()
            file.close()
            self.runJavaScript(content.data().decode())
        if self.webChannel() is None:
            channel = QtWebChannel.QWebChannel(self)
            self.setWebChannel(channel)

    def add_objects(self, objects):
        if self.webChannel() is not None:
            initial_script = ""
            end_script = ""
            self.webChannel().registerObjects(objects)
            for name, obj in objects.items():
                initial_script += "var {helper};".format(helper=name)
                end_script += "{helper} = channel.objects.{helper};".format(helper=name)
            js = initial_script + \
                 "new QWebChannel(qt.webChannelTransport, function (channel) {" + \
                 end_script + \
                 "} );"
            self.runJavaScript(js)

    def run_scripts_on_load(self):
        pass


class WebRTCPageView(WebEnginePage):
    def __init__(self, *args, **kwargs):
        super(WebRTCPageView, self).__init__(*args, **kwargs)
        self.featurePermissionRequested.connect(self.onFeaturePermissionRequested)
        self.load(QtCore.QUrl("https://test.webrtc.org/"))

    @QtCore.Slot(QtCore.QUrl, QtWebEngineWidgets.QWebEnginePage.Feature)
    def onFeaturePermissionRequested(self, url, feature):
        if feature in (QtWebEngineWidgets.QWebEnginePage.MediaAudioCapture,
                       QtWebEngineWidgets.QWebEnginePage.MediaVideoCapture,
                       QtWebEngineWidgets.QWebEnginePage.MediaAudioVideoCapture):
            self.setFeaturePermission(url, feature, QtWebEngineWidgets.QWebEnginePage.PermissionGrantedByUser)
        else:
            self.setFeaturePermission(url, feature, QtWebEngineWidgets.WebEnginePage.PermissionDeniedByUser)

    def run_scripts_on_load(self):
        if self.url() == QtCore.QUrl("https://test.webrtc.org/"):
            self.add_objects({"jshelper": self})
            js = '''
                var button = document.getElementById("startButton");
                button.addEventListener("click", function(){ jshelper.on_clicked() });
            '''
            self.runJavaScript(js)

    @QtCore.Slot()
    def on_clicked(self):
        print("clicked on startButton")
        QtCore.QCoreApplication.quit()


if __name__ == '__main__':
    import sys

    app = QtWidgets.QApplication(sys.argv)

    w = QtWidgets.QWidget()
    lay = QtWidgets.QVBoxLayout(w)

    button = QtWidgets.QToolButton()
    button.setStyleSheet('''
        QToolButton{
            border: 1px; 
            border-color: black; 
            border-style: outset
        }
        QToolButton[success="true"]{
            background-color: red; 
        }
        QToolButton[success="false"]{
            background-color: green; 
        }
    ''')

    def refresh_button(ok):
        button.setProperty("success", ok)
        button.style().unpolish(button)
        button.style().polish(button)

    refresh_button(False)

    view = QtWebEngineWidgets.QWebEngineView()
    page = WebRTCPageView()
    page.profile().clearHttpCache()
    view.setPage(page)

    progressbar = QtWidgets.QProgressBar()
    page.loadProgress.connect(progressbar.setValue)
    page.loadFinished.connect(refresh_button)

    hlay = QtWidgets.QHBoxLayout()
    hlay.addWidget(progressbar)
    hlay.addWidget(button)

    lay.addWidget(view)
    lay.addLayout(hlay)
    w.resize(640, 480)
    w.show()
    sys.exit(app.exec_())