ogn-python/app/main/routes.py

import datetime

from flask import request, render_template, send_file

from app import db
from app import cache
from app.model import Airport, Country, Device, Logbook, Receiver

from app.main import bp


@cache.cached(key_prefix="countries_in_receivers")
def get_countries_in_receivers():
    query = db.session.query(Country.iso2).filter(Country.gid == Receiver.country_id).order_by(Country.iso2).distinct(Country.iso2)

    return [{"iso2": country[0]} for country in query.all()]


@cache.cached(key_prefix="countries_in_logbook")
def get_countries_in_logbook():
    query = db.session.query(Country.iso2).filter(Country.iso2 == Airport.country_code).filter(Logbook.takeoff_airport_id == Airport.id).order_by(Country.iso2).distinct(Country.iso2)

    return [{"iso2": country[0]} for country in query.all()]


@cache.memoize()
def get_airports_in_country(sel_country):
    query = db.session.query(Airport.id, Airport.name).filter(Airport.country_code == sel_country).filter(Logbook.takeoff_airport_id == Airport.id).order_by(Airport.name).distinct(Airport.name)

    return [{"id": airport[0], "name": airport[1]} for airport in query.all()]


@cache.memoize()
def get_dates_for_airport(sel_airport):
    query = (
        db.session.query(db.func.date(Logbook.reftime), db.func.count(Logbook.id).label("logbook_count"))
        .filter(Airport.id == sel_airport)
        .filter(db.or_(Airport.id == Logbook.takeoff_airport_id, Airport.id == Logbook.landing_airport_id))
        .group_by(db.func.date(Logbook.reftime))
        .order_by(db.func.date(Logbook.reftime).desc())
    )

    return [{"date": date, "logbook_count": logbook_count} for (date, logbook_count) in query.all()]


@bp.route("/")
@bp.route("/index.html")
def index():
    return render_template("base.html")


@bp.route("/devices.html", methods=["GET", "POST"])
def devices():
    devices = db.session.query(Device).order_by(Device.address)
    return render_template("devices.html", devices=devices)


@bp.route("/device_detail.html", methods=["GET", "POST"])
def device_detail():
    device_name = request.args.get("device_name")
    device = db.session.query(Device).filter(Device.name == device_name).one()

    return render_template("device_detail.html", title="Device", device=device)


@bp.route("/receivers.html")
def receivers():
    sel_country = request.args.get("country")

    countries = get_countries_in_receivers()

    # Get receiver selection list
    if sel_country:
        receivers = db.session.query(Receiver).filter(db.and_(Receiver.country_id == Country.gid, Country.iso2 == sel_country)).order_by(Receiver.name)
    else:
        receivers = db.session.query(Receiver).order_by(Receiver.name)

    return render_template("receivers.html", title="Receivers", sel_country=sel_country, countries=countries, receivers=receivers)


@bp.route("/receiver_detail.html")
def receiver_detail():
    receiver_name = request.args.get("receiver_name")

    receiver = db.session.query(Receiver).filter(Receiver.name == receiver_name).one()

    airport = (
        db.session.query(Airport)
        .filter(
            db.and_(
                Receiver.name == receiver_name,
                db.func.st_contains(db.func.st_buffer(Receiver.location_wkt, 0.5), Airport.location_wkt),
                db.func.st_distance_sphere(Airport.location_wkt, Receiver.location_wkt) < 1000,
            )
        )
        .filter(Airport.style.in_((2, 4, 5)))
    )
    return render_template("receiver_detail.html", title="Receiver Detail", receiver=receiver, airport=airport.first())


@bp.route("/airports.html", methods=["GET", "POST"])
def airports():
    sel_country = request.args.get("country")

    countries = get_countries_in_logbook()

    if sel_country:
        airports = get_airports_in_country(sel_country)
    else:
        airports = []

    page = request.args.get("page", 1, type=int)

    return render_template("airports.html", sel_country=sel_country, countries=countries, airports=airports)


@bp.route("/airport_detail.html")
def airport_detail():
    sel_airport = request.args.get("airport")

    airport = db.session.query(Airport).filter(Airport.id == sel_airport)

    devices = db.session.query(Device).join(Logbook).filter(Logbook.takeoff_airport_id == sel_airport).order_by(Device.address)

    return render_template("airport_detail.html", title="Airport Detail", airport=airport.one(), devices=devices)


@bp.route("/logbook.html", methods=["GET", "POST"])
def logbook():
    sel_country = request.args.get("country")
    sel_airport = request.args.get("airport")
    sel_date = request.args.get("date")

    sel_device_id = request.args.get("device_id")

    countries = get_countries_in_logbook()

    if sel_country:
        airports = get_airports_in_country(sel_country)
    else:
        airports = []

    if sel_airport:
        sel_airport = int(sel_airport)
        if sel_airport not in [airport["id"] for airport in airports]:
            sel_airport = None
            sel_date = None
        dates = get_dates_for_airport(sel_airport)
    else:
        dates = []

    if sel_date:
        sel_date = datetime.datetime.strptime(sel_date, "%Y-%m-%d").date()
        if sel_date not in [entry["date"] for entry in dates]:
            sel_date = dates[0]["date"]
    elif len(dates) > 0:
        sel_date = dates[0]["date"]

    # Get Logbook
    filters = []
    if sel_airport:
        filters.append(db.or_(Logbook.takeoff_airport_id == sel_airport, Logbook.landing_airport_id == sel_airport))

    if sel_date:
        filters.append(db.func.date(Logbook.reftime) == sel_date)

    if sel_device_id:
        filters.append(Logbook.device_id == sel_device_id)

    if len(filters) > 0:
        logbook = db.session.query(Logbook).filter(*filters).order_by(Logbook.reftime)
    else:
        logbook = None

    return render_template("logbook.html", title="Logbook", sel_country=sel_country, countries=countries, sel_airport=sel_airport, airports=airports, sel_date=sel_date, dates=dates, logbook=logbook)


@bp.route("/download.html")
def download_flight():
    from io import StringIO

    buffer = StringIO()
    buffer.write("Moin moin\nAlter Verwalter")
    buffer.seek(0)

    return send_file(buffer, as_attachment=True, attachment_filename="wtf.igc", mimetype="text/plain")