Files
lightweight-charts-python/lightweight_charts/chart.py
louisnw a2d29ed829 - Significant refactoring resulting in a 34% reduction in size of the codebase (excluding the Lightweight Charts package) and greater efficiency.
- Upgraded to Lightweight Charts v4.0.1.
- Added a ‘hover’ item to the returning dictionary from subscribe_click.
- Markers and SubCharts no longer use a UUID for identification, but an 8 character string.
2023-05-21 15:44:10 +01:00

92 lines
3.0 KiB
Python

import webview
import multiprocessing as mp
from lightweight_charts.js import LWC
class PyWV:
def __init__(self, q, exit, loaded, html, js_api, width, height, x, y, on_top, debug):
self.queue = q
self.exit = exit
self.loaded = loaded
self.debug = debug
self.js_api = js_api
self.webview = webview.create_window('', html=html, on_top=on_top, js_api=js_api,
width=width, height=height, x=x, y=y)
self.webview.events.loaded += self.on_js_load
self.loop()
def loop(self):
while 1:
arg = self.queue.get()
if arg in ('start', 'show', 'hide', 'exit'):
webview.start(debug=self.debug) if arg == 'start' else getattr(self.webview, arg)()
self.exit.set() if arg in ('start', 'exit') else None
elif arg == 'subscribe':
func, c_id = (self.queue.get() for _ in range(2))
self.js_api.click_funcs[str(c_id)] = func
else:
try:
self.webview.evaluate_js(arg)
except KeyError:
return
def on_js_load(self):
self.loaded.set(), self.loop()
class Chart(LWC):
def __init__(self, volume_enabled: bool = True, width: int = 800, height: int = 600, x: int = None, y: int = None,
on_top: bool = False, debug: bool = False,
inner_width: float = 1.0, inner_height: float = 1.0):
super().__init__(volume_enabled, inner_width, inner_height)
self._js_api_code = 'pywebview.api.onClick'
self._q = mp.Queue()
self._script_func = self._q.put
self._exit = mp.Event()
self._loaded = mp.Event()
self._process = mp.Process(target=PyWV, args=(self._q, self._exit, self._loaded, self._html, self._js_api,
width, height, x, y, on_top, debug,), daemon=True)
self._process.start()
self._create_chart()
def show(self, block: bool = False):
"""
Shows the chart window.\n
:param block: blocks execution until the chart is closed.
"""
if not self.loaded:
self._q.put('start')
self._loaded.wait()
self._on_js_load()
else:
self._q.put('show')
if block:
try:
self._exit.wait()
except KeyboardInterrupt:
return
self._exit.clear()
def hide(self):
"""
Hides the chart window.\n
"""
self._q.put('hide')
def exit(self):
"""
Exits and destroys the chart window.\n
"""
self._q.put('exit')
self._exit.wait()
self._process.terminate()
del self
def subscribe_click(self, function: object):
self._q.put('subscribe')
self._q.put(function)
self._q.put(self.id)
super().subscribe_click(function)