Simplify return home heuristics now that we have comprehensive travel data
Remove complex country-based return home assumptions since we now track actual train and ferry travel back to the UK. The system can rely on real travel data rather than heuristics. - Remove NEARBY_BALKAN_COUNTRIES constant - Remove geographic assumptions about returning home after trips - Keep only UK trip ending logic (if trip ends in UK, you're home) With complete train/ferry tracking, actual return travel is captured rather than inferred. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
parent
e370049bcb
commit
f396d8a62f
|
@ -11,20 +11,6 @@ from . import events_yaml, get_country, travel
|
||||||
from .event import Event
|
from .event import Event
|
||||||
from .types import StrDict, Trip
|
from .types import StrDict, Trip
|
||||||
|
|
||||||
NEARBY_BALKAN_COUNTRIES = {
|
|
||||||
"GR",
|
|
||||||
"AL",
|
|
||||||
"XK",
|
|
||||||
"HR",
|
|
||||||
"SI",
|
|
||||||
"MK",
|
|
||||||
"BA",
|
|
||||||
"ME",
|
|
||||||
"RS",
|
|
||||||
"BG",
|
|
||||||
"RO",
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def busy_event(e: Event) -> bool:
|
def busy_event(e: Event) -> bool:
|
||||||
"""Busy."""
|
"""Busy."""
|
||||||
|
@ -522,19 +508,6 @@ def _check_return_home_heuristic(
|
||||||
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 (None, get_country("gb"))
|
return (None, get_country("gb"))
|
||||||
|
|
||||||
# For short trips to nearby countries or international trips
|
|
||||||
# (ended >=1 day ago), assume returned home if no subsequent travel data
|
|
||||||
if days_since_trip >= 1 and (
|
|
||||||
# European countries (close by rail/ferry)
|
|
||||||
final_alpha_2 in {"BE", "NL", "FR", "DE", "CH", "AT", "IT", "ES"}
|
|
||||||
# Nearby Balkan countries
|
|
||||||
or final_alpha_2 in NEARBY_BALKAN_COUNTRIES
|
|
||||||
# International trips (assume return home after trip ends)
|
|
||||||
or final_alpha_2
|
|
||||||
in {"US", "CA", "IN", "JP", "CN", "AU", "NZ", "BR", "AR", "ZA"}
|
|
||||||
):
|
|
||||||
return (None, get_country("gb"))
|
|
||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue