pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Import adjustments to allow msg codec overriding in `tractor`

tractor_typed_msg_hackin
Tyler Goodlet 2022-07-07 16:57:21 -04:00
parent d31c38ef51
commit 869aa8251a
3 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
piker/clearing/_paper_engine.py
View File

@ -30,8 +30,8 @@ import trio
import tractor import tractor
from dataclasses import dataclass from dataclasses import dataclass
from .. import data
from ..data._source import Symbol from ..data._source import Symbol
from ..data.feed import open_feed
from ..pp import Position from ..pp import Position
from ..data._normalize import iterticks from ..data._normalize import iterticks
from ..data._source import unpack_fqsn from ..data._source import unpack_fqsn
@ -441,14 +441,11 @@ async def trades_dialogue(
) -> None: ) -> None:
tractor.log.get_console_log(loglevel) tractor.log.get_console_log(loglevel)
async with ( async with open_feed(
data.open_feed(
[fqsn], [fqsn],
loglevel=loglevel, loglevel=loglevel,
) as feed, ) as feed:
):
# TODO: load paper positions per broker from .toml config file # TODO: load paper positions per broker from .toml config file
# and pass as symbol to position data mapping: ``dict[str, dict]`` # and pass as symbol to position data mapping: ``dict[str, dict]``
# await ctx.started(all_positions) # await ctx.started(all_positions)

12
piker/data/__init__.py
View File

@ -30,19 +30,19 @@ from ._sharedmem import (
get_shm_token, get_shm_token,
ShmArray, ShmArray,
) )
from .feed import ( # from .feed import (
open_feed, # # open_feed,
_setup_persistent_brokerd, # _setup_persistent_brokerd,
) # )
__all__ = [ __all__ = [
'open_feed', # 'open_feed',
'ShmArray', 'ShmArray',
'iterticks', 'iterticks',
'maybe_open_shm_array', 'maybe_open_shm_array',
'attach_shm_array', 'attach_shm_array',
'open_shm_array', 'open_shm_array',
'get_shm_token', 'get_shm_token',
'_setup_persistent_brokerd', # '_setup_persistent_brokerd',
] ]

7
piker/data/_sharedmem.py
View File

@ -128,12 +128,15 @@ class _Token(Struct, frozen=True):
@classmethod @classmethod
def from_msg(cls, msg: dict) -> _Token: def from_msg(cls, msg: dict) -> _Token:
if isinstance(msg, _Token):
return msg
# TODO: native struct decoding # TODO: native struct decoding
# return _token_dec.decode(msg) # return _token_dec.decode(msg)
if isinstance(msg, _Token):
return msg
# assert 0
msg['dtype_descr'] = tuple(map(tuple, msg['dtype_descr'])) msg['dtype_descr'] = tuple(map(tuple, msg['dtype_descr']))
return _Token(**msg) return _Token(**msg)