From bbd5883e5238ff8a0f9540aff420bfc4f52929b2 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Tue, 18 May 2021 08:35:39 -0400
Subject: [PATCH] Add search pause configs to backends

---
 piker/brokers/ib.py     | 35 ++++++++++++++++++++++++-----------
 piker/brokers/kraken.py | 11 ++++++++---
 2 files changed, 32 insertions(+), 14 deletions(-)

diff --git a/piker/brokers/ib.py b/piker/brokers/ib.py
index eea26b62..c56a70a8 100644
--- a/piker/brokers/ib.py
+++ b/piker/brokers/ib.py
@@ -89,7 +89,15 @@ _time_frames = {
     'Y': 'OneYear',
 }
 
-_show_wap_in_history = False
+_show_wap_in_history: bool = False
+
+# optional search config the backend can register for
+# it's symbol search handling (in this case we avoid
+# accepting patterns before the kb has settled more then
+# a quarter second).
+_search_conf = {
+    'pause_period': 6/16,
+}
 
 
 # overrides to sidestep pretty questionable design decisions in
@@ -156,7 +164,7 @@ _adhoc_futes_set = {
     'mes.globex',
 }
 
-    # https://misc.interactivebrokers.com/cstools/contract_info/v3.10/index.php?action=Conid%20Info&wlId=IB&conid=69067924
+# https://misc.interactivebrokers.com/cstools/contract_info/v3.10/index.php?action=Conid%20Info&wlId=IB&conid=69067924
 
 _enters = 0
 
@@ -875,7 +883,16 @@ async def backfill_bars(
         i = 0
         while i < count:
 
-            bars, bars_array, next_dt = await get_bars(sym, end_dt=next_dt)
+            out = await get_bars(sym, end_dt=next_dt)
+
+            if out is None:
+                # could be trying to retreive bars over weekend
+                # TODO: add logic here to handle tradable hours and only grab
+                # valid bars in the range
+                log.error(f"Can't grab bars starting at {next_dt}!?!?")
+                continue
+
+            bars, bars_array, next_dt = out
             shm.push(bars_array, prepend=True)
             i += 1
 
@@ -1255,6 +1272,10 @@ async def open_symbol_search(
                 except trio.WouldBlock:
                     pass
 
+            if not pattern:
+                log.warning(f'empty pattern received, skipping..')
+                continue
+
             log.debug(f'searching for {pattern}')
             # await tractor.breakpoint()
             last = time.time()
@@ -1264,14 +1285,6 @@ async def open_symbol_search(
                 upto=5,
             )
             log.debug(f'got results {results.keys()}')
-            # results = await client.search_stocks(
-            #     pattern=pattern, upto=5)
-
-            # if cs.cancelled_caught:
-            #     print(f'timed out search for {pattern} !?')
-            #     # await tractor.breakpoint()
-            #     await stream.send({})
-            #     continue
 
             log.debug("fuzzy matching")
             matches = fuzzy.extractBests(
diff --git a/piker/brokers/kraken.py b/piker/brokers/kraken.py
index 09a92c90..62fc6115 100644
--- a/piker/brokers/kraken.py
+++ b/piker/brokers/kraken.py
@@ -57,6 +57,11 @@ log = get_logger(__name__)
 _url = 'https://api.kraken.com/0'
 
 
+_search_conf = {
+    'pause_period': 0.0616
+}
+
+
 # Broker specific ohlc schema which includes a vwap field
 _ohlc_dtype = [
     ('index', int),
@@ -231,6 +236,7 @@ class Client:
         if since is None:
             since = arrow.utcnow().floor('minute').shift(
                 minutes=-count).timestamp()
+
         # UTC 2017-07-02 12:53:20 is oldest seconds value
         since = str(max(1499000000, since))
         json = await self._public(
@@ -488,12 +494,12 @@ async def open_autorecon_ws(url):
 
 
 async def backfill_bars(
+
     sym: str,
     shm: ShmArray,  # type: ignore # noqa
-
     count: int = 10,  # NOTE: any more and we'll overrun the underlying buffer
-
     task_status: TaskStatus[trio.CancelScope] = trio.TASK_STATUS_IGNORED,
+
 ) -> None:
     """Fill historical bars into shared mem / storage afap.
     """
@@ -639,7 +645,6 @@ async def stream_quotes(
                 await send_chan.send({topic: quote})
 
 
-
 @tractor.context
 async def open_symbol_search(
     ctx: tractor.Context,