345b37e0f3
- Added the chart.spinner method, which when set to `True` shows a loading spinner on the chart (a nice visual for API calls, large datasets etc). - If an empty data frame is passed to set (eg.`chart.set(pd.DataFrame())`) the volume series and candle series will be cleared. - added the `cumulative_volume` parameter to `update_from_tick`, which adds the given volume tick onto the latest bar. - Added `vert_visible` and `horz_visible` parameters to `crosshair`. - Small style improvements to the searchbox and topbar. - Fixed a bug preventing callbacks within `WxChart` and `QtChart` Thanks to @emma-uw for the following fixes and enhancements! - Methods `hide_data`, `show_data` and `price_line` can be used within Charts, Subcharts and Lines to change the visibility of data, price lines, and the price line labels. - Added the `delete` method to Line, which irreversably deletes the Line on the chart as well as its objects in JavaScript and Python. - Added the `lines` common method, which returns a list of all Line objects for the chart. - Added the `fit` method to the common methods, which uses the `fitContent()` method from Lightweight Charts. - Fixed a big which caused synced SubCharts to be out of sync upon loading. BETA: Polygon.io integration - Added the `PolygonChart` and `polygon` method, allowing for direct integration of polygon.io’s API. - This feature is still in beta, and there will be a full announcement and update once the feature is complete!
93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
import re
|
|
from random import choices
|
|
from string import ascii_lowercase
|
|
from typing import Literal
|
|
|
|
|
|
class MissingColumn(KeyError):
|
|
def __init__(self, message):
|
|
super().__init__(message)
|
|
self.msg = message
|
|
|
|
def __str__(self):
|
|
return f'{self.msg}'
|
|
|
|
|
|
class ColorError(ValueError):
|
|
def __init__(self, message):
|
|
super().__init__(message)
|
|
self.msg = message
|
|
|
|
def __str__(self):
|
|
return f'{self.msg}'
|
|
|
|
|
|
class IDGen(list):
|
|
def generate(self):
|
|
var = ''.join(choices(ascii_lowercase, k=8))
|
|
if var not in self:
|
|
self.append(var)
|
|
return var
|
|
self.generate()
|
|
|
|
|
|
def _valid_color(string):
|
|
if string[:3] == 'rgb' or string[:4] == 'rgba' or string[0] == '#':
|
|
return True
|
|
raise ColorError('Colors must be in the format of either rgb, rgba or hex.')
|
|
|
|
|
|
def _js_bool(b: bool): return 'true' if b is True else 'false' if b is False else None
|
|
|
|
|
|
LINE_STYLE = Literal['solid', 'dotted', 'dashed', 'large_dashed', 'sparse_dotted']
|
|
|
|
MARKER_POSITION = Literal['above', 'below', 'inside']
|
|
|
|
MARKER_SHAPE = Literal['arrow_up', 'arrow_down', 'circle', 'square']
|
|
|
|
CROSSHAIR_MODE = Literal['normal', 'magnet']
|
|
|
|
PRICE_SCALE_MODE = Literal['normal', 'logarithmic', 'percentage', 'index100']
|
|
|
|
|
|
def _line_style(line: LINE_STYLE):
|
|
js = 'LightweightCharts.LineStyle.'
|
|
return js+line[:line.index('_')].title() + line[line.index('_') + 1:].title() if '_' in line else js+line.title()
|
|
|
|
|
|
def _crosshair_mode(mode: CROSSHAIR_MODE):
|
|
return f'LightweightCharts.CrosshairMode.{mode.title()}' if mode else None
|
|
|
|
|
|
def _price_scale_mode(mode: PRICE_SCALE_MODE):
|
|
return f"LightweightCharts.PriceScaleMode.{'IndexedTo100' if mode == 'index100' else mode.title() if mode else None}"
|
|
|
|
|
|
def _marker_shape(shape: MARKER_SHAPE):
|
|
return shape[:shape.index('_')]+shape[shape.index('_')+1:].title() if '_' in shape else shape.title()
|
|
|
|
|
|
def _marker_position(p: MARKER_POSITION):
|
|
return {
|
|
'above': 'aboveBar',
|
|
'below': 'belowBar',
|
|
'inside': 'inBar',
|
|
None: None,
|
|
}[p]
|
|
|
|
|
|
def _convert_timeframe(timeframe):
|
|
spans = {
|
|
'min': 'minute',
|
|
'H': 'hour',
|
|
'D': 'day',
|
|
'W': 'week',
|
|
'M': 'month',
|
|
}
|
|
try:
|
|
multiplier = re.findall(r'\d+', timeframe)[0]
|
|
except IndexError:
|
|
return 1, spans[timeframe]
|
|
timespan = spans[timeframe.replace(multiplier, '')]
|
|
return multiplier, timespan |