80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
import datetime
|
|
|
|
import webview
|
|
from multiprocessing import Queue
|
|
|
|
from lightweight_charts.js import LWC
|
|
|
|
_q = Queue()
|
|
_result_q = Queue()
|
|
|
|
DEBUG = True
|
|
|
|
|
|
class API:
|
|
def __init__(self):
|
|
self.click_func = None
|
|
|
|
def onClick(self, data):
|
|
if isinstance(data['time'], int):
|
|
data['time'] = datetime.datetime.fromtimestamp(data['time'])
|
|
else:
|
|
data['time'] = datetime.datetime(data['time']['year'], data['time']['month'], data['time']['day'])
|
|
self.click_func(data) if self.click_func else None
|
|
|
|
|
|
class Webview(LWC):
|
|
def __init__(self, chart):
|
|
super().__init__(chart.volume_enabled)
|
|
self.chart = chart
|
|
self.started = False
|
|
|
|
self.js_api = API()
|
|
self.webview = webview.create_window('', html=self._html, on_top=chart.on_top, js_api=self.js_api,
|
|
width=chart.width, height=chart.height, x=chart.x, y=chart.y)
|
|
self.webview.events.loaded += self._on_js_load
|
|
|
|
def run_script(self, script): self.webview.evaluate_js(script)
|
|
|
|
def _on_js_load(self):
|
|
self.loaded = True
|
|
while len(self.js_queue) > 0:
|
|
func, args, kwargs = self.js_queue[0]
|
|
getattr(self, func)(*args)
|
|
del self.js_queue[0]
|
|
_loop(self.chart, controller=self)
|
|
|
|
def show(self):
|
|
if self.loaded:
|
|
self.webview.show()
|
|
else:
|
|
webview.start(debug=self.chart.debug)
|
|
|
|
def subscribe_click(self, function):
|
|
if self._stored('subscribe_click', function):
|
|
return None
|
|
|
|
self.js_api.click_func = function
|
|
self.run_script('isSubscribed = true')
|
|
|
|
def create_line(self, color: str = 'rgba(214, 237, 255, 0.6)', width: int = 2):
|
|
return super().create_line(color, width).id
|
|
|
|
def hide(self): self.webview.hide()
|
|
|
|
def exit(self): self.webview.destroy()
|
|
|
|
|
|
def _loop(chart, controller=None):
|
|
wv = Webview(chart) if not controller else controller
|
|
while 1:
|
|
func, args = chart._q.get()
|
|
try:
|
|
result = getattr(wv, func)(*args)
|
|
except KeyError as e:
|
|
return
|
|
if func == 'show':
|
|
chart._exit.set()
|
|
chart._result_q.put(result) if result is not None else None
|
|
|