Move code around

agenda/holidays.py

@@ -3,11 +3,30 @@
 import collections
 from datetime import date, timedelta
 
+import flask
+
 import agenda.uk_holiday
 import holidays
 
 from .event import Event
-from .types import Holiday
+from .types import Holiday, Trip
+
+
+def get_trip_holidays(trip: Trip) -> list[Holiday]:
+    """Get holidays happening during trip."""
+    if not trip.end:
+        return []
+    countries = {c.alpha_2 for c in trip.countries}
+    return sorted(
+        (
+            hol
+            for hol in get_all(
+                trip.start, trip.end, flask.current_app.config["DATA_DIR"]
+            )
+            if hol.country.upper() in countries
+        ),
+        key=lambda item: (item.date, item.country),
+    )
 
 
 def us_holidays(start_date: date, end_date: date) -> list[Holiday]: