-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwebengineview.py
51 lines (40 loc) · 1.75 KB
/
webengineview.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
from PySide6.QtWebEngineWidgets import QWebEngineView
from PySide6.QtWebChannel import QWebChannel
from PySide6.QtCore import QFile, QIODevice, QDir
from PySide6.QtWidgets import QFileDialog
from backend import Backend
from xmlhelper import XmlHelper
class WebEngineView(QWebEngineView):
def __init__(self, parent=None):
super().__init__(parent)
self._backend = Backend()
self._xmlHelper = XmlHelper()
self.page().loadFinished.connect(self.handleLoadFinished)
self._backend.openFileClicked.connect(self.handleOpenFileClicked)
def handleOpenFileClicked(self):
fileName = QFileDialog.getOpenFileName(self, "Open File", QDir().currentPath(), "XML files (*.xml)")
if fileName[0] == '':
return
xmlDict = self._xmlHelper.parse(fileName[0])
self._backend.setGraphData(xmlDict)
def handleLoadFinished(self, ok):
if ok:
self.loadQWebChannel()
self.loadObject()
def loadQWebChannel(self):
file = QFile(":/qtwebchannel/qwebchannel.js")
if file.open(QIODevice.ReadOnly):
content = file.readAll()
file.close()
self.page().runJavaScript(content.data().decode())
if self.page().webChannel() is None:
channel = QWebChannel(self.page())
self.page().setWebChannel(channel)
def loadObject(self):
if self.page().webChannel() is not None:
self.page().webChannel().registerObject("backend", self._backend)
script = r"""
new QWebChannel(qt.webChannelTransport, function (channel) {
window.backend = channel.objects.backend;
});"""
self.page().runJavaScript(script)