first commit

lightweight_charts/pywebview.py

@@ -0,0 +1,79 @@
+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
+