- Added support for PyQt.

- Refactored widgets to catch ModuleNotFoundErrors.
This commit is contained in:
louisnw
2023-05-15 12:28:25 +01:00
parent d8424d6da3
commit 7c125b24ec
4 changed files with 44 additions and 13 deletions

View File

@ -1,8 +1,4 @@
from .chart import Chart
from .js import LWC
try:
import wx.html2
from .widgets import WxChart
except:
pass

View File

@ -1,14 +1,22 @@
from lightweight_charts.js import LWC
try:
import wx.html2
except ImportError:
pass
try:
from PyQt5.QtWebEngineWidgets import QWebEngineView
except ImportError:
pass
from lightweight_charts.js import LWC
class WxChart(LWC):
def __init__(self, parent, volume_enabled=True):
super().__init__(volume_enabled)
self.webview = wx.html2.WebView.New(parent)
try:
self.webview = wx.html2.WebView.New(parent)
except NameError:
raise ModuleNotFoundError('wx.html2 was not found, and must be installed to use WxChart.')
self.webview.Bind(wx.html2.EVT_WEBVIEW_LOADED, self._on_js_load)
self.webview.SetPage(self._html, '')
@ -17,7 +25,7 @@ class WxChart(LWC):
def run_script(self, script): self.webview.RunScript(script)
def _on_js_load(self, e: wx.html2.WebViewEvent):
def _on_js_load(self, e):
if not self.second_load:
self.second_load = True
return
@ -25,4 +33,26 @@ class WxChart(LWC):
for func, args, kwargs in self.js_queue:
getattr(super(), func)(*args, **kwargs)
def get_webview(self): return self.webview
def get_webview(self): return self.webview
class QtChart(LWC):
def __init__(self, widget=None, volume_enabled=True):
super().__init__(volume_enabled)
try:
self.webview = QWebEngineView(widget)
except NameError:
raise ModuleNotFoundError('QWebEngineView was not found, and must be installed to use QtChart.')
self.webview.loadFinished.connect(self._on_js_load)
self.webview.page().setHtml(self._html)
def run_script(self, script): self.webview.page().runJavaScript(script)
def _on_js_load(self):
self.loaded = True
for func, args, kwargs in self.js_queue:
getattr(super(), func)(*args, **kwargs)
def get_webview(self): return self.webview