Airlines

"""Airline lookup response models for /airlines/search."""from __future__ import annotationsfrom dataclasses import dataclass@dataclassclass AirlineLookupRequest:    """Query input for GET /airlines/search - provide one code."""    icao: str | None = None    iata: str | None = None@dataclassclass AirlineRecord:    id: int    name: str    alias: str | None    iata: str | None    icao: str | None    callsign: str | None    country: str | None    active: str    logo: str | None@dataclassclass AirlineSummary:    """Shape returned by airline_summary() helper."""    icao: str | None    iata: str | None    name: str    callsign: str | None    country: str | None    is_active: bool    logo: str | None

import osimport timefrom dataclasses import dataclassimport requestsHEADERS = {    "x-api-key": os.getenv("SKYLINK_API_KEY", "YOUR_API_KEY")}BASE = "https://data.skylinkapi.com/v3"CACHE: dict[str, tuple[dict, float]] = {}AIRLINE_TTL = 86_400  # 24 h@dataclassclass AirlineSummary:    icao: str | None    iata: str | None    name: str    callsign: str | None    country: str | None    is_active: bool    logo: str | Nonedef fetch_airline(*, icao: str | None = None, iata: str | None = None) -> dict | None:    if bool(icao) == bool(iata):        raise ValueError("Provide exactly one of icao or iata")    code = (icao or iata or "").upper()    key = f"airline:{icao or ''}:{iata or ''}"    cached = CACHE.get(key)    if cached and time.time() < cached[1]:        return cached[0]    params = {"icao": code} if icao else {"iata": code}    try:        r = requests.get(f"{BASE}/airlines/search", headers=HEADERS, params=params, timeout=(10, 15))    except requests.Timeout as exc:        raise RuntimeError(f"SkyLink timeout fetching airline {code}") from exc    if r.status_code == 404:        return None    r.raise_for_status()    items = r.json()    if not items:        return None    airline = items[0]    CACHE[key] = (airline, time.time() + AIRLINE_TTL)    return airlinedef airline_summary(icao: str) -> AirlineSummary | None:    airline = fetch_airline(icao=icao)    if not airline:        return None    return AirlineSummary(        icao=airline.get("icao"),        iata=airline.get("iata"),        name=airline["name"],        callsign=airline.get("callsign"),        country=airline.get("country"),        is_active=airline.get("active") == "Y",        logo=airline.get("logo"),    )if __name__ == "__main__":    import json    from dataclasses import asdict    summary = airline_summary("AAL")    print(json.dumps(asdict(summary), indent=2) if summary else "Airline not found")