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

Return accounts in `bidict`

fsp_feeds
Tyler Goodlet 2021-09-08 14:01:54 -04:00
parent 063788499a
commit 0d2cddec9a
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
piker/config.py
View File

@ -22,6 +22,7 @@ from os.path import dirname
import shutil import shutil
from typing import Optional from typing import Optional
from bidict import bidict
import toml import toml
import click import click
@ -104,10 +105,13 @@ def write(
return toml.dump(config, cf) return toml.dump(config, cf)
def load_accounts() -> dict[str, Optional[str]]: def load_accounts(
provider: Optional[str] = None
) -> bidict[str, Optional[str]]:
# our default paper engine entry # our default paper engine entry
accounts: dict[str, Optional[str]] = {'paper': None} accounts = bidict({'paper': None})
conf, path = load() conf, path = load()
section = conf.get('accounts') section = conf.get('accounts')
@ -116,6 +120,10 @@ def load_accounts() -> dict[str, Optional[str]]:
else: else:
for brokername, account_labels in section.items(): for brokername, account_labels in section.items():
if (
provider is None or
provider and brokername == provider
):
for name, value in account_labels.items(): for name, value in account_labels.items():
accounts[f'{brokername}.{name}'] = value accounts[f'{brokername}.{name}'] = value