pikers
/
piker
3
0
Fork 0
Code Issues 2 Pull Requests 5 Packages Projects Releases Wiki Activity

`ib`: again, only *update* ledger records from API

rekt_pps
Tyler Goodlet 2023-03-27 16:49:17 -04:00
parent 2cb59fe450
commit a336def65f
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
piker/brokers/ib/broker.py
View File

@ -612,36 +612,39 @@ async def trades_dialogue(
# update trades ledgers for all accounts from connected # update trades ledgers for all accounts from connected
# api clients which report trades for **this session**. # api clients which report trades for **this session**.
trades = await proxy.trades() api_trades = await proxy.trades()
if trades: if api_trades:
trans_by_acct: dict[str, Transaction] trans_by_acct: dict[str, Transaction]
api_to_ledger_entries: dict[str, dict] api_to_ledger_entries: dict[str, dict]
( (
trans_by_acct, trans_by_acct,
api_to_ledger_entries, api_to_ledger_entries,
) = await update_ledger_from_api_trades( ) = await update_ledger_from_api_trades(
trades, api_trades,
proxy, proxy,
accounts_def_inv, accounts_def_inv,
) )
# if new trades are detected from the API, prepare # if new api_trades are detected from the API, prepare
# them for the ledger file and update the pptable. # them for the ledger file and update the pptable.
if api_to_ledger_entries: if api_to_ledger_entries:
trade_entries = api_to_ledger_entries.get(acctid) trade_entries = api_to_ledger_entries.get(acctid)
await tractor.breakpoint()
if trade_entries: if trade_entries:
# write ledger with all new trades # write ledger with all new api_trades
# **AFTER** we've updated the # **AFTER** we've updated the `pps.toml`
# `pps.toml` from the original # from the original ledger state! (i.e. this
# ledger state! (i.e. this is # is currently done on exit)
# currently done on exit) for tid, entry in trade_entries.items():
ledger.update(trade_entries) ledger.setdefault(tid, {}).update(entry)
trans = trans_by_acct.get(acctid) trans = trans_by_acct.get(acctid)
if trans: if trans:
table.update_from_trans(trans) table.update_from_trans(trans)
# update position table with latest ledger from all
# gathered transactions: ledger file + api records.
trans = norm_trade_records(ledger) trans = norm_trade_records(ledger)
table.update_from_trans(trans) table.update_from_trans(trans)
@ -653,7 +656,8 @@ async def trades_dialogue(
# -> collect all ib-pp reported positions so that we can be # -> collect all ib-pp reported positions so that we can be
# sure know which positions to update from the ledger if # sure know which positions to update from the ledger if
# any are missing from the ``pps.toml`` # any are missing from the ``pps.toml``
pos: IbPosition # named tuple actually
pos: IbPosition # named tuple subtype
for pos in client.positions(): for pos in client.positions():
# NOTE XXX: we skip options for now since we don't # NOTE XXX: we skip options for now since we don't