From 725909a94ccf07537f68a32e335422cf6ffb5877 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Tue, 7 Jun 2022 16:22:46 -0400
Subject: [PATCH] Convert accounts table to `bidict` after config load

---
 piker/brokers/ib/api.py | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/piker/brokers/ib/api.py b/piker/brokers/ib/api.py
index b12e723b..044415fc 100644
--- a/piker/brokers/ib/api.py
+++ b/piker/brokers/ib/api.py
@@ -38,6 +38,7 @@ import time
 from types import SimpleNamespace
 
 
+from bidict import bidict
 import trio
 import tractor
 from tractor import to_asyncio
@@ -261,12 +262,11 @@ class Client:
 
         # NOTE: the ib.client here is "throttled" to 45 rps by default
 
-    async def trades(
-        self,
-        # api_only: bool = False,
-
-    ) -> dict[str, Any]:
+    async def trades(self) -> dict[str, Any]:
+        '''
+        Return list of trades from current session in ``dict``.
 
+        '''
         # orders = await self.ib.reqCompletedOrdersAsync(
         #     apiOnly=api_only
         # )
@@ -811,10 +811,23 @@ _scan_ignore: set[tuple[str, int]] = set()
 
 def get_config() -> dict[str, Any]:
 
-    conf, path = config.load()
-
+    conf, path = config.load('brokers')
     section = conf.get('ib')
 
+    accounts = section.get('accounts')
+    if not accounts:
+        raise ValueError(
+            'brokers.toml -> `ib.accounts` must be defined\n'
+            f'location: {path}'
+        )
+
+    names = list(accounts.keys())
+    accts = section['accounts'] = bidict(accounts)
+    log.info(
+        f'brokers.toml defines {len(accts)} accounts: '
+        f'{pformat(names)}'
+    )
+
     if section is None:
         log.warning(f'No config section found for ib in {path}')
         return {}