From 553d0557b6c2d8b606f8a22830e4e7d0aa99525b Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Wed, 26 Oct 2022 10:28:59 -0400
Subject: [PATCH] Raise `DataUnavailable` when a contract's 'earliest time' is
 hit

---
 piker/brokers/ib/feed.py | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/piker/brokers/ib/feed.py b/piker/brokers/ib/feed.py
index 957b6249..3eb4b8df 100644
--- a/piker/brokers/ib/feed.py
+++ b/piker/brokers/ib/feed.py
@@ -39,7 +39,11 @@ import tractor
 import trio
 from trio_typing import TaskStatus
 
-from .._util import SymbolNotFound, NoData
+from .._util import (
+    NoData,
+    DataUnavailable,
+    SymbolNotFound,
+)
 from .api import (
     # _adhoc_futes_set,
     con2fqsn,
@@ -126,6 +130,8 @@ async def open_history_client(
         mean: float = 0
         count: int = 0
 
+        head_dt = await proxy.get_head_time(fqsn=symbol)
+
         async def get_hist(
             timeframe: float,
             end_dt: Optional[datetime] = None,
@@ -153,7 +159,9 @@ async def open_history_client(
                     f'mean: {mean}'
                 )
 
-            if out is None:
+            if (
+                out is None
+            ):
                 # could be trying to retreive bars over weekend
                 log.error(f"Can't grab bars starting at {end_dt}!?!?")
                 raise NoData(
@@ -161,6 +169,9 @@ async def open_history_client(
                     # frame_size=2000,
                 )
 
+            if end_dt and end_dt <= head_dt:
+                raise DataUnavailable(f'First timestamp is {head_dt}')
+
             bars, bars_array, first_dt, last_dt = out
 
             # volume cleaning since there's -ve entries,
@@ -363,8 +374,12 @@ async def get_bars(
 
                         # try to decrement start point and look further back
                         # end_dt = end_dt.subtract(seconds=2000)
-                        end_dt = end_dt.subtract(days=1)
-                        print("SUBTRACTING DAY")
+                        if end_dt is not None:
+                            end_dt = end_dt.subtract(days=1)
+                            print("SUBTRACTING DAY")
+                        else:
+                            end_dt = pendulum.now().subtract(days=1)
+
                         continue
 
                     elif 'API historical data query cancelled' in err.message: