#!/usr/bin/python3 """Web page to show upcoming events.""" import inspect import operator import os.path import sys import traceback from datetime import date, datetime import flask import werkzeug import werkzeug.debug.tbtools import yaml import agenda.data import agenda.error_mail import agenda.thespacedevs from agenda import format_list_with_ampersand, travel from agenda.types import StrDict, Trip app = flask.Flask(__name__) app.debug = False app.config.from_object("config.default") agenda.error_mail.setup_error_mail(app) @app.errorhandler(werkzeug.exceptions.InternalServerError) def exception_handler(e: werkzeug.exceptions.InternalServerError) -> tuple[str, int]: """Handle exception.""" exec_type, exc_value, current_traceback = sys.exc_info() assert exc_value tb = werkzeug.debug.tbtools.DebugTraceback(exc_value) summary = tb.render_traceback_html(include_title=False) exc_lines = "".join(tb._te.format_exception_only()) last_frame = list(traceback.walk_tb(current_traceback))[-1][0] last_frame_args = inspect.getargs(last_frame.f_code) return ( flask.render_template( "show_error.html", plaintext=tb.render_traceback_text(), exception=exc_lines, exception_type=tb._te.exc_type.__name__, summary=summary, last_frame=last_frame, last_frame_args=last_frame_args, ), 500, ) @app.route("/") async def index() -> str: """Index page.""" now = datetime.now() data = await agenda.data.get_data(now, app.config) return flask.render_template("index.html", today=now.date(), **data) @app.route("/launches") async def launch_list() -> str: """Web page showing List of space launches.""" now = datetime.now() data_dir = app.config["DATA_DIR"] rocket_dir = os.path.join(data_dir, "thespacedevs") rockets = await agenda.thespacedevs.get_launches(rocket_dir, limit=100) return flask.render_template("launches.html", rockets=rockets, now=now) @app.route("/gaps") async def gaps_page() -> str: """List of available gaps.""" now = datetime.now() data = await agenda.data.get_data(now, app.config) return flask.render_template("gaps.html", today=now.date(), gaps=data["gaps"]) @app.route("/travel") def travel_list() -> str: """Page showing a list of upcoming travel.""" data_dir = app.config["PERSONAL_DATA"] flights = travel.parse_yaml("flights", data_dir) trains = travel.parse_yaml("trains", data_dir) return flask.render_template("travel.html", flights=flights, trains=trains) def as_date(d: date | datetime) -> date: """Date of event.""" return d.date() if isinstance(d, datetime) else d @app.route("/conference") def conference_list() -> str: """Page showing a list of conferences.""" data_dir = app.config["PERSONAL_DATA"] filepath = os.path.join(data_dir, "conferences.yaml") item_list = yaml.safe_load(open(filepath)) today = date.today() for conf in item_list: conf["start_date"] = as_date(conf["start"]) conf["end_date"] = as_date(conf["end"]) item_list.sort(key=operator.itemgetter("start_date")) current = [ conf for conf in item_list if conf["start_date"] <= today and conf["end_date"] >= today ] past = [conf for conf in item_list if conf["end_date"] < today] future = [conf for conf in item_list if conf["start_date"] > today] return flask.render_template( "conference_list.html", current=current, past=past, future=future, today=today, get_country=agenda.get_country, ) @app.route("/accommodation") def accommodation_list() -> str: """Page showing a list of past, present and future accommodation.""" data_dir = app.config["PERSONAL_DATA"] items = travel.parse_yaml("accommodation", data_dir) stays_in_2024 = [item for item in items if item["from"].year == 2024] total_nights_2024 = sum( (stay["to"].date() - stay["from"].date()).days for stay in stays_in_2024 ) nights_abroad_2024 = sum( (stay["to"].date() - stay["from"].date()).days for stay in stays_in_2024 if stay["country"] != "gb" ) return flask.render_template( "accommodation.html", items=items, total_nights_2024=total_nights_2024, nights_abroad_2024=nights_abroad_2024, get_country=agenda.get_country, ) def load_travel(travel_type: str) -> list[StrDict]: """Read flight and train journeys.""" data_dir = app.config["PERSONAL_DATA"] items = travel.parse_yaml(travel_type + "s", data_dir) for item in items: item["type"] = travel_type return items def load_trains() -> list[StrDict]: """Load trains.""" data_dir = app.config["PERSONAL_DATA"] trains = load_travel("train") stations = travel.parse_yaml("stations", data_dir) by_name = {station["name"]: station for station in stations} for train in trains: assert train["from"] in by_name assert train["to"] in by_name train["from_station"] = by_name[train["from"]] train["to_station"] = by_name[train["to"]] for leg in train["legs"]: assert leg["from"] in by_name assert leg["to"] in by_name leg["from_station"] = by_name[leg["from"]] leg["to_station"] = by_name[leg["to"]] return trains def load_flights() -> list[StrDict]: """Load flights.""" data_dir = app.config["PERSONAL_DATA"] flights = load_travel("flight") airports = travel.parse_yaml("airports", data_dir) for flight in flights: if flight["from"] in airports: flight["from_airport"] = airports[flight["from"]] if flight["to"] in airports: flight["to_airport"] = airports[flight["to"]] return flights def build_trip_list() -> list[Trip]: """Generate list of trips.""" trips: dict[date, Trip] = {} data_dir = app.config["PERSONAL_DATA"] travel_items = sorted( load_flights() + load_trains(), key=operator.itemgetter("depart") ) data = { "travel": travel_items, "accommodation": travel.parse_yaml("accommodation", data_dir), "conferences": travel.parse_yaml("conferences", data_dir), "events": travel.parse_yaml("events", data_dir), } for key, item_list in data.items(): assert isinstance(item_list, list) for item in item_list: if not (start := item.get("trip")): continue if start not in trips: trips[start] = Trip(start=start) getattr(trips[start], key).append(item) return [trip for _, trip in sorted(trips.items())] @app.route("/trip") def trip_list() -> str: """Page showing a list of trips.""" trip_list = build_trip_list() today = date.today() current = [ item for item in trip_list if item.start <= today and (item.end or item.start) >= today ] past = [item for item in trip_list if (item.end or item.start) < today] future = [item for item in trip_list if item.start > today] return flask.render_template( "trips.html", current=current, past=past, future=future, today=today, get_country=agenda.get_country, format_list_with_ampersand=format_list_with_ampersand, ) def collect_trip_coordinates(trip: Trip) -> list[StrDict]: """Extract and deduplicate airport and station coordinates from trip.""" stations = {} station_list = [] airports = {} for t in trip.travel: if t["type"] == "train": station_list += [t["from_station"], t["to_station"]] for leg in t["legs"]: station_list.append(leg["from_station"]) station_list.append(leg["to_station"]) else: assert t["type"] == "flight" for field in "from_airport", "to_airport": if field in t: airports[t[field]["iata"]] = t[field] for s in station_list: if s["uic"] in stations: continue stations[s["uic"]] = s return [ { "name": s["name"], "type": "station", "latitude": s["latitude"], "longitude": s["longitude"], } for s in stations.values() ] + [ { "name": s["name"], "type": "airport", "latitude": s["latitude"], "longitude": s["longitude"], } for s in airports.values() ] def latlon_tuple(stop: StrDict) -> tuple[float, float]: """Given a transport stop return the lat/lon as a tuple.""" return (stop["latitude"], stop["longitude"]) def read_geojson(filename: str) -> str: data_dir = app.config["PERSONAL_DATA"] return open(os.path.join(data_dir, "train_routes", filename + ".geojson")).read() def get_trip_routes(trip: Trip) -> list[StrDict]: routes = [] seen_geojson = set() for t in trip.travel: if t["type"] == "flight": if "from_airport" not in t or "to_airport" not in t: continue fly_from, fly_to = t["from_airport"], t["to_airport"] routes.append( { "type": "flight", "from": latlon_tuple(fly_from), "to": latlon_tuple(fly_to), } ) else: assert t["type"] == "train" for leg in t["legs"]: train_from, train_to = leg["from_station"], leg["to_station"] geojson_filename = train_from.get("routes", {}).get(train_to["uic"]) print(train_from["name"], train_to["name"], geojson_filename) if not geojson_filename: routes.append( { "type": "train", "from": latlon_tuple(train_from), "to": latlon_tuple(train_to), } ) continue if geojson_filename in seen_geojson: continue seen_geojson.add(geojson_filename) routes.append( { "type": "train", "geojson": read_geojson(geojson_filename), } ) return routes @app.route("/trip/") def trip_page(start: str) -> str: """Individual trip page.""" trip_list = build_trip_list() today = date.today() trip = next((trip for trip in trip_list if trip.start.isoformat() == start), None) if not trip: flask.abort(404) coordinates = collect_trip_coordinates(trip) routes = get_trip_routes(trip) return flask.render_template( "trip_page.html", trip=trip, today=today, coordinates=coordinates, routes=routes, get_country=agenda.get_country, format_list_with_ampersand=format_list_with_ampersand, ) if __name__ == "__main__": app.run(host="0.0.0.0")