increase lookback days on trip metrics ingestion (#94)

transitmatters · Apr 6, 2024 · 592350a · 592350a
1 parent cf921b1
commit 592350a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/ingestor/app.py b/ingestor/app.py
@@ -124,7 +124,7 @@ def update_time_predictions(event):
 # 4:40am UTC -> 2:40/3:40am ET every day
 @app.schedule(Cron(40, 7, "*", "*", "?", "*"))
 def update_trip_metrics(event):
-    trip_metrics.ingest_trip_metrics_yesterday()
+    trip_metrics.ingest_recent_trip_metrics(lookback_days=7)
 
 
 # Manually triggered lambda for populating daily trip metric tables. Only needs to be ran once.

diff --git a/ingestor/chalicelib/trip_metrics/__init__.py b/ingestor/chalicelib/trip_metrics/__init__.py
@@ -1,3 +1,3 @@
-__all__ = ["ingest_trip_metrics_yesterday", "ingest_trip_metrics"]
+__all__ = ["ingest_recent_trip_metrics", "ingest_trip_metrics"]
 
-from .ingest import ingest_trip_metrics_yesterday, ingest_trip_metrics
+from .ingest import ingest_recent_trip_metrics, ingest_trip_metrics
diff --git a/ingestor/chalicelib/trip_metrics/ingest.py b/ingestor/chalicelib/trip_metrics/ingest.py
@@ -179,7 +179,7 @@ def ingest_trip_metrics(start_date: date, end_date: date):
     dynamo.dynamo_batch_write(row_dicts, "DeliveredTripMetricsExtended")
 
 
-def ingest_trip_metrics_yesterday():
-    today = date.today()
-    yesterday = today - timedelta(days=1)
-    ingest_trip_metrics(yesterday, today)
+def ingest_recent_trip_metrics(lookback_days: int = 1):
+    end = date.today()
+    start = end - timedelta(days=lookback_days)
+    ingest_trip_metrics(start, end)