d43e7c24e7
- Hotkeys can now use any character, and modifier keys are not required. - Refactored the colors of the topbar, searchbox, toolbox, and widgets for consistency. - Toolbox/interval refactoring and simplification. - Histograms now show up in the legend, and will use shorthand notation by default (e.g 34k rather than 34000).
103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
import asyncio
|
|
from typing import Dict, Literal
|
|
|
|
from .util import jbool, Pane
|
|
|
|
|
|
ALIGN = Literal['left', 'right']
|
|
|
|
|
|
class Widget(Pane):
|
|
def __init__(self, topbar, value, func=None):
|
|
super().__init__(topbar.win)
|
|
self.value = value
|
|
|
|
def wrapper(v):
|
|
self.value = v
|
|
func(topbar._chart)
|
|
|
|
async def async_wrapper(v):
|
|
self.value = v
|
|
await func(topbar._chart)
|
|
|
|
self.win.handlers[self.id] = async_wrapper if asyncio.iscoroutinefunction(func) else wrapper
|
|
|
|
|
|
class TextWidget(Widget):
|
|
def __init__(self, topbar, initial_text, align):
|
|
super().__init__(topbar, value=initial_text)
|
|
self.run_script(f'{self.id} = {topbar.id}.makeTextBoxWidget("{initial_text}", "{align}")')
|
|
|
|
def set(self, string):
|
|
self.value = string
|
|
self.run_script(f'{self.id}.innerText = "{string}"')
|
|
|
|
|
|
class SwitcherWidget(Widget):
|
|
def __init__(self, topbar, options, default, align, func):
|
|
super().__init__(topbar, value=default, func=func)
|
|
self.run_script(f'{self.id} = {topbar.id}.makeSwitcher({list(options)}, "{default}", "{self.id}", "{align}")')
|
|
|
|
|
|
class MenuWidget(Widget):
|
|
def __init__(self, topbar, options, default, separator, align, func):
|
|
super().__init__(topbar, value=default, func=func)
|
|
self.run_script(f'''
|
|
{self.id} = {topbar.id}.makeMenu({list(options)}, "{default}", {jbool(separator)}, "{self.id}", "{align}")
|
|
''')
|
|
|
|
|
|
class ButtonWidget(Widget):
|
|
def __init__(self, topbar, button, separator, align, func):
|
|
super().__init__(topbar, value=button, func=func)
|
|
self.run_script(
|
|
f'{self.id} = {topbar.id}.makeButton("{button}", "{self.id}", {jbool(separator)}, true, "{align}")')
|
|
|
|
def set(self, string):
|
|
self.value = string
|
|
self.run_script(f'{self.id}.elem.innerText = "{string}"')
|
|
|
|
|
|
class TopBar(Pane):
|
|
def __init__(self, chart):
|
|
super().__init__(chart.win)
|
|
self._chart = chart
|
|
self._widgets: Dict[str, Widget] = {}
|
|
self._created = False
|
|
|
|
def _create(self):
|
|
if self._created:
|
|
return
|
|
from lightweight_charts.abstract import JS
|
|
self._created = True
|
|
self.run_script(JS['callback'])
|
|
self.run_script(f'{self.id} = new TopBar({self._chart.id})')
|
|
|
|
def __getitem__(self, item):
|
|
if widget := self._widgets.get(item):
|
|
return widget
|
|
raise KeyError(f'Topbar widget "{item}" not found.')
|
|
|
|
def get(self, widget_name):
|
|
return self._widgets.get(widget_name)
|
|
|
|
def switcher(self, name, options: tuple, default: str = None,
|
|
align: ALIGN = 'left', func: callable = None):
|
|
self._create()
|
|
self._widgets[name] = SwitcherWidget(self, options, default if default else options[0], align, func)
|
|
|
|
def menu(self, name, options: tuple, default: str = None, separator: bool = True,
|
|
align: ALIGN = 'left', func: callable = None):
|
|
self._create()
|
|
self._widgets[name] = MenuWidget(self, options, default if default else options[0], separator, align, func)
|
|
|
|
def textbox(self, name: str, initial_text: str = '',
|
|
align: ALIGN = 'left'):
|
|
self._create()
|
|
self._widgets[name] = TextWidget(self, initial_text, align)
|
|
|
|
def button(self, name, button_text: str, separator: bool = True,
|
|
align: ALIGN = 'left', func: callable = None):
|
|
self._create()
|
|
self._widgets[name] = ButtonWidget(self, button_text, separator, align, func)
|