f72baf95ba
- Added async methods to polygon. - The `requests` library is no longer required, with `urllib` being used instead. - Added the `get_bar_data` function, which returns a dataframe of aggregate data from polygon. - Opened up the `subscribe` and `unsubscribe` functions Enhancements: - Tables will now scroll when the rows exceed table height. Bugs: - Fixed a bug preventing async functions being used with horizontal line event. - Fixed a bug causing the legend to show duplicate lines if the line was created after the legend. - Fixed a bug causing the line hide icon to persist within the legend after deletion (#75) - Fixed a bug causing the search box to be unfocused when the chart is loaded.
87 lines
3.4 KiB
Python
87 lines
3.4 KiB
Python
import random
|
|
from typing import Union
|
|
|
|
from .util import jbool, Pane, NUM
|
|
|
|
|
|
class Footer:
|
|
def __init__(self, table):
|
|
self._table = table
|
|
|
|
def __setitem__(self, key, value):
|
|
self._table.run_script(f'{self._table.id}.footer[{key}].innerText = "{value}"')
|
|
|
|
def __call__(self, number_of_text_boxes: int):
|
|
self._table.run_script(f'{self._table.id}.makeFooter({number_of_text_boxes})')
|
|
|
|
|
|
class Row(dict):
|
|
def __init__(self, table, id, items):
|
|
super().__init__()
|
|
self.run_script = table.run_script
|
|
self._table = table
|
|
self.id = id
|
|
self.meta = {}
|
|
self.run_script(f'{self._table.id}.newRow({list(items.values())}, "{self.id}")')
|
|
for key, val in items.items():
|
|
self[key] = val
|
|
|
|
def __setitem__(self, column, value):
|
|
if isinstance(column, tuple):
|
|
return [self.__setitem__(col, val) for col, val in zip(column, value)]
|
|
original_value = value
|
|
if column in self._table._formatters:
|
|
value = self._table._formatters[column].replace(self._table.VALUE, str(value))
|
|
self.run_script(f'{self._table.id}.updateCell("{self.id}", "{column}", "{value}")')
|
|
return super().__setitem__(column, original_value)
|
|
|
|
def background_color(self, column, color): self._style('backgroundColor', column, color)
|
|
|
|
def text_color(self, column, color): self._style('textColor', column, color)
|
|
|
|
def _style(self, style, column, arg):
|
|
self.run_script(f"{self._table.id}.rows[{self.id}]['{column}'].style.{style} = '{arg}'")
|
|
|
|
def delete(self):
|
|
self.run_script(f"{self._table.id}.deleteRow('{self.id}')")
|
|
self._table.pop(self.id)
|
|
|
|
|
|
class Table(Pane, dict):
|
|
VALUE = 'CELL__~__VALUE__~__PLACEHOLDER'
|
|
|
|
def __init__(self, window, width: NUM, height: NUM, headings: tuple, widths: tuple = None, alignments: tuple = None, position='left', draggable: bool = False, func: callable = None):
|
|
dict.__init__(self)
|
|
Pane.__init__(self, window)
|
|
self._formatters = {}
|
|
self.headings = headings
|
|
self.is_shown = True
|
|
self.win.handlers[self.id] = lambda rId: func(self[rId])
|
|
headings = list(headings)
|
|
widths = list(widths) if widths else []
|
|
alignments = list(alignments) if alignments else []
|
|
|
|
self.run_script(f'{self.id} = new Table({width}, {height}, {headings}, {widths}, {alignments}, "{position}", {jbool(draggable)})')
|
|
self.run_script(f'{self.id}.callbackName = "{self.id}"') if func else None
|
|
self.footer = Footer(self)
|
|
|
|
def new_row(self, *values, id=None) -> Row:
|
|
row_id = random.randint(0, 99_999_999) if not id else id
|
|
self[row_id] = Row(self, row_id, {heading: item for heading, item in zip(self.headings, values)})
|
|
return self[row_id]
|
|
|
|
def clear(self): self.run_script(f"{self.id}.clearRows()"), super().clear()
|
|
|
|
def get(self, __key: Union[int, str]) -> Row: return super().get(int(__key))
|
|
|
|
def __getitem__(self, item): return super().__getitem__(int(item))
|
|
|
|
def format(self, column: str, format_str: str): self._formatters[column] = format_str
|
|
|
|
def visible(self, visible: bool):
|
|
self.is_shown = visible
|
|
self.run_script(f"""
|
|
{self.id}.container.style.display = '{'block' if visible else 'none'}'
|
|
{self.id}.container.{'add' if visible else 'remove'}EventListener('mousedown', {self.id}.onMouseDown)
|
|
""")
|