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

Make robinhood the default broker backend 

- Add a rate limit cli option
- Allow broker backends to define a max quote query limit
- Add an index ETF list to demonstrate robinhood's real-time prices
kivy_mainline_and_py3.8
Tyler Goodlet 2018-03-21 17:28:40 -04:00
parent 200526da8a
commit 178e091f41
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
piker/cli.py
View File

@ -12,6 +12,7 @@ from .log import get_console_log, colorize_json, get_logger
from .brokers import core from .brokers import core
log = get_logger('cli') log = get_logger('cli')
DEFAULT_BROKER = 'robinhood'
def run(main, loglevel='info'): def run(main, loglevel='info'):
@ -32,7 +33,7 @@ def cli():
@cli.command() @cli.command()
@click.option('--broker', '-b', default='questrade', @click.option('--broker', '-b', default=DEFAULT_BROKER,
help='Broker backend to use') help='Broker backend to use')
@click.option('--loglevel', '-l', default='warning', help='Logging level') @click.option('--loglevel', '-l', default='warning', help='Logging level')
@click.option('--keys', '-k', multiple=True, @click.option('--keys', '-k', multiple=True,
@ -71,7 +72,7 @@ def api(meth, kwargs, loglevel, broker, keys):
@cli.command() @cli.command()
@click.option('--broker', '-b', default='questrade', @click.option('--broker', '-b', default=DEFAULT_BROKER,
help='Broker backend to use') help='Broker backend to use')
@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,
@ -100,15 +101,16 @@ def quote(loglevel, broker, tickers, df_output):
@cli.command() @cli.command()
@click.option('--broker', '-b', default='questrade', @click.option('--broker', '-b', default=DEFAULT_BROKER,
help='Broker backend to use') help='Broker backend to use')
@click.option('--loglevel', '-l', default='warning', help='Logging level') @click.option('--loglevel', '-l', default='warning', help='Logging level')
@click.option('--rate', '-r', default=5, help='Logging level')
@click.argument('name', nargs=1, required=True) @click.argument('name', nargs=1, required=True)
def watch(loglevel, broker, name): def watch(loglevel, broker, rate, name):
"""Spawn a watchlist. """Spawn a watchlist.
""" """
from .ui.watchlist import _async_main from .ui.watchlist import _async_main
get_console_log(loglevel) # activate console logging log = get_console_log(loglevel) # activate console logging
brokermod = import_module('.' + broker, 'piker.brokers') brokermod = import_module('.' + broker, 'piker.brokers')
watchlists = { watchlists = {
@ -122,15 +124,15 @@ def watch(loglevel, broker, name):
'SEED.TO', 'HMJR.TO', 'CMED.TO', 'PAS.VN', 'SEED.TO', 'HMJR.TO', 'CMED.TO', 'PAS.VN',
'CRON', 'CRON',
], ],
'dad': [ 'dad': ['GM', 'TSLA', 'DOL.TO', 'CIM', 'SPY', 'SHOP.TO'],
'GM', 'TSLA', 'DOL.TO', 'CIM', 'SPY', 'pharma': ['ATE.VN'],
'SHOP.TO', 'indexes': ['SPY', 'DAX', 'QQQ', 'DIA'],
],
'pharma': [
'ATE.VN'
],
} }
# broker_conf_path = os.path.join( # broker_conf_path = os.path.join(
# click.get_app_dir('piker'), 'watchlists.json') # click.get_app_dir('piker'), 'watchlists.json')
# from piker.testing import _quote_streamer as brokermod # from piker.testing import _quote_streamer as brokermod
trio.run(_async_main, name, watchlists[name], brokermod) broker_limit = getattr(brokermod, '_rate_limit', float('inf'))
if broker_limit < rate:
rate = broker_limit
log.warn(f"Limiting {brokermod.__name__} query rate to {rate}/sec")
trio.run(_async_main, name, watchlists[name], brokermod, rate)