Optionally load `MktPair` in `Flume`s
parent
2cc80d53ca
commit
452cd7db8a
|
@ -22,7 +22,7 @@ real-time data processing data-structures.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from decimal import Decimal
|
# from decimal import Decimal
|
||||||
from typing import (
|
from typing import (
|
||||||
TYPE_CHECKING,
|
TYPE_CHECKING,
|
||||||
)
|
)
|
||||||
|
@ -193,19 +193,30 @@ class Flume(Struct):
|
||||||
return msg
|
return msg
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_msg(cls, msg: dict) -> dict:
|
def from_msg(
|
||||||
|
cls,
|
||||||
|
msg: dict,
|
||||||
|
|
||||||
|
) -> dict:
|
||||||
|
'''
|
||||||
|
Load from an IPC msg presumably in either `dict` or
|
||||||
|
`msgspec.Struct` form.
|
||||||
|
|
||||||
|
'''
|
||||||
|
sym_msg = msg.pop('symbol')
|
||||||
|
|
||||||
|
if 'dst' in sym_msg:
|
||||||
|
mkt = MktPair.from_msg(sym_msg)
|
||||||
|
|
||||||
|
else:
|
||||||
# XXX NOTE: ``msgspec`` can encode `Decimal`
|
# XXX NOTE: ``msgspec`` can encode `Decimal`
|
||||||
# but it doesn't decide to it by default since
|
# but it doesn't decide to it by default since
|
||||||
# we aren't spec-cing these msgs as structs...
|
# we aren't spec-cing these msgs as structs, SO
|
||||||
symbol = Symbol(**msg.pop('symbol'))
|
# we have to ensure we do a struct type case (which `.copy()`
|
||||||
symbol.tick_size = Decimal(symbol.tick_size)
|
# does) to ensure we get the right type!
|
||||||
symbol.lot_tick_size = Decimal(symbol.lot_tick_size)
|
mkt = Symbol(**sym_msg).copy()
|
||||||
|
|
||||||
return cls(
|
return cls(symbol=mkt, **msg)
|
||||||
symbol=symbol,
|
|
||||||
**msg,
|
|
||||||
)
|
|
||||||
|
|
||||||
def get_index(
|
def get_index(
|
||||||
self,
|
self,
|
||||||
|
|
Loading…
Reference in New Issue