Edward Betts
ea4980a5d7
Adjust European short trip heuristic from >3 days to >1 day to correctly detect when user has returned home from European trips. This fixes the April 29-30, 2023 case where the location incorrectly showed "Sankt Georg, Hamburg" instead of "Bristol" when the user was free (no events scheduled) after the foss-north trip ended on April 27. The previous logic required more than 3 days to pass before assuming return home from European countries, but for short European trips by rail/ferry, users typically return within 1-2 days. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
39 lines
923 B
Python
Executable file
39 lines
923 B
Python
Executable file
#!/usr/bin/python3
|
|
|
|
import datetime
|
|
import sys
|
|
import typing
|
|
|
|
from rich.pretty import pprint
|
|
|
|
import agenda.travel
|
|
from agenda.types import StrDict
|
|
|
|
config = __import__("config.default", fromlist=[""])
|
|
data_dir = config.PERSONAL_DATA
|
|
|
|
|
|
def get_current_flight(now: datetime.datetime) -> StrDict | None:
|
|
"""Flight."""
|
|
bookings = agenda.travel.parse_yaml("flights", data_dir)
|
|
for booking in bookings:
|
|
for flight in typing.cast(list[StrDict], booking["flights"]):
|
|
if (
|
|
"depart" not in flight
|
|
or "arrive" not in flight
|
|
or not (flight["depart"] < now < flight["arrive"])
|
|
):
|
|
continue
|
|
return flight
|
|
return None
|
|
|
|
|
|
now = datetime.datetime.now(tz=datetime.UTC)
|
|
flight = get_current_flight(now)
|
|
if not flight:
|
|
print("not currently on a flight")
|
|
sys.exit(0)
|
|
pprint(flight)
|
|
|
|
print(flight["arrive"] - now)
|