ib.feed: start drafting out `get_mkt_info()` endpoint

master
Tyler Goodlet 2023-05-15 15:35:57 -04:00
parent 1263835034
commit f20e2d6ee2
1 changed files with 55 additions and 42 deletions

View File

@ -19,7 +19,10 @@ Data feed endpoints pre-wrapped and ready for use with ``tractor``/``trio``.
"""
from __future__ import annotations
import asyncio
from contextlib import asynccontextmanager as acm
from contextlib import (
asynccontextmanager as acm,
nullcontext,
)
from decimal import Decimal
from dataclasses import asdict
from datetime import datetime
@ -59,6 +62,9 @@ from .api import (
Contract,
)
from ._util import data_reset_hack
from piker._cacheables import (
async_lifo_cache,
)
# https://interactivebrokers.github.io/tws-api/tick_types.html
@ -733,22 +739,36 @@ def normalize(
return data
# TODO!
# async def get_mkt_info(
# fqme: str,
@async_lifo_cache()
async def get_mkt_info(
fqme: str,
# _cache: dict[str, MktPair] = {}
proxy: MethodProxy | None = None,
# ) -> tuple[MktPair, Pair]:
) -> tuple[MktPair, Pair]:
# both = _cache.get(fqme)
# if both:
# return both
# we don't need to split off any fqme broker part?
# bs_fqme, _, broker = fqme.partition('.')
# proxy: MethodProxy
# async with open_data_client() as proxy:
proxy: MethodProxy
if proxy is not None:
client_ctx = nullcontext(proxy)
else:
client_ctx = open_data_client
async with client_ctx as proxy:
try:
(
con, # Contract
details, # ContractDetails
) = await proxy.get_sym_details(symbol=fqme)
except ConnectionError:
log.exception(f'Proxy is ded {proxy._aio_ns}')
raise
# pair: Pair = await client.exch_info(fqme.upper())
# mkt = MktPair(
# dst=Asset(
# name=pair.baseAsset,
@ -765,9 +785,9 @@ def normalize(
# bs_mktid=pair.symbol,
# broker='binance',
# )
# both = mkt, pair
# _cache[fqme] = both
# return both
return con, details
async def stream_quotes(
@ -794,18 +814,11 @@ async def stream_quotes(
proxy: MethodProxy
async with open_data_client() as proxy:
try:
(
con, # Contract
first_ticker, # Ticker
details, # ContractDetails
) = await proxy.get_sym_details(symbol=sym)
except ConnectionError:
log.exception(f'Proxy is ded {proxy._aio_ns}')
raise
con, details = await get_mkt_info(sym, proxy=proxy)
first_ticker = await proxy.get_quote(contract=con)
first_quote = normalize(first_ticker)
# print(f'first quote: {first_quote}')
log.runtime(f'FIRST QUOTE: {first_quote}')
def mk_init_msgs() -> dict[str, dict]:
'''
@ -879,7 +892,7 @@ async def stream_quotes(
# TODO: we should instead spawn a task that waits on a feed to start
# and let it wait indefinitely..instead of this hard coded stuff.
with trio.move_on_after(1):
contract, first_ticker, details = await proxy.get_quote(symbol=sym)
first_ticker = await proxy.get_quote(contract=con)
# it might be outside regular trading hours so see if we can at
# least grab history.