fix_deribit_hist_queries #10
|
@ -28,6 +28,8 @@ from decimal import (
|
||||||
Decimal,
|
Decimal,
|
||||||
)
|
)
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
from pathlib import Path
|
||||||
|
from pprint import pformat
|
||||||
import time
|
import time
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
|
@ -37,8 +39,6 @@ from typing import (
|
||||||
|
|
||||||
from pendulum import now
|
from pendulum import now
|
||||||
import trio
|
import trio
|
||||||
from trio_typing import TaskStatus
|
|
||||||
from rapidfuzz import process as fuzzy
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from tractor.trionics import (
|
from tractor.trionics import (
|
||||||
broadcast_receiver,
|
broadcast_receiver,
|
||||||
|
@ -55,8 +55,10 @@ from cryptofeed.defines import (
|
||||||
OPTION, CALL, PUT
|
OPTION, CALL, PUT
|
||||||
)
|
)
|
||||||
from cryptofeed.symbols import Symbol
|
from cryptofeed.symbols import Symbol
|
||||||
|
|
||||||
# types for managing the cb callbacks.
|
# types for managing the cb callbacks.
|
||||||
# from cryptofeed.types import L1Book
|
# from cryptofeed.types import L1Book
|
||||||
|
from piker.brokers import SymbolNotFound
|
||||||
from .venues import (
|
from .venues import (
|
||||||
_ws_url,
|
_ws_url,
|
||||||
MarketType,
|
MarketType,
|
||||||
|
@ -64,9 +66,9 @@ from .venues import (
|
||||||
Pair,
|
Pair,
|
||||||
OptionPair,
|
OptionPair,
|
||||||
JSONRPCResult,
|
JSONRPCResult,
|
||||||
JSONRPCChannel,
|
# JSONRPCChannel,
|
||||||
KLinesResult,
|
KLinesResult,
|
||||||
Trade,
|
# Trade,
|
||||||
LastTradesResult,
|
LastTradesResult,
|
||||||
)
|
)
|
||||||
from piker.accounting import (
|
from piker.accounting import (
|
||||||
|
@ -77,7 +79,7 @@ from piker.accounting import (
|
||||||
from piker.data import (
|
from piker.data import (
|
||||||
def_iohlcv_fields,
|
def_iohlcv_fields,
|
||||||
match_from_pairs,
|
match_from_pairs,
|
||||||
Struct,
|
# Struct,
|
||||||
)
|
)
|
||||||
from piker.data._web_bs import (
|
from piker.data._web_bs import (
|
||||||
open_jsonrpc_session
|
open_jsonrpc_session
|
||||||
|
@ -97,7 +99,7 @@ _spawn_kwargs = {
|
||||||
|
|
||||||
|
|
||||||
# convert datetime obj timestamp to unixtime in milliseconds
|
# convert datetime obj timestamp to unixtime in milliseconds
|
||||||
def deribit_timestamp(when):
|
def deribit_timestamp(when) -> int:
|
||||||
return int((when.timestamp() * 1000) + (when.microsecond / 1000))
|
return int((when.timestamp() * 1000) + (when.microsecond / 1000))
|
||||||
|
|
||||||
|
|
||||||
|
@ -121,14 +123,20 @@ def str_to_cb_sym(name: str) -> Symbol:
|
||||||
type=OPTION,
|
type=OPTION,
|
||||||
strike_price=strike_price,
|
strike_price=strike_price,
|
||||||
option_type=option_type,
|
option_type=option_type,
|
||||||
expiry_date=new_expiry_date)
|
expiry_date=new_expiry_date
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def piker_sym_to_cb_sym(name: str) -> Symbol:
|
def piker_sym_to_cb_sym(name: str) -> Symbol:
|
||||||
base, expiry_date, strike_price, option_type = tuple(
|
(
|
||||||
|
base,
|
||||||
|
expiry_date,
|
||||||
|
strike_price,
|
||||||
|
option_type,
|
||||||
|
)= tuple(
|
||||||
name.upper().split('-'))
|
name.upper().split('-'))
|
||||||
|
|
||||||
quote = base
|
quote: str = base
|
||||||
|
|
||||||
if option_type == 'P':
|
if option_type == 'P':
|
||||||
option_type = PUT
|
option_type = PUT
|
||||||
|
@ -143,7 +151,8 @@ def piker_sym_to_cb_sym(name: str) -> Symbol:
|
||||||
type=OPTION,
|
type=OPTION,
|
||||||
strike_price=strike_price,
|
strike_price=strike_price,
|
||||||
option_type=option_type,
|
option_type=option_type,
|
||||||
expiry_date=expiry_date)
|
expiry_date=expiry_date
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def cb_sym_to_deribit_inst(sym: Symbol):
|
def cb_sym_to_deribit_inst(sym: Symbol):
|
||||||
|
@ -179,16 +188,18 @@ def get_config() -> dict[str, Any]:
|
||||||
|
|
||||||
conf: dict
|
conf: dict
|
||||||
path: Path
|
path: Path
|
||||||
|
|
||||||
conf, path = config.load(
|
conf, path = config.load(
|
||||||
conf_name='brokers',
|
conf_name='brokers',
|
||||||
touch_if_dne=True,
|
touch_if_dne=True,
|
||||||
)
|
)
|
||||||
section: dict = {}
|
section: dict|None = conf.get('deribit')
|
||||||
section = conf.get('deribit')
|
|
||||||
if section is None:
|
if section is None:
|
||||||
log.warning(f'No config section found for deribit in {path}')
|
raise ValueError(
|
||||||
return {}
|
f'No `[deribit]` section found in\n'
|
||||||
|
f'{path!r}\n\n'
|
||||||
|
f'See the template config from the core repo for samples..\n'
|
||||||
|
# f'<TODO put repo link here??>'
|
||||||
|
)
|
||||||
|
|
||||||
conf_option = section.get('option', {})
|
conf_option = section.get('option', {})
|
||||||
section.clear # clear the dict to reuse it
|
section.clear # clear the dict to reuse it
|
||||||
|
@ -204,7 +215,10 @@ def get_config() -> dict[str, Any]:
|
||||||
|
|
||||||
|
|
||||||
class Client:
|
class Client:
|
||||||
|
'''
|
||||||
|
Hi-level interface for the jsron-RPC over websocket API.
|
||||||
|
|
||||||
|
'''
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
|
||||||
|
@ -223,7 +237,11 @@ class Client:
|
||||||
self._auth_ts = None
|
self._auth_ts = None
|
||||||
self._auth_renew_ts = 5 # seconds to renew auth
|
self._auth_renew_ts = 5 # seconds to renew auth
|
||||||
|
|
||||||
async def _json_rpc_auth_wrapper(self, *args, **kwargs) -> JSONRPCResult:
|
async def _json_rpc_auth_wrapper(
|
||||||
|
self,
|
||||||
|
*args,
|
||||||
|
**kwargs,
|
||||||
|
) -> JSONRPCResult:
|
||||||
|
|
||||||
"""Background task that adquires a first access token and then will
|
"""Background task that adquires a first access token and then will
|
||||||
refresh the access token.
|
refresh the access token.
|
||||||
|
@ -250,9 +268,6 @@ class Client:
|
||||||
|
|
||||||
return await self.json_rpc(*args, **kwargs)
|
return await self.json_rpc(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def get_balances(
|
async def get_balances(
|
||||||
self,
|
self,
|
||||||
kind: str = 'option'
|
kind: str = 'option'
|
||||||
|
@ -277,23 +292,29 @@ class Client:
|
||||||
venue: str | None = None,
|
venue: str | None = None,
|
||||||
|
|
||||||
) -> dict[str, Asset]:
|
) -> dict[str, Asset]:
|
||||||
"""Return the set of asset balances for this account
|
'''
|
||||||
by symbol.
|
Return the set of asset balances for this account
|
||||||
"""
|
by (deribit's) symbol.
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
assets = {}
|
assets = {}
|
||||||
resp = await self._json_rpc_auth_wrapper(
|
resp = await self._json_rpc_auth_wrapper(
|
||||||
'public/get_currencies',
|
'public/get_currencies',
|
||||||
params={}
|
params={}
|
||||||
)
|
)
|
||||||
currencies = resp.result
|
currencies: list[dict] = resp.result
|
||||||
for currency in currencies:
|
for currency in currencies:
|
||||||
name = currency['currency']
|
name: str = currency['currency']
|
||||||
tx_tick = digits_to_dec(currency['fee_precision'])
|
tx_tick: Decimal = digits_to_dec(currency['fee_precision'])
|
||||||
atype='crypto_currency'
|
|
||||||
|
# TODO, handling of options, futures, perps etc. more
|
||||||
|
# specifically with diff `.atype`s?
|
||||||
assets[name] = Asset(
|
assets[name] = Asset(
|
||||||
name=name,
|
name=name,
|
||||||
atype=atype,
|
atype='crypto_currency',
|
||||||
tx_tick=tx_tick)
|
tx_tick=tx_tick,
|
||||||
|
)
|
||||||
|
|
||||||
instruments = await self.symbol_info(currency=name)
|
instruments = await self.symbol_info(currency=name)
|
||||||
for instrument in instruments:
|
for instrument in instruments:
|
||||||
|
@ -301,7 +322,8 @@ class Client:
|
||||||
assets[pair.symbol] = Asset(
|
assets[pair.symbol] = Asset(
|
||||||
name=pair.symbol,
|
name=pair.symbol,
|
||||||
atype=pair.venue,
|
atype=pair.venue,
|
||||||
tx_tick=pair.size_tick)
|
tx_tick=pair.size_tick,
|
||||||
|
)
|
||||||
|
|
||||||
return assets
|
return assets
|
||||||
|
|
||||||
|
@ -358,6 +380,19 @@ class Client:
|
||||||
return cached_pair
|
return cached_pair
|
||||||
|
|
||||||
if sym:
|
if sym:
|
||||||
|
opt: OptionPair|None = pair_table.get(sym)
|
||||||
|
if not opt:
|
||||||
|
closest_matches: dict[str, Pair] = match_from_pairs(
|
||||||
|
pairs=pair_table,
|
||||||
|
query=sym,
|
||||||
|
score_cutoff=40,
|
||||||
|
)
|
||||||
|
closest_syms: list[str] = list(closest_matches.keys())
|
||||||
|
raise ValueError(
|
||||||
|
f'No contract found for {sym!r}\n\n'
|
||||||
|
f'Closest {len(closest_syms)} available contracts:\n\n'
|
||||||
|
f'{pformat(closest_syms)}\n'
|
||||||
|
)
|
||||||
return pair_table[sym]
|
return pair_table[sym]
|
||||||
else:
|
else:
|
||||||
return self._pairs
|
return self._pairs
|
||||||
|
@ -381,7 +416,7 @@ class Client:
|
||||||
params: dict[str, str] = {
|
params: dict[str, str] = {
|
||||||
'currency': currency.upper(),
|
'currency': currency.upper(),
|
||||||
'kind': kind,
|
'kind': kind,
|
||||||
'expired': str(expired).lower()
|
'expired': expired,
|
||||||
}
|
}
|
||||||
|
|
||||||
resp: JSONRPCResult = await self._json_rpc_auth_wrapper(
|
resp: JSONRPCResult = await self._json_rpc_auth_wrapper(
|
||||||
|
@ -389,7 +424,7 @@ class Client:
|
||||||
params,
|
params,
|
||||||
)
|
)
|
||||||
# convert to symbol-keyed table
|
# convert to symbol-keyed table
|
||||||
pair_type: Type = PAIRTYPES[kind]
|
pair_type: Pair = PAIRTYPES[kind]
|
||||||
results: list[dict] | None = resp.result
|
results: list[dict] | None = resp.result
|
||||||
|
|
||||||
instruments: dict[str, Pair] = {}
|
instruments: dict[str, Pair] = {}
|
||||||
|
@ -427,12 +462,15 @@ class Client:
|
||||||
mkt_pairs = await self.symbol_info()
|
mkt_pairs = await self.symbol_info()
|
||||||
|
|
||||||
if not mkt_pairs:
|
if not mkt_pairs:
|
||||||
raise SymbolNotFound(f'No market pairs found!?:\n{resp}')
|
raise SymbolNotFound(
|
||||||
|
f'No market pairs found!?:\n'
|
||||||
|
f'{mkt_pairs}'
|
||||||
|
)
|
||||||
|
|
||||||
pairs_view_subtable: dict[str, Pair] = {}
|
pairs_view_subtable: dict[str, Pair] = {}
|
||||||
|
|
||||||
for instrument in mkt_pairs:
|
for instrument in mkt_pairs:
|
||||||
pair_type: Type = PAIRTYPES[venue]
|
pair_type: Pair|OptionPair = PAIRTYPES[venue]
|
||||||
|
|
||||||
pair: Pair = pair_type(**mkt_pairs[instrument].to_dict())
|
pair: Pair = pair_type(**mkt_pairs[instrument].to_dict())
|
||||||
|
|
||||||
|
@ -480,12 +518,14 @@ class Client:
|
||||||
if end_dt is None:
|
if end_dt is None:
|
||||||
end_dt = now('UTC')
|
end_dt = now('UTC')
|
||||||
|
|
||||||
|
_orig_start_dt = start_dt
|
||||||
if start_dt is None:
|
if start_dt is None:
|
||||||
start_dt = end_dt.start_of(
|
start_dt = end_dt.start_of(
|
||||||
'minute').subtract(minutes=limit)
|
'minute'
|
||||||
|
).subtract(minutes=limit)
|
||||||
|
|
||||||
start_time = deribit_timestamp(start_dt)
|
start_time: int = deribit_timestamp(start_dt)
|
||||||
end_time = deribit_timestamp(end_dt)
|
end_time: int = deribit_timestamp(end_dt)
|
||||||
|
|
||||||
# https://docs.deribit.com/#public-get_tradingview_chart_data
|
# https://docs.deribit.com/#public-get_tradingview_chart_data
|
||||||
resp = await self._json_rpc_auth_wrapper(
|
resp = await self._json_rpc_auth_wrapper(
|
||||||
|
@ -499,8 +539,12 @@ class Client:
|
||||||
|
|
||||||
result = KLinesResult(**resp.result)
|
result = KLinesResult(**resp.result)
|
||||||
new_bars: list[tuple] = []
|
new_bars: list[tuple] = []
|
||||||
for i in range(len(result.close)):
|
# if _orig_start_dt is None:
|
||||||
|
# if not new_bars:
|
||||||
|
# import tractor
|
||||||
|
# await tractor.pause()
|
||||||
|
|
||||||
|
for i in range(len(result.close)):
|
||||||
row = [
|
row = [
|
||||||
(start_time + (i * (60 * 1000))) / 1000.0, # time
|
(start_time + (i * (60 * 1000))) / 1000.0, # time
|
||||||
result.open[i],
|
result.open[i],
|
||||||
|
@ -575,18 +619,33 @@ async def aio_price_feed_relay(
|
||||||
from_trio: asyncio.Queue,
|
from_trio: asyncio.Queue,
|
||||||
to_trio: trio.abc.SendChannel,
|
to_trio: trio.abc.SendChannel,
|
||||||
) -> None:
|
) -> None:
|
||||||
async def _trade(data: dict, receipt_timestamp):
|
|
||||||
to_trio.send_nowait(('trade', {
|
async def _trade(
|
||||||
|
data: dict,
|
||||||
|
receipt_timestamp: int,
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Send `cryptofeed.FeedHandler` quotes to `piker`-side
|
||||||
|
`trio.Task`.
|
||||||
|
|
||||||
|
'''
|
||||||
|
to_trio.send_nowait((
|
||||||
|
'trade', {
|
||||||
'symbol': cb_sym_to_deribit_inst(
|
'symbol': cb_sym_to_deribit_inst(
|
||||||
str_to_cb_sym(data.symbol)).lower(),
|
str_to_cb_sym(data.symbol)).lower(),
|
||||||
'last': data,
|
'last': data,
|
||||||
'broker_ts': time.time(),
|
'broker_ts': time.time(),
|
||||||
'data': data.to_dict(),
|
'data': data.to_dict(),
|
||||||
'receipt': receipt_timestamp
|
'receipt': receipt_timestamp,
|
||||||
}))
|
},
|
||||||
|
))
|
||||||
|
|
||||||
async def _l1(data: dict, receipt_timestamp):
|
async def _l1(
|
||||||
to_trio.send_nowait(('l1', {
|
data: dict,
|
||||||
|
receipt_timestamp: int,
|
||||||
|
) -> None:
|
||||||
|
to_trio.send_nowait((
|
||||||
|
'l1', {
|
||||||
'symbol': cb_sym_to_deribit_inst(
|
'symbol': cb_sym_to_deribit_inst(
|
||||||
str_to_cb_sym(data.symbol)).lower(),
|
str_to_cb_sym(data.symbol)).lower(),
|
||||||
'ticks': [
|
'ticks': [
|
||||||
|
@ -611,7 +670,8 @@ async def aio_price_feed_relay(
|
||||||
'size': float(data.ask_size)
|
'size': float(data.ask_size)
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}))
|
},
|
||||||
|
))
|
||||||
sym: Symbol = piker_sym_to_cb_sym(instrument)
|
sym: Symbol = piker_sym_to_cb_sym(instrument)
|
||||||
fh.add_feed(
|
fh.add_feed(
|
||||||
DERIBIT,
|
DERIBIT,
|
||||||
|
@ -668,68 +728,69 @@ async def maybe_open_price_feed(
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def aio_order_feed_relay(
|
# TODO, move all to `.broker` submod!
|
||||||
fh: FeedHandler,
|
# async def aio_order_feed_relay(
|
||||||
instrument: Symbol,
|
# fh: FeedHandler,
|
||||||
from_trio: asyncio.Queue,
|
# instrument: Symbol,
|
||||||
to_trio: trio.abc.SendChannel,
|
# from_trio: asyncio.Queue,
|
||||||
) -> None:
|
# to_trio: trio.abc.SendChannel,
|
||||||
async def _fill(data: dict, receipt_timestamp):
|
# ) -> None:
|
||||||
breakpoint()
|
# async def _fill(data: dict, receipt_timestamp):
|
||||||
|
# breakpoint()
|
||||||
|
|
||||||
async def _order_info(data: dict, receipt_timestamp):
|
# async def _order_info(data: dict, receipt_timestamp):
|
||||||
breakpoint()
|
# breakpoint()
|
||||||
|
|
||||||
fh.add_feed(
|
# fh.add_feed(
|
||||||
DERIBIT,
|
# DERIBIT,
|
||||||
channels=[FILLS, ORDER_INFO],
|
# channels=[FILLS, ORDER_INFO],
|
||||||
symbols=[instrument.upper()],
|
# symbols=[instrument.upper()],
|
||||||
callbacks={
|
# callbacks={
|
||||||
FILLS: _fill,
|
# FILLS: _fill,
|
||||||
ORDER_INFO: _order_info,
|
# ORDER_INFO: _order_info,
|
||||||
})
|
# })
|
||||||
|
|
||||||
if not fh.running:
|
# if not fh.running:
|
||||||
fh.run(
|
# fh.run(
|
||||||
start_loop=False,
|
# start_loop=False,
|
||||||
install_signal_handlers=False)
|
# install_signal_handlers=False)
|
||||||
|
|
||||||
# sync with trio
|
# # sync with trio
|
||||||
to_trio.send_nowait(None)
|
# to_trio.send_nowait(None)
|
||||||
|
|
||||||
await asyncio.sleep(float('inf'))
|
# await asyncio.sleep(float('inf'))
|
||||||
|
|
||||||
|
|
||||||
@acm
|
# @acm
|
||||||
async def open_order_feed(
|
# async def open_order_feed(
|
||||||
instrument: list[str]
|
# instrument: list[str]
|
||||||
) -> trio.abc.ReceiveStream:
|
# ) -> trio.abc.ReceiveStream:
|
||||||
async with maybe_open_feed_handler() as fh:
|
# async with maybe_open_feed_handler() as fh:
|
||||||
async with to_asyncio.open_channel_from(
|
# async with to_asyncio.open_channel_from(
|
||||||
partial(
|
# partial(
|
||||||
aio_order_feed_relay,
|
# aio_order_feed_relay,
|
||||||
fh,
|
# fh,
|
||||||
instrument
|
# instrument
|
||||||
)
|
# )
|
||||||
) as (first, chan):
|
# ) as (first, chan):
|
||||||
yield chan
|
# yield chan
|
||||||
|
|
||||||
|
|
||||||
@acm
|
# @acm
|
||||||
async def maybe_open_order_feed(
|
# async def maybe_open_order_feed(
|
||||||
instrument: str
|
# instrument: str
|
||||||
) -> trio.abc.ReceiveStream:
|
# ) -> trio.abc.ReceiveStream:
|
||||||
|
|
||||||
# TODO: add a predicate to maybe_open_context
|
# # TODO: add a predicate to maybe_open_context
|
||||||
async with maybe_open_context(
|
# async with maybe_open_context(
|
||||||
acm_func=open_order_feed,
|
# acm_func=open_order_feed,
|
||||||
kwargs={
|
# kwargs={
|
||||||
'instrument': instrument.split('.')[0],
|
# 'instrument': instrument.split('.')[0],
|
||||||
'fh': fh
|
# 'fh': fh
|
||||||
},
|
# },
|
||||||
key=f'{instrument.split('.')[0]}-order',
|
# key=f'{instrument.split('.')[0]}-order',
|
||||||
) as (cache_hit, feed):
|
# ) as (cache_hit, feed):
|
||||||
if cache_hit:
|
# if cache_hit:
|
||||||
yield broadcast_receiver(feed, 10)
|
# yield broadcast_receiver(feed, 10)
|
||||||
else:
|
# else:
|
||||||
yield feed
|
# yield feed
|
||||||
|
|
|
@ -18,56 +18,57 @@
|
||||||
Deribit backend.
|
Deribit backend.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
from __future__ import annotations
|
||||||
from contextlib import asynccontextmanager as acm
|
from contextlib import asynccontextmanager as acm
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Any, Optional, Callable
|
from typing import (
|
||||||
from pprint import pformat
|
# Any,
|
||||||
|
# Optional,
|
||||||
|
Callable,
|
||||||
|
)
|
||||||
|
# from pprint import pformat
|
||||||
import time
|
import time
|
||||||
|
|
||||||
|
import cryptofeed
|
||||||
import trio
|
import trio
|
||||||
from trio_typing import TaskStatus
|
from trio_typing import TaskStatus
|
||||||
from pendulum import (
|
from pendulum import (
|
||||||
from_timestamp,
|
from_timestamp,
|
||||||
now,
|
|
||||||
)
|
)
|
||||||
from rapidfuzz import process as fuzzy
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
from piker.accounting import (
|
from piker.accounting import (
|
||||||
|
Asset,
|
||||||
MktPair,
|
MktPair,
|
||||||
unpack_fqme,
|
unpack_fqme,
|
||||||
)
|
)
|
||||||
from piker.brokers import (
|
from piker.brokers import (
|
||||||
open_cached_client,
|
open_cached_client,
|
||||||
NoData,
|
NoData,
|
||||||
|
DataUnavailable,
|
||||||
)
|
)
|
||||||
from piker._cacheables import (
|
from piker._cacheables import (
|
||||||
async_lifo_cache,
|
async_lifo_cache,
|
||||||
)
|
)
|
||||||
from piker.log import get_logger, get_console_log
|
from piker.log import (
|
||||||
from piker.data import ShmArray
|
get_logger,
|
||||||
|
mk_repr,
|
||||||
|
)
|
||||||
from piker.data.validate import FeedInit
|
from piker.data.validate import FeedInit
|
||||||
from piker.brokers._util import (
|
|
||||||
BrokerError,
|
|
||||||
DataUnavailable,
|
|
||||||
)
|
|
||||||
|
|
||||||
from cryptofeed import FeedHandler
|
|
||||||
from cryptofeed.defines import (
|
|
||||||
DERIBIT, L1_BOOK, TRADES, OPTION, CALL, PUT
|
|
||||||
)
|
|
||||||
from cryptofeed.symbols import Symbol
|
|
||||||
|
|
||||||
from .api import (
|
from .api import (
|
||||||
Client, Trade,
|
Client,
|
||||||
get_config,
|
# get_config,
|
||||||
piker_sym_to_cb_sym, cb_sym_to_deribit_inst,
|
piker_sym_to_cb_sym,
|
||||||
|
cb_sym_to_deribit_inst,
|
||||||
maybe_open_price_feed
|
maybe_open_price_feed
|
||||||
)
|
)
|
||||||
from .venues import (
|
from .venues import (
|
||||||
Pair,
|
Pair,
|
||||||
OptionPair,
|
OptionPair,
|
||||||
|
Trade,
|
||||||
)
|
)
|
||||||
|
|
||||||
_spawn_kwargs = {
|
_spawn_kwargs = {
|
||||||
|
@ -86,6 +87,10 @@ async def open_history_client(
|
||||||
# TODO implement history getter for the new storage layer.
|
# TODO implement history getter for the new storage layer.
|
||||||
async with open_cached_client('deribit') as client:
|
async with open_cached_client('deribit') as client:
|
||||||
|
|
||||||
|
pair: OptionPair = client._pairs[mkt.dst.name]
|
||||||
|
# XXX NOTE, the cuckers use ms !!!
|
||||||
|
creation_time_s: int = pair.creation_timestamp/1000
|
||||||
|
|
||||||
async def get_ohlc(
|
async def get_ohlc(
|
||||||
timeframe: float,
|
timeframe: float,
|
||||||
end_dt: datetime | None = None,
|
end_dt: datetime | None = None,
|
||||||
|
@ -105,6 +110,31 @@ async def open_history_client(
|
||||||
end_dt=end_dt,
|
end_dt=end_dt,
|
||||||
)
|
)
|
||||||
if len(array) == 0:
|
if len(array) == 0:
|
||||||
|
if (
|
||||||
|
end_dt is None
|
||||||
|
):
|
||||||
|
raise DataUnavailable(
|
||||||
|
'No history seems to exist yet?\n\n'
|
||||||
|
f'{mkt}'
|
||||||
|
)
|
||||||
|
elif (
|
||||||
|
end_dt
|
||||||
|
and
|
||||||
|
end_dt.timestamp() < creation_time_s
|
||||||
|
):
|
||||||
|
# the contract can't have history
|
||||||
|
# before it was created.
|
||||||
|
pair_type_str: str = type(pair).__name__
|
||||||
|
create_dt: datetime = from_timestamp(creation_time_s)
|
||||||
|
raise DataUnavailable(
|
||||||
|
f'No history prior to\n'
|
||||||
|
f'`{pair_type_str}.creation_timestamp: int = '
|
||||||
|
f'{pair.creation_timestamp}\n\n'
|
||||||
|
f'------ deribit sux ------\n'
|
||||||
|
f'WHICH IN "NORMAL PEOPLE WHO USE EPOCH TIME" form is,\n'
|
||||||
|
f'creation_time_s: {creation_time_s}\n'
|
||||||
|
f'create_dt: {create_dt}\n'
|
||||||
|
)
|
||||||
raise NoData(
|
raise NoData(
|
||||||
f'No frame for {start_dt} -> {end_dt}\n'
|
f'No frame for {start_dt} -> {end_dt}\n'
|
||||||
)
|
)
|
||||||
|
@ -126,14 +156,20 @@ async def open_history_client(
|
||||||
|
|
||||||
return array, start_dt, end_dt
|
return array, start_dt, end_dt
|
||||||
|
|
||||||
yield get_ohlc, {'erlangs': 3, 'rate': 3}
|
yield (
|
||||||
|
get_ohlc,
|
||||||
|
{ # backfill config
|
||||||
|
'erlangs': 3,
|
||||||
|
'rate': 3,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@async_lifo_cache()
|
@async_lifo_cache()
|
||||||
async def get_mkt_info(
|
async def get_mkt_info(
|
||||||
fqme: str,
|
fqme: str,
|
||||||
|
|
||||||
) -> tuple[MktPair, Pair] | None:
|
) -> tuple[MktPair, Pair|OptionPair] | None:
|
||||||
|
|
||||||
# uppercase since kraken bs_mktid is always upper
|
# uppercase since kraken bs_mktid is always upper
|
||||||
if 'deribit' not in fqme.lower():
|
if 'deribit' not in fqme.lower():
|
||||||
|
@ -149,7 +185,7 @@ async def get_mkt_info(
|
||||||
# returns, always!
|
# returns, always!
|
||||||
expiry: str = expiry.upper()
|
expiry: str = expiry.upper()
|
||||||
venue: str = venue.upper()
|
venue: str = venue.upper()
|
||||||
venue_lower: str = venue.lower()
|
# venue_lower: str = venue.lower()
|
||||||
|
|
||||||
mkt_mode: str = 'option'
|
mkt_mode: str = 'option'
|
||||||
|
|
||||||
|
@ -175,52 +211,64 @@ async def get_mkt_info(
|
||||||
price_tick=pair.price_tick,
|
price_tick=pair.price_tick,
|
||||||
size_tick=pair.size_tick,
|
size_tick=pair.size_tick,
|
||||||
bs_mktid=pair.symbol,
|
bs_mktid=pair.symbol,
|
||||||
expiry=pair.expiry,
|
|
||||||
venue=mkt_mode,
|
venue=mkt_mode,
|
||||||
broker='deribit',
|
broker='deribit',
|
||||||
_atype=mkt_mode,
|
_atype=mkt_mode,
|
||||||
_fqme_without_src=True,
|
_fqme_without_src=True,
|
||||||
|
|
||||||
|
# expiry=pair.expiry,
|
||||||
|
# XXX TODO, currently we don't use it since it's
|
||||||
|
# already "described" in the `OptionPair.symbol: str`
|
||||||
|
# and if we slap in the ISO repr it's kinda hideous..
|
||||||
|
# -[ ] figure out the best either std
|
||||||
)
|
)
|
||||||
return mkt, pair
|
return mkt, pair
|
||||||
|
|
||||||
|
|
||||||
async def stream_quotes(
|
async def stream_quotes(
|
||||||
|
|
||||||
send_chan: trio.abc.SendChannel,
|
send_chan: trio.abc.SendChannel,
|
||||||
symbols: list[str],
|
symbols: list[str],
|
||||||
feed_is_live: trio.Event,
|
feed_is_live: trio.Event,
|
||||||
loglevel: str = None,
|
|
||||||
|
|
||||||
# startup sync
|
# startup sync
|
||||||
task_status: TaskStatus[tuple[dict, dict]] = trio.TASK_STATUS_IGNORED,
|
task_status: TaskStatus[tuple[dict, dict]] = trio.TASK_STATUS_IGNORED,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
# XXX: required to propagate ``tractor`` loglevel to piker logging
|
'''
|
||||||
get_console_log(loglevel or tractor.current_actor().loglevel)
|
Open a live quote stream for the market set defined by `symbols`.
|
||||||
|
|
||||||
|
'''
|
||||||
sym = symbols[0].split('.')[0]
|
sym = symbols[0].split('.')[0]
|
||||||
|
|
||||||
init_msgs: list[FeedInit] = []
|
init_msgs: list[FeedInit] = []
|
||||||
|
|
||||||
|
# multiline nested `dict` formatter (since rn quote-msgs are
|
||||||
|
# just that).
|
||||||
|
pfmt: Callable[[str], str] = mk_repr()
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
open_cached_client('deribit') as client,
|
open_cached_client('deribit') as client,
|
||||||
send_chan as send_chan
|
send_chan as send_chan
|
||||||
):
|
):
|
||||||
|
mkt: MktPair
|
||||||
|
pair: Pair
|
||||||
mkt, pair = await get_mkt_info(sym)
|
mkt, pair = await get_mkt_info(sym)
|
||||||
|
|
||||||
# build out init msgs according to latest spec
|
# build out init msgs according to latest spec
|
||||||
init_msgs.append(
|
init_msgs.append(
|
||||||
FeedInit(mkt_info=mkt)
|
FeedInit(
|
||||||
|
mkt_info=mkt,
|
||||||
)
|
)
|
||||||
nsym = piker_sym_to_cb_sym(sym)
|
)
|
||||||
|
# build `cryptofeed` feed-handle
|
||||||
|
cf_sym: cryptofeed.Symbol = piker_sym_to_cb_sym(sym)
|
||||||
|
|
||||||
async with maybe_open_price_feed(sym) as stream:
|
async with maybe_open_price_feed(sym) as stream:
|
||||||
|
last_trades = (
|
||||||
cache = client._pairs
|
await client.last_trades(
|
||||||
|
cb_sym_to_deribit_inst(cf_sym),
|
||||||
last_trades = (await client.last_trades(
|
count=1,
|
||||||
cb_sym_to_deribit_inst(nsym), count=1)).trades
|
)
|
||||||
|
).trades
|
||||||
|
|
||||||
if len(last_trades) == 0:
|
if len(last_trades) == 0:
|
||||||
last_trade = None
|
last_trade = None
|
||||||
|
@ -243,13 +291,29 @@ async def stream_quotes(
|
||||||
'broker_ts': last_trade.timestamp
|
'broker_ts': last_trade.timestamp
|
||||||
}]
|
}]
|
||||||
}
|
}
|
||||||
task_status.started((init_msgs, first_quote))
|
task_status.started((
|
||||||
|
init_msgs,
|
||||||
|
first_quote,
|
||||||
|
))
|
||||||
|
|
||||||
feed_is_live.set()
|
feed_is_live.set()
|
||||||
|
|
||||||
|
# NOTE XXX, static for now!
|
||||||
|
# => since this only handles ONE mkt feed at a time we
|
||||||
|
# don't need a lookup table to map interleaved quotes
|
||||||
|
# from multiple possible mkt-pairs
|
||||||
|
topic: str = mkt.bs_fqme
|
||||||
|
|
||||||
|
# deliver until cancelled
|
||||||
async for typ, quote in stream:
|
async for typ, quote in stream:
|
||||||
topic = quote['symbol']
|
sym: str = quote['symbol']
|
||||||
await send_chan.send({topic: quote})
|
log.info(
|
||||||
|
f'deribit {typ!r} quote for {sym!r}\n\n'
|
||||||
|
f'{pfmt(quote)}\n'
|
||||||
|
)
|
||||||
|
await send_chan.send({
|
||||||
|
topic: quote,
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
|
@ -259,13 +323,14 @@ async def open_symbol_search(
|
||||||
async with open_cached_client('deribit') as client:
|
async with open_cached_client('deribit') as client:
|
||||||
|
|
||||||
# load all symbols locally for fast search
|
# load all symbols locally for fast search
|
||||||
cache = client._pairs
|
# cache = client._pairs
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
|
|
||||||
async with ctx.open_stream() as stream:
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
pattern: str
|
pattern: str
|
||||||
async for pattern in stream:
|
async for pattern in stream:
|
||||||
|
|
||||||
# NOTE: pattern fuzzy-matching is done within
|
# NOTE: pattern fuzzy-matching is done within
|
||||||
# the methd impl.
|
# the methd impl.
|
||||||
pairs: dict[str, Pair] = await client.search_symbols(
|
pairs: dict[str, Pair] = await client.search_symbols(
|
||||||
|
|
|
@ -22,11 +22,10 @@ from __future__ import annotations
|
||||||
import pendulum
|
import pendulum
|
||||||
from typing import (
|
from typing import (
|
||||||
Literal,
|
Literal,
|
||||||
|
Optional,
|
||||||
)
|
)
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
|
|
||||||
from msgspec import field
|
|
||||||
|
|
||||||
from piker.types import Struct
|
from piker.types import Struct
|
||||||
|
|
||||||
|
|
||||||
|
@ -111,18 +110,21 @@ class OptionPair(Pair, frozen=True):
|
||||||
block_trade_min_trade_amount: int # '25'
|
block_trade_min_trade_amount: int # '25'
|
||||||
block_trade_commission: float # '0.003'
|
block_trade_commission: float # '0.003'
|
||||||
|
|
||||||
|
|
||||||
# NOTE: see `.data._symcache.SymbologyCache.load()` for why
|
# NOTE: see `.data._symcache.SymbologyCache.load()` for why
|
||||||
ns_path: str = 'piker.brokers.deribit:OptionPair'
|
ns_path: str = 'piker.brokers.deribit:OptionPair'
|
||||||
|
|
||||||
|
# TODO, impl this without the MM:SS part of
|
||||||
|
# the `'THH:MM:SS..'` etc..
|
||||||
@property
|
@property
|
||||||
def expiry(self) -> str:
|
def expiry(self) -> str:
|
||||||
iso_date = pendulum.from_timestamp(self.expiration_timestamp / 1000).isoformat()
|
iso_date = pendulum.from_timestamp(
|
||||||
|
self.expiration_timestamp / 1000
|
||||||
|
).isoformat()
|
||||||
return iso_date
|
return iso_date
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def venue(self) -> str:
|
def venue(self) -> str:
|
||||||
return 'option'
|
return f'{self.instrument_type}_option'
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def bs_fqme(self) -> str:
|
def bs_fqme(self) -> str:
|
||||||
|
|
|
@ -30,6 +30,7 @@ import time
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
AsyncIterator,
|
AsyncIterator,
|
||||||
|
Callable,
|
||||||
TYPE_CHECKING,
|
TYPE_CHECKING,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -54,6 +55,9 @@ from ._util import (
|
||||||
get_console_log,
|
get_console_log,
|
||||||
)
|
)
|
||||||
from ..service import maybe_spawn_daemon
|
from ..service import maybe_spawn_daemon
|
||||||
|
from piker.log import (
|
||||||
|
mk_repr,
|
||||||
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ._sharedmem import (
|
from ._sharedmem import (
|
||||||
|
@ -575,7 +579,6 @@ async def open_sample_stream(
|
||||||
|
|
||||||
|
|
||||||
async def sample_and_broadcast(
|
async def sample_and_broadcast(
|
||||||
|
|
||||||
bus: _FeedsBus, # noqa
|
bus: _FeedsBus, # noqa
|
||||||
rt_shm: ShmArray,
|
rt_shm: ShmArray,
|
||||||
hist_shm: ShmArray,
|
hist_shm: ShmArray,
|
||||||
|
@ -596,11 +599,22 @@ async def sample_and_broadcast(
|
||||||
|
|
||||||
overruns = Counter()
|
overruns = Counter()
|
||||||
|
|
||||||
|
# multiline nested `dict` formatter (since rn quote-msgs are
|
||||||
|
# just that).
|
||||||
|
pfmt: Callable[[str], str] = mk_repr()
|
||||||
|
|
||||||
# iterate stream delivered by broker
|
# iterate stream delivered by broker
|
||||||
async for quotes in quote_stream:
|
async for quotes in quote_stream:
|
||||||
# print(quotes)
|
|
||||||
|
|
||||||
# TODO: ``numba`` this!
|
# XXX WARNING XXX only enable for debugging bc ow can cost
|
||||||
|
# ALOT of perf with HF-feedz!!!
|
||||||
|
#
|
||||||
|
# log.info(
|
||||||
|
# 'Rx live quotes:\n'
|
||||||
|
# f'{pfmt(quotes)}'
|
||||||
|
# )
|
||||||
|
|
||||||
|
# TODO: `numba` this!
|
||||||
for broker_symbol, quote in quotes.items():
|
for broker_symbol, quote in quotes.items():
|
||||||
# TODO: in theory you can send the IPC msg *before* writing
|
# TODO: in theory you can send the IPC msg *before* writing
|
||||||
# to the sharedmem array to decrease latency, however, that
|
# to the sharedmem array to decrease latency, however, that
|
||||||
|
@ -673,6 +687,18 @@ async def sample_and_broadcast(
|
||||||
sub_key: str = broker_symbol.lower()
|
sub_key: str = broker_symbol.lower()
|
||||||
subs: set[Sub] = bus.get_subs(sub_key)
|
subs: set[Sub] = bus.get_subs(sub_key)
|
||||||
|
|
||||||
|
if not subs:
|
||||||
|
all_bs_fqmes: list[str] = list(
|
||||||
|
bus._subscribers.keys()
|
||||||
|
)
|
||||||
|
log.warning(
|
||||||
|
f'No subscribers for {brokername!r} live-quote ??\n'
|
||||||
|
f'broker_symbol: {broker_symbol}\n\n'
|
||||||
|
|
||||||
|
f'Maybe the backend-sys symbol does not match one of,\n'
|
||||||
|
f'{pfmt(all_bs_fqmes)}\n'
|
||||||
|
)
|
||||||
|
|
||||||
# NOTE: by default the broker backend doesn't append
|
# NOTE: by default the broker backend doesn't append
|
||||||
# it's own "name" into the fqme schema (but maybe it
|
# it's own "name" into the fqme schema (but maybe it
|
||||||
# should?) so we have to manually generate the correct
|
# should?) so we have to manually generate the correct
|
||||||
|
|
|
@ -360,7 +360,7 @@ async def open_autorecon_ws(
|
||||||
|
|
||||||
'''
|
'''
|
||||||
JSONRPC response-request style machinery for transparent multiplexing
|
JSONRPC response-request style machinery for transparent multiplexing
|
||||||
of msgs over a NoBsWs.
|
of msgs over a `NoBsWs`.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
@ -377,6 +377,16 @@ async def open_jsonrpc_session(
|
||||||
url: str,
|
url: str,
|
||||||
start_id: int = 0,
|
start_id: int = 0,
|
||||||
response_type: type = JSONRPCResult,
|
response_type: type = JSONRPCResult,
|
||||||
|
msg_recv_timeout: float = float('inf'),
|
||||||
|
# ^NOTE, since only `deribit` is using this jsonrpc stuff atm
|
||||||
|
# and options mkts are generally "slow moving"..
|
||||||
|
#
|
||||||
|
# FURTHER if we break the underlying ws connection then since we
|
||||||
|
# don't pass a `fixture` to the task that manages `NoBsWs`, i.e.
|
||||||
|
# `_reconnect_forever()`, the jsonrpc "transport pipe" get's
|
||||||
|
# broken and never restored with wtv init sequence is required to
|
||||||
|
# re-establish a working req-resp session.
|
||||||
|
|
||||||
# request_type: Optional[type] = None,
|
# request_type: Optional[type] = None,
|
||||||
# request_hook: Optional[Callable] = None,
|
# request_hook: Optional[Callable] = None,
|
||||||
# error_hook: Optional[Callable] = None,
|
# error_hook: Optional[Callable] = None,
|
||||||
|
@ -388,12 +398,18 @@ async def open_jsonrpc_session(
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
trio.open_nursery() as n,
|
trio.open_nursery() as n,
|
||||||
open_autorecon_ws(url) as ws
|
open_autorecon_ws(
|
||||||
|
url=url,
|
||||||
|
msg_recv_timeout=msg_recv_timeout,
|
||||||
|
) as ws
|
||||||
):
|
):
|
||||||
rpc_id: Iterable[int] = count(start_id)
|
rpc_id: Iterable[int] = count(start_id)
|
||||||
rpc_results: dict[int, dict] = {}
|
rpc_results: dict[int, dict] = {}
|
||||||
|
|
||||||
async def json_rpc(method: str, params: dict) -> dict:
|
async def json_rpc(
|
||||||
|
method: str,
|
||||||
|
params: dict,
|
||||||
|
) -> dict:
|
||||||
'''
|
'''
|
||||||
perform a json rpc call and wait for the result, raise exception in
|
perform a json rpc call and wait for the result, raise exception in
|
||||||
case of error field present on response
|
case of error field present on response
|
||||||
|
|
|
@ -540,7 +540,10 @@ async def open_feed_bus(
|
||||||
# subscription since the backend isn't (yet) expected to
|
# subscription since the backend isn't (yet) expected to
|
||||||
# append it's own name to the fqme, so we filter on keys
|
# append it's own name to the fqme, so we filter on keys
|
||||||
# which *do not* include that name (e.g .ib) .
|
# which *do not* include that name (e.g .ib) .
|
||||||
bus._subscribers.setdefault(bs_fqme, set())
|
bus._subscribers.setdefault(
|
||||||
|
bs_fqme,
|
||||||
|
set(),
|
||||||
|
)
|
||||||
|
|
||||||
# sync feed subscribers with flume handles
|
# sync feed subscribers with flume handles
|
||||||
await ctx.started(
|
await ctx.started(
|
||||||
|
|
28
piker/log.py
28
piker/log.py
|
@ -18,7 +18,11 @@
|
||||||
Log like a forester!
|
Log like a forester!
|
||||||
"""
|
"""
|
||||||
import logging
|
import logging
|
||||||
|
import reprlib
|
||||||
import json
|
import json
|
||||||
|
from typing import (
|
||||||
|
Callable,
|
||||||
|
)
|
||||||
|
|
||||||
import tractor
|
import tractor
|
||||||
from pygments import (
|
from pygments import (
|
||||||
|
@ -84,3 +88,27 @@ def colorize_json(
|
||||||
# likeable styles: algol_nu, tango, monokai
|
# likeable styles: algol_nu, tango, monokai
|
||||||
formatters.TerminalTrueColorFormatter(style=style)
|
formatters.TerminalTrueColorFormatter(style=style)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def mk_repr(
|
||||||
|
**repr_kws,
|
||||||
|
) -> Callable[[str], str]:
|
||||||
|
'''
|
||||||
|
Allocate and deliver a `repr.Repr` instance with provided input
|
||||||
|
settings using the std-lib's `reprlib` mod,
|
||||||
|
* https://docs.python.org/3/library/reprlib.html
|
||||||
|
|
||||||
|
------ Ex. ------
|
||||||
|
An up to 6-layer-nested `dict` as multi-line:
|
||||||
|
- https://stackoverflow.com/a/79102479
|
||||||
|
- https://docs.python.org/3/library/reprlib.html#reprlib.Repr.maxlevel
|
||||||
|
|
||||||
|
'''
|
||||||
|
def_kws: dict[str, int] = dict(
|
||||||
|
indent=2,
|
||||||
|
maxlevel=6, # recursion levels
|
||||||
|
maxstring=66, # match editor line-len limit
|
||||||
|
)
|
||||||
|
def_kws |= repr_kws
|
||||||
|
reprr = reprlib.Repr(**def_kws)
|
||||||
|
return reprr.repr
|
||||||
|
|
|
@ -161,7 +161,12 @@ class NativeStorageClient:
|
||||||
|
|
||||||
def index_files(self):
|
def index_files(self):
|
||||||
for path in self._datadir.iterdir():
|
for path in self._datadir.iterdir():
|
||||||
if path.name in {'borked', 'expired',}:
|
if (
|
||||||
|
path.name in {'borked', 'expired',}
|
||||||
|
or
|
||||||
|
'.parquet' not in str(path)
|
||||||
|
):
|
||||||
|
# ignore all non-apache files (for now)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
key: str = path.name.rstrip('.parquet')
|
key: str = path.name.rstrip('.parquet')
|
||||||
|
|
|
@ -44,8 +44,10 @@ import trio
|
||||||
from trio_typing import TaskStatus
|
from trio_typing import TaskStatus
|
||||||
import tractor
|
import tractor
|
||||||
from pendulum import (
|
from pendulum import (
|
||||||
|
Interval,
|
||||||
DateTime,
|
DateTime,
|
||||||
Duration,
|
Duration,
|
||||||
|
duration as mk_duration,
|
||||||
from_timestamp,
|
from_timestamp,
|
||||||
)
|
)
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
@ -214,7 +216,8 @@ async def maybe_fill_null_segments(
|
||||||
# pair, immediately stop backfilling?
|
# pair, immediately stop backfilling?
|
||||||
if (
|
if (
|
||||||
start_dt
|
start_dt
|
||||||
and end_dt < start_dt
|
and
|
||||||
|
end_dt < start_dt
|
||||||
):
|
):
|
||||||
await tractor.pause()
|
await tractor.pause()
|
||||||
break
|
break
|
||||||
|
@ -262,6 +265,7 @@ async def maybe_fill_null_segments(
|
||||||
except tractor.ContextCancelled:
|
except tractor.ContextCancelled:
|
||||||
# log.exception
|
# log.exception
|
||||||
await tractor.pause()
|
await tractor.pause()
|
||||||
|
raise
|
||||||
|
|
||||||
null_segs_detected.set()
|
null_segs_detected.set()
|
||||||
# RECHECK for more null-gaps
|
# RECHECK for more null-gaps
|
||||||
|
@ -349,7 +353,7 @@ async def maybe_fill_null_segments(
|
||||||
|
|
||||||
async def start_backfill(
|
async def start_backfill(
|
||||||
get_hist,
|
get_hist,
|
||||||
frame_types: dict[str, Duration] | None,
|
def_frame_duration: Duration,
|
||||||
mod: ModuleType,
|
mod: ModuleType,
|
||||||
mkt: MktPair,
|
mkt: MktPair,
|
||||||
shm: ShmArray,
|
shm: ShmArray,
|
||||||
|
@ -379,22 +383,23 @@ async def start_backfill(
|
||||||
update_start_on_prepend: bool = False
|
update_start_on_prepend: bool = False
|
||||||
if backfill_until_dt is None:
|
if backfill_until_dt is None:
|
||||||
|
|
||||||
# TODO: drop this right and just expose the backfill
|
# TODO: per-provider default history-durations?
|
||||||
# limits inside a [storage] section in conf.toml?
|
# -[ ] inside the `open_history_client()` config allow
|
||||||
# when no tsdb "last datum" is provided, we just load
|
# declaring the history duration limits instead of
|
||||||
# some near-term history.
|
# guessing and/or applying the same limits to all?
|
||||||
# periods = {
|
#
|
||||||
# 1: {'days': 1},
|
# -[ ] allow declaring (default) per-provider backfill
|
||||||
# 60: {'days': 14},
|
# limits inside a [storage] sub-section in conf.toml?
|
||||||
# }
|
#
|
||||||
|
# NOTE, when no tsdb "last datum" is provided, we just
|
||||||
# do a decently sized backfill and load it into storage.
|
# load some near-term history by presuming a "decently
|
||||||
|
# large" 60s duration limit and a much shorter 1s range.
|
||||||
periods = {
|
periods = {
|
||||||
1: {'days': 2},
|
1: {'days': 2},
|
||||||
60: {'years': 6},
|
60: {'years': 6},
|
||||||
}
|
}
|
||||||
period_duration: int = periods[timeframe]
|
period_duration: int = periods[timeframe]
|
||||||
update_start_on_prepend = True
|
update_start_on_prepend: bool = True
|
||||||
|
|
||||||
# NOTE: manually set the "latest" datetime which we intend to
|
# NOTE: manually set the "latest" datetime which we intend to
|
||||||
# backfill history "until" so as to adhere to the history
|
# backfill history "until" so as to adhere to the history
|
||||||
|
@ -416,7 +421,6 @@ async def start_backfill(
|
||||||
f'backfill_until_dt: {backfill_until_dt}\n'
|
f'backfill_until_dt: {backfill_until_dt}\n'
|
||||||
f'last_start_dt: {last_start_dt}\n'
|
f'last_start_dt: {last_start_dt}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
(
|
(
|
||||||
array,
|
array,
|
||||||
|
@ -426,37 +430,58 @@ async def start_backfill(
|
||||||
timeframe,
|
timeframe,
|
||||||
end_dt=last_start_dt,
|
end_dt=last_start_dt,
|
||||||
)
|
)
|
||||||
|
|
||||||
except NoData as _daterr:
|
except NoData as _daterr:
|
||||||
# 3 cases:
|
orig_last_start_dt: datetime = last_start_dt
|
||||||
# - frame in the middle of a legit venue gap
|
gap_report: str = (
|
||||||
# - history actually began at the `last_start_dt`
|
f'EMPTY FRAME for `end_dt: {last_start_dt}`?\n'
|
||||||
# - some other unknown error (ib blocking the
|
f'{mod.name} -> tf@fqme: {timeframe}@{mkt.fqme}\n'
|
||||||
# history bc they don't want you seeing how they
|
f'last_start_dt: {orig_last_start_dt}\n\n'
|
||||||
# cucked all the tinas..)
|
f'bf_until: {backfill_until_dt}\n'
|
||||||
if dur := frame_types.get(timeframe):
|
|
||||||
# decrement by a frame's worth of duration and
|
|
||||||
# retry a few times.
|
|
||||||
last_start_dt.subtract(
|
|
||||||
seconds=dur.total_seconds()
|
|
||||||
)
|
)
|
||||||
log.warning(
|
# EMPTY FRAME signal with 3 (likely) causes:
|
||||||
f'{mod.name} -> EMPTY FRAME for end_dt?\n'
|
#
|
||||||
f'tf@fqme: {timeframe}@{mkt.fqme}\n'
|
# 1. range contains legit gap in venue history
|
||||||
'bf_until <- last_start_dt:\n'
|
# 2. history actually (edge case) **began** at the
|
||||||
f'{backfill_until_dt} <- {last_start_dt}\n'
|
# value `last_start_dt`
|
||||||
f'Decrementing `end_dt` by {dur} and retry..\n'
|
# 3. some other unknown error (ib blocking the
|
||||||
|
# history-query bc they don't want you seeing how
|
||||||
|
# they cucked all the tinas.. like with options
|
||||||
|
# hist)
|
||||||
|
#
|
||||||
|
if def_frame_duration:
|
||||||
|
# decrement by a duration's (frame) worth of time
|
||||||
|
# as maybe indicated by the backend to see if we
|
||||||
|
# can get older data before this possible
|
||||||
|
# "history gap".
|
||||||
|
last_start_dt: datetime = last_start_dt.subtract(
|
||||||
|
seconds=def_frame_duration.total_seconds()
|
||||||
)
|
)
|
||||||
|
gap_report += (
|
||||||
|
f'Decrementing `end_dt` and retrying with,\n'
|
||||||
|
f'def_frame_duration: {def_frame_duration}\n'
|
||||||
|
f'(new) last_start_dt: {last_start_dt}\n'
|
||||||
|
)
|
||||||
|
log.warning(gap_report)
|
||||||
|
# skip writing to shm/tsdb and try the next
|
||||||
|
# duration's worth of prior history.
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
else:
|
||||||
|
# await tractor.pause()
|
||||||
|
raise DataUnavailable(gap_report)
|
||||||
|
|
||||||
# broker says there never was or is no more history to pull
|
# broker says there never was or is no more history to pull
|
||||||
except DataUnavailable:
|
except DataUnavailable as due:
|
||||||
|
message: str = due.args[0]
|
||||||
log.warning(
|
log.warning(
|
||||||
f'NO-MORE-DATA in range?\n'
|
f'Provider {mod.name!r} halted backfill due to,\n\n'
|
||||||
f'`{mod.name}` halted history:\n'
|
|
||||||
f'tf@fqme: {timeframe}@{mkt.fqme}\n'
|
f'{message}\n'
|
||||||
'bf_until <- last_start_dt:\n'
|
|
||||||
f'{backfill_until_dt} <- {last_start_dt}\n'
|
f'fqme: {mkt.fqme}\n'
|
||||||
|
f'timeframe: {timeframe}\n'
|
||||||
|
f'last_start_dt: {last_start_dt}\n'
|
||||||
|
f'bf_until: {backfill_until_dt}\n'
|
||||||
)
|
)
|
||||||
# UGH: what's a better way?
|
# UGH: what's a better way?
|
||||||
# TODO: backends are responsible for being correct on
|
# TODO: backends are responsible for being correct on
|
||||||
|
@ -465,34 +490,54 @@ async def start_backfill(
|
||||||
# to halt the request loop until the condition is
|
# to halt the request loop until the condition is
|
||||||
# resolved or should the backend be entirely in
|
# resolved or should the backend be entirely in
|
||||||
# charge of solving such faults? yes, right?
|
# charge of solving such faults? yes, right?
|
||||||
# if timeframe > 1:
|
|
||||||
# await tractor.pause()
|
|
||||||
return
|
return
|
||||||
|
|
||||||
|
time: np.ndarray = array['time']
|
||||||
assert (
|
assert (
|
||||||
array['time'][0]
|
time[0]
|
||||||
==
|
==
|
||||||
next_start_dt.timestamp()
|
next_start_dt.timestamp()
|
||||||
)
|
)
|
||||||
|
|
||||||
diff = last_start_dt - next_start_dt
|
assert time[-1] == next_end_dt.timestamp()
|
||||||
frame_time_diff_s = diff.seconds
|
|
||||||
|
expected_dur: Interval = last_start_dt - next_start_dt
|
||||||
|
|
||||||
# frame's worth of sample-period-steps, in seconds
|
# frame's worth of sample-period-steps, in seconds
|
||||||
frame_size_s: float = len(array) * timeframe
|
frame_size_s: float = len(array) * timeframe
|
||||||
expected_frame_size_s: float = frame_size_s + timeframe
|
recv_frame_dur: Duration = (
|
||||||
if frame_time_diff_s > expected_frame_size_s:
|
from_timestamp(array[-1]['time'])
|
||||||
|
-
|
||||||
|
from_timestamp(array[0]['time'])
|
||||||
|
)
|
||||||
|
if (
|
||||||
|
(lt_frame := (recv_frame_dur < expected_dur))
|
||||||
|
or
|
||||||
|
(null_frame := (frame_size_s == 0))
|
||||||
|
# ^XXX, should NEVER hit now!
|
||||||
|
):
|
||||||
# XXX: query result includes a start point prior to our
|
# XXX: query result includes a start point prior to our
|
||||||
# expected "frame size" and thus is likely some kind of
|
# expected "frame size" and thus is likely some kind of
|
||||||
# history gap (eg. market closed period, outage, etc.)
|
# history gap (eg. market closed period, outage, etc.)
|
||||||
# so just report it to console for now.
|
# so just report it to console for now.
|
||||||
|
if lt_frame:
|
||||||
|
reason = 'Possible GAP (or first-datum)'
|
||||||
|
else:
|
||||||
|
assert null_frame
|
||||||
|
reason = 'NULL-FRAME'
|
||||||
|
|
||||||
|
missing_dur: Interval = expected_dur.end - recv_frame_dur.end
|
||||||
log.warning(
|
log.warning(
|
||||||
'GAP DETECTED:\n'
|
f'{timeframe}s-series {reason} detected!\n'
|
||||||
f'last_start_dt: {last_start_dt}\n'
|
f'fqme: {mkt.fqme}\n'
|
||||||
f'diff: {diff}\n'
|
f'last_start_dt: {last_start_dt}\n\n'
|
||||||
f'frame_time_diff_s: {frame_time_diff_s}\n'
|
f'recv interval: {recv_frame_dur}\n'
|
||||||
|
f'expected interval: {expected_dur}\n\n'
|
||||||
|
|
||||||
|
f'Missing duration of history of {missing_dur.in_words()!r}\n'
|
||||||
|
f'{missing_dur}\n'
|
||||||
)
|
)
|
||||||
|
# await tractor.pause()
|
||||||
|
|
||||||
to_push = diff_history(
|
to_push = diff_history(
|
||||||
array,
|
array,
|
||||||
|
@ -567,7 +612,8 @@ async def start_backfill(
|
||||||
# long-term storage.
|
# long-term storage.
|
||||||
if (
|
if (
|
||||||
storage is not None
|
storage is not None
|
||||||
and write_tsdb
|
and
|
||||||
|
write_tsdb
|
||||||
):
|
):
|
||||||
log.info(
|
log.info(
|
||||||
f'Writing {ln} frame to storage:\n'
|
f'Writing {ln} frame to storage:\n'
|
||||||
|
@ -688,7 +734,7 @@ async def back_load_from_tsdb(
|
||||||
last_tsdb_dt
|
last_tsdb_dt
|
||||||
and latest_start_dt
|
and latest_start_dt
|
||||||
):
|
):
|
||||||
backfilled_size_s = (
|
backfilled_size_s: Duration = (
|
||||||
latest_start_dt - last_tsdb_dt
|
latest_start_dt - last_tsdb_dt
|
||||||
).seconds
|
).seconds
|
||||||
# if the shm buffer len is not large enough to contain
|
# if the shm buffer len is not large enough to contain
|
||||||
|
@ -911,6 +957,8 @@ async def tsdb_backfill(
|
||||||
f'{pformat(config)}\n'
|
f'{pformat(config)}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# concurrently load the provider's most-recent-frame AND any
|
||||||
|
# pre-existing tsdb history already saved in `piker` storage.
|
||||||
dt_eps: list[DateTime, DateTime] = []
|
dt_eps: list[DateTime, DateTime] = []
|
||||||
async with trio.open_nursery() as tn:
|
async with trio.open_nursery() as tn:
|
||||||
tn.start_soon(
|
tn.start_soon(
|
||||||
|
@ -921,7 +969,6 @@ async def tsdb_backfill(
|
||||||
timeframe,
|
timeframe,
|
||||||
config,
|
config,
|
||||||
)
|
)
|
||||||
|
|
||||||
tsdb_entry: tuple = await load_tsdb_hist(
|
tsdb_entry: tuple = await load_tsdb_hist(
|
||||||
storage,
|
storage,
|
||||||
mkt,
|
mkt,
|
||||||
|
@ -950,6 +997,25 @@ async def tsdb_backfill(
|
||||||
mr_end_dt,
|
mr_end_dt,
|
||||||
) = dt_eps
|
) = dt_eps
|
||||||
|
|
||||||
|
first_frame_dur_s: Duration = (mr_end_dt - mr_start_dt).seconds
|
||||||
|
calced_frame_size: Duration = mk_duration(
|
||||||
|
seconds=first_frame_dur_s,
|
||||||
|
)
|
||||||
|
# NOTE, attempt to use the backend declared default frame
|
||||||
|
# sizing (as allowed by their time-series query APIs) and
|
||||||
|
# if not provided try to construct a default from the
|
||||||
|
# first frame received above.
|
||||||
|
def_frame_durs: dict[
|
||||||
|
int,
|
||||||
|
Duration,
|
||||||
|
]|None = config.get('frame_types', None)
|
||||||
|
if def_frame_durs:
|
||||||
|
def_frame_size: Duration = def_frame_durs[timeframe]
|
||||||
|
assert def_frame_size == calced_frame_size
|
||||||
|
else:
|
||||||
|
# use what we calced from first frame above.
|
||||||
|
def_frame_size = calced_frame_size
|
||||||
|
|
||||||
# NOTE: when there's no offline data, there's 2 cases:
|
# NOTE: when there's no offline data, there's 2 cases:
|
||||||
# - data backend doesn't support timeframe/sample
|
# - data backend doesn't support timeframe/sample
|
||||||
# period (in which case `dt_eps` should be `None` and
|
# period (in which case `dt_eps` should be `None` and
|
||||||
|
@ -980,7 +1046,7 @@ async def tsdb_backfill(
|
||||||
partial(
|
partial(
|
||||||
start_backfill,
|
start_backfill,
|
||||||
get_hist=get_hist,
|
get_hist=get_hist,
|
||||||
frame_types=config.get('frame_types', None),
|
def_frame_duration=def_frame_size,
|
||||||
mod=mod,
|
mod=mod,
|
||||||
mkt=mkt,
|
mkt=mkt,
|
||||||
shm=shm,
|
shm=shm,
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -25,11 +25,11 @@ build-backend = "poetry.core.masonry.api"
|
||||||
ignore = []
|
ignore = []
|
||||||
|
|
||||||
# https://docs.astral.sh/ruff/settings/#lint_per-file-ignores
|
# https://docs.astral.sh/ruff/settings/#lint_per-file-ignores
|
||||||
"piker/ui/qt.py" = [
|
# "piker/ui/qt.py" = [
|
||||||
"E402",
|
# "E402",
|
||||||
'F401', # unused imports (without __all__ or blah as blah)
|
# 'F401', # unused imports (without __all__ or blah as blah)
|
||||||
# "F841", # unused variable rules
|
# # "F841", # unused variable rules
|
||||||
]
|
# ]
|
||||||
# ignore-init-module-imports = false
|
# ignore-init-module-imports = false
|
||||||
|
|
||||||
# ------ - ------
|
# ------ - ------
|
||||||
|
@ -72,12 +72,8 @@ httpx = "^0.27.0"
|
||||||
cryptofeed = "^2.4.0"
|
cryptofeed = "^2.4.0"
|
||||||
pyarrow = "^17.0.0"
|
pyarrow = "^17.0.0"
|
||||||
|
|
||||||
[tool.poetry.dependencies.tractor]
|
tractor = {path = "../tractor", develop = true}
|
||||||
develop = true
|
websockets = "12.0"
|
||||||
git = 'https://pikers.dev/goodboy/tractor.git'
|
|
||||||
branch = 'aio_abandons'
|
|
||||||
# path = "../tractor"
|
|
||||||
|
|
||||||
[tool.poetry.dependencies.asyncvnc]
|
[tool.poetry.dependencies.asyncvnc]
|
||||||
git = 'https://github.com/pikers/asyncvnc.git'
|
git = 'https://github.com/pikers/asyncvnc.git'
|
||||||
branch = 'main'
|
branch = 'main'
|
||||||
|
|
Loading…
Reference in New Issue