From 03f262f020f0ea855d3ca6f78bad1dad92a2f0dd Mon Sep 17 00:00:00 2001 From: Giacomo Sarrocco Date: Sat, 16 Dec 2023 09:27:03 +0100 Subject: [PATCH] Remove upper bound constraint for orig_dep_date --- server/base/source.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/server/base/source.py b/server/base/source.py index 587fe84..c76cbbc 100644 --- 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)