Do not upload stop times with orig_dep_date more than 2 days into the…

… future (#167)
gsarrco · Jan 17, 2024 · 47a5706 · 47a5706
2 parents b24615e + ae98a5b
commit 47a5706
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/base/source.py b/server/base/source.py
@@ -356,6 +356,9 @@ def get_source_stations(self) -> list[list[Station, str]]:
         return list(results.values())
 
     def upload_trip_stop_time_to_postgres(self, stop_time: TripStopTime):
+        if stop_time.orig_dep_date > date.today() + timedelta(days=2):
+            return
+
         stop_id = self.name + '_' + stop_time.station.id if self.name != 'treni' else stop_time.station.id
 
         stmt = insert(StopTime).values(stop_id=stop_id, sched_arr_dt=stop_time.arr_time,