diff --git a/ingestor/chalicelib/speed_restrictions.py b/ingestor/chalicelib/speed_restrictions.py index 54a48b7..de5640e 100644 --- a/ingestor/chalicelib/speed_restrictions.py +++ b/ingestor/chalicelib/speed_restrictions.py @@ -2,6 +2,7 @@ import zipfile import csv import boto3 +from pathlib import PurePath from datetime import date, datetime from dataclasses import dataclass from io import BytesIO, TextIOWrapper @@ -87,8 +88,9 @@ def bucket_entries_by_key(entries: Iterator[SpeedRestrictionEntry]) -> Dict[Entr def csv_is_too_old(csv_file_name: str, max_lookback_months: Union[None, int]) -> bool: if not max_lookback_months: return False - csv_date = datetime.strptime(csv_file_name[:7], "%Y-%m").date() - print(csv_date) + file_name_only = PurePath(csv_file_name).name + date_part = file_name_only[:7] + csv_date = datetime.strptime(date_part, "%Y-%m").date() return (date.today() - csv_date).days > (1 + max_lookback_months) * 30