From 62259880fd0d9ac40c455347666924b0714ea7b0 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Wed, 5 Apr 2023 19:34:29 -0400 Subject: [PATCH] paper: on no input fqme, load all mktinfos from pos table --- piker/clearing/_paper_engine.py | 37 +++++++++++++++++++-------------- 1 file changed, 21 insertions(+), 16 deletions(-) diff --git a/piker/clearing/_paper_engine.py b/piker/clearing/_paper_engine.py index 086263a2..f2fcbbd3 100644 --- a/piker/clearing/_paper_engine.py +++ b/piker/clearing/_paper_engine.py @@ -557,24 +557,29 @@ async def trades_dialogue( # existing brokerd daemon? # - alternatively we can possibly expect and use # a `.broker.norm_trade_records()` ep? - mkt: MktPair | None = None - brokermod = get_brokermod(broker) - gmi = getattr(brokermod, 'get_mkt_info', None) - if gmi: - mkt, pair = await brokermod.get_mkt_info( - fqme.rstrip(f'.{broker}'), + fqmes: list[str] = [fqme] + if fqme is None: + fqmes = list(ppt.pps) + + for fqme in fqmes: + mkt: MktPair | None = None + brokermod = get_brokermod(broker) + gmi = getattr(brokermod, 'get_mkt_info', None) + if gmi: + mkt, pair = await brokermod.get_mkt_info( + fqme.rstrip(f'.{broker}'), + ) + + # update pos table from ledger history + ppt.update_from_trans( + ledger.to_trans(), + + # NOTE: here we pass in any `MktPair` provided by the + # backend broker instead of assuming the pps.toml contains + # the correct contents! + force_mkt=mkt ) - # update pos table from ledger history - ppt.update_from_trans( - ledger.to_trans(), - - # NOTE: here we pass in any `MktPair` provided by the - # backend broker instead of assuming the pps.toml contains - # the correct contents! - force_mkt=mkt - ) - pp_msgs: list[BrokerdPosition] = [] pos: Position token: str # f'{symbol}.{self.broker}'