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

Support specifying daemon host address

kivy_mainline_and_py3.8
Tyler Goodlet 2018-05-08 15:10:10 -04:00
parent 3d6b14ec3f
commit 5a9c079c10
1 changed files with 20 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
piker/cli.py
View File

@ -23,23 +23,17 @@ _watchlists_data_path = os.path.join(_config_dir, 'watchlists.json')
def run(main, loglevel='info'): def run(main, loglevel='info'):
log = get_console_log(loglevel) get_console_log(loglevel)
# main sandwich
try:
return trio.run(main) return trio.run(main)
except Exception as err:
log.exception(err)
finally:
log.debug("Exiting piker")
@click.command() @click.command()
@click.option('--loglevel', '-l', default='warning', help='Logging level') @click.option('--loglevel', '-l', default='warning', help='Logging level')
def pikerd(loglevel): @click.option('--host', '-h', default='127.0.0.1', help='Host address to bind')
def pikerd(loglevel, host):
"""Spawn the piker daemon. """Spawn the piker daemon.
""" """
run(_daemon_main, loglevel) run(partial(_daemon_main, host), loglevel)
@click.group() @click.group()
@ -120,8 +114,10 @@ def quote(loglevel, broker, tickers, df_output):
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.option('--rate', '-r', default=5, help='Logging level')
@click.option('--dhost', '-dh', default='127.0.0.1',
help='Daemon host address to connect to')
@click.argument('name', nargs=1, required=True) @click.argument('name', nargs=1, required=True)
def watch(loglevel, broker, rate, name): def watch(loglevel, broker, rate, name, dhost):
"""Spawn a real-time watchlist. """Spawn a real-time watchlist.
""" """
from .ui.watchlist import _async_main from .ui.watchlist import _async_main
@ -131,17 +127,20 @@ def watch(loglevel, broker, rate, name):
watchlists = wl.merge_watchlist(watchlist_from_file, wl._builtins) watchlists = wl.merge_watchlist(watchlist_from_file, wl._builtins)
tickers = watchlists[name] tickers = watchlists[name]
async def main(timeout=1): async def launch_client(sleep=0.5, tries=10):
async def subscribe(client): async def subscribe(client):
# initial request for symbols price streams # initial request for symbols price streams
await client.send((brokermod.name, tickers)) await client.send((brokermod.name, tickers))
client = Client(('127.0.0.1', 1616), subscribe) client = Client((dhost, 1616), subscribe)
for _ in range(tries): # try for 5 seconds
try: try:
await client.connect() await client.connect()
break
except OSError as oserr: except OSError as oserr:
await trio.sleep(0.5) await trio.sleep(sleep)
else:
# will raise indicating child proc should be spawned # will raise indicating child proc should be spawned
await client.connect() await client.connect()
@ -155,18 +154,18 @@ def watch(loglevel, broker, rate, name):
await client.aclose() await client.aclose()
try: try:
trio.run(main) trio.run(partial(launch_client, tries=1))
except OSError as oserr: except OSError as oserr:
log.warn("No broker daemon could be found") log.warn("No broker daemon could be found")
log.warn(oserr) log.warn(oserr)
log.warning("Spawning local broker-daemon...") log.warning("Spawning local broker-daemon...")
child = Process( child = Process(
target=run, target=run,
args=(_daemon_main, loglevel), args=(partial(_daemon_main, dhost), loglevel),
daemon=True, daemon=True,
) )
child.start() child.start()
trio.run(main, 5) trio.run(launch_client, 5)
child.join() child.join()