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

Port watchlist app to tractor api

kivy_mainline_and_py3.8
Tyler Goodlet 2018-06-27 11:50:02 -04:00
parent f32ecb3d65
commit e22f17bfe9
1 changed files with 24 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
piker/ui/watchlist.py
View File

@ -6,7 +6,7 @@ Launch with ``piker watch <watchlist name>``.
(Currently there's a bunch of questrade specific stuff in here) (Currently there's a bunch of questrade specific stuff in here)
""" """
from itertools import chain from itertools import chain
from types import ModuleType from types import ModuleType, AsyncGeneratorType
import trio import trio
from kivy.uix.boxlayout import BoxLayout from kivy.uix.boxlayout import BoxLayout
@ -319,7 +319,7 @@ async def update_quotes(
nursery: 'Nursery', nursery: 'Nursery',
brokermod: ModuleType, brokermod: ModuleType,
widgets: dict, widgets: dict,
client: 'Client', agen: AsyncGeneratorType,
symbol_data: dict, symbol_data: dict,
first_quotes: dict first_quotes: dict
): ):
@ -359,7 +359,7 @@ async def update_quotes(
grid.render_rows(cache) grid.render_rows(cache)
# core cell update loop # core cell update loop
async for quotes in client.aiter_recv(): # new quotes data only async for quotes in agen: # new quotes data only
for symbol, quote in quotes.items(): for symbol, quote in quotes.items():
record, displayable = brokermod.format_quote( record, displayable = brokermod.format_quote(
quote, symbol_data=symbol_data) quote, symbol_data=symbol_data)
@ -375,28 +375,26 @@ async def update_quotes(
nursery.cancel_scope.cancel() nursery.cancel_scope.cancel()
async def run_kivy(root, nursery): async def _async_main(name, portal, tickers, brokermod, rate):
'''Trio-kivy entry point.
'''
await async_runTouchApp(root) # run kivy
nursery.cancel_scope.cancel() # cancel all other tasks that may be running
async def _async_main(name, client, tickers, brokermod, rate):
'''Launch kivy app + all other related tasks. '''Launch kivy app + all other related tasks.
This is started with cli command `piker watch`. This is started with cli command `piker watch`.
''' '''
# subscribe for tickers # subscribe for tickers (this performs a possible filtering
await client.send((brokermod.name, tickers)) # where invalid symbols are discarded)
# get initial symbol data (long term data including last days close price) sd = await portal.run(
# TODO: need something better this this toy protocol "piker.brokers.core", 'symbol_data',
sd = await client.recv() broker=brokermod.name, tickers=tickers)
# an async generator instance
agen = await portal.run(
"piker.brokers.core", 'start_quote_stream',
broker=brokermod.name, tickers=tickers)
async with trio.open_nursery() as nursery: async with trio.open_nursery() as nursery:
# get first quotes response # get first quotes response
log.debug("Waiting on first quote...") log.debug("Waiting on first quote...")
quotes = await client.recv() quotes = await agen.__anext__()
first_quotes = [ first_quotes = [
brokermod.format_quote(quote, symbol_data=sd)[0] brokermod.format_quote(quote, symbol_data=sd)[0]
for quote in quotes.values()] for quote in quotes.values()]
@ -455,6 +453,13 @@ async def _async_main(name, client, tickers, brokermod, rate):
'header': header, 'header': header,
'pager': pager, 'pager': pager,
} }
nursery.start_soon(run_kivy, widgets['root'], nursery) # nursery.start_soon(run_kivy, widgets['root'], nursery)
nursery.start_soon( nursery.start_soon(
update_quotes, nursery, brokermod, widgets, client, sd, quotes) update_quotes, nursery, brokermod, widgets, agen, sd, quotes)
# Trio-kivy entry point.
await async_runTouchApp(widgets['root']) # run kivy
await agen.aclose() # cancel aysnc gen call
await portal.run(
"piker.brokers.core", 'modify_quote_stream',
broker=brokermod.name, tickers=[])