diff --git a/agenda/busy.py b/agenda/busy.py index 72793c3..87ccf1d 100644 --- a/agenda/busy.py +++ b/agenda/busy.py @@ -84,9 +84,9 @@ def get_busy_events( def get_location_for_date( target_date: date, trips: list[Trip], - bookings: list[dict], - accommodations: list[dict], - airports: dict, + bookings: list[StrDict], + accommodations: list[StrDict], + airports: StrDict, ) -> tuple[str | None, pycountry.db.Country | None]: """Get location (city, country) for a specific date using travel history.""" # UK airports that indicate being home @@ -211,17 +211,17 @@ def get_location_for_date( if len(locations) == 1: city, country = locations[0] return (city, country) - + # Multiple locations: use progression through the trip trip_duration = (trip.end - trip.start).days + 1 days_into_trip = (target_date - trip.start).days - + # Simple progression: first half at first location, second half at last location if days_into_trip <= trip_duration // 2: city, country = locations[0] else: city, country = locations[-1] - + return (city, country) # Find most recent flight or accommodation before this date