edward/agenda
1
0
Fork 0
Code Issues 16 Pull requests Packages Projects Releases Wiki Activity

Run black to format code

Browse source
This commit is contained in:
Edward Betts 2025-07-16 07:42:20 +02:00
parent 3ad11b070a
commit eb65f720bf
1 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
agenda/busy.py
View file

@ -282,12 +282,9 @@ def get_location_for_date(
if locations:
final_city, final_country = locations[-1]
days_since_trip = (target_date - trip.end).days
# If trip ended in UK, you should be home now
if (
hasattr(final_country, "alpha_2")
and final_country.alpha_2 == "GB"
):
if hasattr(final_country, "alpha_2") and final_country.alpha_2 == "GB":
return ("Bristol", get_country("gb"))
# For short trips to nearby countries or international trips
@ -297,11 +294,26 @@ def get_location_for_date(
and hasattr(final_country, "alpha_2")
and (
# European countries (close by rail/ferry)
final_country.alpha_2 in {"BE", "NL", "FR", "DE", "CH", "AT", "IT", "ES"}
final_country.alpha_2
in {"BE", "NL", "FR", "DE", "CH", "AT", "IT", "ES"}
# Nearby Balkan countries
or final_country.alpha_2 in {"GR", "AL", "XK", "HR", "SI", "MK", "BA", "ME", "RS", "BG", "RO"}
or final_country.alpha_2
in {
"GR",
"AL",
"XK",
"HR",
"SI",
"MK",
"BA",
"ME",
"RS",
"BG",
"RO",
}
# International trips (assume return home after trip ends)
or final_country.alpha_2 in {"US", "CA", "IN", "JP", "CN", "AU", "NZ", "BR", "AR", "ZA"}
or final_country.alpha_2
in {"US", "CA", "IN", "JP", "CN", "AU", "NZ", "BR", "AR", "ZA"}
)
):
return ("Bristol", get_country("gb"))