pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Use brokermod util functions in cli

kivy_mainline_and_py3.8
Tyler Goodlet 2018-03-27 16:27:30 -04:00
parent 8544c22cdc
commit 19303cf0ad
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
piker/cli.py
View File

@ -2,14 +2,13 @@
Console interface to broker client/daemons. Console interface to broker client/daemons.
""" """
from functools import partial from functools import partial
from importlib import import_module
import click import click
import trio import trio
import pandas as pd import pandas as pd
from .log import get_console_log, colorize_json, get_logger from .log import get_console_log, colorize_json, get_logger
from .brokers import core from .brokers import core, get_brokermod
log = get_logger('cli') log = get_logger('cli')
DEFAULT_BROKER = 'robinhood' DEFAULT_BROKER = 'robinhood'
@ -44,7 +43,7 @@ def api(meth, kwargs, loglevel, broker, keys):
"""client for testing broker API methods with pretty printing of output. """client for testing broker API methods with pretty printing of output.
""" """
log = get_console_log(loglevel) log = get_console_log(loglevel)
brokermod = import_module('.' + broker, 'piker.brokers') brokermod = get_brokermod(broker)
_kwargs = {} _kwargs = {}
for kwarg in kwargs: for kwarg in kwargs:
@ -77,11 +76,11 @@ def api(meth, kwargs, loglevel, broker, keys):
@click.option('--loglevel', '-l', default='warning', help='Logging level') @click.option('--loglevel', '-l', default='warning', help='Logging level')
@click.option('--df-output', '-df', flag_value=True, @click.option('--df-output', '-df', flag_value=True,
help='Ouput in `pandas.DataFrame` format') help='Ouput in `pandas.DataFrame` format')
@click.argument('tickers', nargs=-1) @click.argument('tickers', nargs=-1, required=True)
def quote(loglevel, broker, tickers, df_output): def quote(loglevel, broker, tickers, df_output):
"""client for testing broker API methods with pretty printing of output. """client for testing broker API methods with pretty printing of output.
""" """
brokermod = import_module('.' + broker, 'piker.brokers') brokermod = get_brokermod(broker)
quotes = run(partial(core.quote, brokermod, tickers), loglevel=loglevel) quotes = run(partial(core.quote, brokermod, tickers), loglevel=loglevel)
if not quotes: if not quotes:
log.error(f"No quotes could be found for {tickers}?") log.error(f"No quotes could be found for {tickers}?")
@ -111,7 +110,7 @@ def watch(loglevel, broker, rate, name):
""" """
from .ui.watchlist import _async_main from .ui.watchlist import _async_main
log = get_console_log(loglevel) # activate console logging log = get_console_log(loglevel) # activate console logging
brokermod = import_module('.' + broker, 'piker.brokers') brokermod = get_brokermod(broker)
watchlists = { watchlists = {
'cannabis': [ 'cannabis': [