Remove upper bound constraint for orig_dep_date

gsarrco · Dec 16, 2023 · 03f262f · 03f262f
1 parent 8876547
commit 03f262f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/server/base/source.py b/server/base/source.py
@@ -106,10 +106,13 @@ def get_stop_times(self, stops_ids, line, start_dt: datetime, offset: int | tupl
         else:
             stmt = select(StopTime)
 
-        day = start_dt.date()
-        day_minus_one = day - timedelta(days=1)
+        start_day_minus_one = start_dt.date() - timedelta(days=1)
+        stmt = stmt.filter(StopTime.orig_dep_date >= start_day_minus_one)
+
+        if end_dt:
+            stmt = stmt.filter(StopTime.orig_dep_date <= end_dt.date())
 
-        stmt = stmt.filter(StopTime.orig_dep_date.between(day_minus_one, day), StopTime.stop_id.in_(stops_ids))
+        stmt = stmt.filter(StopTime.stop_id.in_(stops_ids))
 
         if direction == 1:
             stmt = stmt.filter(StopTime.sched_dep_dt >= start_dt)