opensky-cli 0.1.0__py3-none-any.whl

opensky/config.py

@@ -0,0 +1,151 @@
+from __future__ import annotations
+
+import sys
+import tomllib
+from datetime import date, timedelta
+from pathlib import Path
+
+from pydantic import BaseModel, field_validator
+
+
+class DateRange(BaseModel):
+    start: str
+    end: str
+
+    @field_validator("start", "end")
+    @classmethod
+    def validate_date(cls, v: str) -> str:
+        date.fromisoformat(v)
+        return v
+
+    def dates(self) -> list[str]:
+        s = date.fromisoformat(self.start)
+        e = date.fromisoformat(self.end)
+        result = []
+        while s <= e:
+            result.append(s.isoformat())
+            s += timedelta(days=1)
+        return result
+
+
+class SearchConfig(BaseModel):
+    origins: list[str]
+    destinations: list[str]
+    date_range: DateRange
+    cabin: str = "economy"
+    currency: str = "EUR"
+    stops: str = "any"
+    max_price: float = 0
+
+
+_RISK_ALIASES: dict[str, str] = {
+    "avoid": "do_not_fly",
+    "risky": "high_risk",
+    "caution": "caution",
+    "do_not_fly": "do_not_fly",
+    "high_risk": "high_risk",
+}
+
+
+class SafetyConfig(BaseModel):
+    risk_threshold: str = "high_risk"
+
+    @field_validator("risk_threshold")
+    @classmethod
+    def normalize_risk(cls, v: str) -> str:
+        normalized = _RISK_ALIASES.get(v.lower().strip())
+        if normalized is None:
+            valid = ", ".join(_RISK_ALIASES.keys())
+            raise ValueError(f"Unknown risk_threshold '{v}'. Use: {valid}")
+        return normalized
+
+
+class ScoringConfig(BaseModel):
+    price_weight: float = 1.0
+    duration_weight: float = 0.5
+
+
+class ConnectionsConfig(BaseModel):
+    final_destination: str = ""
+    transit_hours: dict[str, float] = {}
+
+
+class ScanConfig(BaseModel):
+    search: SearchConfig
+    safety: SafetyConfig = SafetyConfig()
+    scoring: ScoringConfig = ScoringConfig()
+    connections: ConnectionsConfig = ConnectionsConfig()
+
+
+def _validate_date_range(cfg: ScanConfig) -> None:
+    """Validate date range: start <= end, warn if start is in the past."""
+    s = date.fromisoformat(cfg.search.date_range.start)
+    e = date.fromisoformat(cfg.search.date_range.end)
+    if s > e:
+        raise ValueError(
+            f"Start date {cfg.search.date_range.start} is after end date "
+            f"{cfg.search.date_range.end}. Start must be before end."
+        )
+    if s < date.today():
+        print(
+            f"Warning: start date {cfg.search.date_range.start} is in the past.",
+            file=sys.stderr,
+        )
+
+
+def load_config(path: str | Path) -> ScanConfig:
+    p = Path(path)
+    with open(p, "rb") as f:
+        data = tomllib.load(f)
+    cfg = ScanConfig(**data)
+
+    _validate_date_range(cfg)
+
+    # Resolve city names to IATA codes in origins/destinations
+    from opensky.airports import resolve_airport
+
+    try:
+        cfg.search.origins = [resolve_airport(o, interactive=False) for o in cfg.search.origins]
+        cfg.search.destinations = [resolve_airport(d, interactive=False) for d in cfg.search.destinations]
+    except ValueError as e:
+        raise ValueError(str(e)) from None
+
+    return cfg
+
+
+EXAMPLE_CONFIG = """\
+[search]
+# Airport codes (city names work in CLI: opensky search bangalore hamburg ...)
+origins = ["BLR", "DEL", "BOM", "KUL", "BKK", "SIN"]  # Bangalore, Delhi, Mumbai, Kuala Lumpur, Bangkok, Singapore
+destinations = ["HAM", "FRA", "MUC", "BER", "AMS", "CPH"]  # Hamburg, Frankfurt, Munich, Berlin, Amsterdam, Copenhagen
+cabin = "economy"   # economy, premium_economy, business, first
+currency = "EUR"
+stops = "any"   # any, nonstop, 1, 2
+# max_price = 500  # maximum price (0 = no limit)
+
+[search.date_range]
+start = "2026-03-10"
+end = "2026-03-20"
+
+[safety]
+risk_threshold = "risky"   # hide flights at this level or above: avoid, risky, caution
+
+[scoring]
+price_weight = 1.0
+duration_weight = 0.5
+
+# Optional: if your final destination is one city, add ground travel time
+# from each arrival airport. This helps scoring rank "fly to nearby city +
+# short train" vs "fly direct but expensive".
+[connections]
+final_destination = "Hamburg"
+
+[connections.transit_hours]
+HAM = 0       # direct, no ground travel
+HAJ = 1.5     # Hannover: 1.5h train to Hamburg
+BER = 2       # Berlin: 2h train to Hamburg
+FRA = 4       # Frankfurt: 4h train to Hamburg
+MUC = 5.5
+AMS = 5
+CPH = 5
+"""

opensky/data/conflict_zones.json

@@ -0,0 +1,203 @@
+{
+  "metadata": {
+    "version": "1.0.0",
+    "updated": "2026-03-01",
+    "sources": [
+      "EASA Conflict Zone Information Bulletins (CZIB)",
+      "safeairspace.net",
+      "FAA NOTAM database"
+    ]
+  },
+  "zones": [
+    {
+      "id": "ukraine",
+      "name": "Ukraine",
+      "risk_level": "do_not_fly",
+      "countries": ["UA"],
+      "airports": [],
+      "source": "EASA CZIB 2022-01 R16",
+      "details": "Active conflict zone. All Ukrainian airspace closed to civil aviation.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "iran",
+      "name": "Iran",
+      "risk_level": "do_not_fly",
+      "countries": ["IR"],
+      "airports": [],
+      "source": "EASA CZIB 2026-002",
+      "details": "Military strikes Feb 2026. FIRs OIIX, OIBB closed to overflights.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "iraq",
+      "name": "Iraq",
+      "risk_level": "do_not_fly",
+      "countries": ["IQ"],
+      "airports": [],
+      "source": "EASA CZIB 2026-002",
+      "details": "Airspace restrictions due to regional conflict.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "syria",
+      "name": "Syria",
+      "risk_level": "do_not_fly",
+      "countries": ["SY"],
+      "airports": [],
+      "source": "EASA CZIB 2024-003",
+      "details": "Ongoing conflict. Damascus FIR closed to overflights.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "israel",
+      "name": "Israel",
+      "risk_level": "do_not_fly",
+      "countries": ["IL"],
+      "airports": [],
+      "source": "EASA CZIB 2024-005",
+      "details": "Active conflict zone. Intermittent airspace closures.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "lebanon",
+      "name": "Lebanon",
+      "risk_level": "do_not_fly",
+      "countries": ["LB"],
+      "airports": [],
+      "source": "EASA CZIB 2024-006",
+      "details": "Beirut FIR affected by regional conflict.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "yemen",
+      "name": "Yemen",
+      "risk_level": "do_not_fly",
+      "countries": ["YE"],
+      "airports": [],
+      "source": "EASA CZIB 2024-007",
+      "details": "Ongoing conflict and Houthi activity affecting airspace.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "libya",
+      "name": "Libya",
+      "risk_level": "do_not_fly",
+      "countries": ["LY"],
+      "airports": [],
+      "source": "EASA CZIB 2023-001",
+      "details": "Unstable security situation.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "somalia",
+      "name": "Somalia",
+      "risk_level": "do_not_fly",
+      "countries": ["SO"],
+      "airports": [],
+      "source": "EASA CZIB 2023-002",
+      "details": "Ongoing conflict. Limited ATC capability.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "afghanistan",
+      "name": "Afghanistan",
+      "risk_level": "do_not_fly",
+      "countries": ["AF"],
+      "airports": [],
+      "source": "EASA CZIB 2022-002",
+      "details": "No functioning civil aviation authority.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "north_korea",
+      "name": "North Korea",
+      "risk_level": "do_not_fly",
+      "countries": ["KP"],
+      "airports": [],
+      "source": "FAA NOTAM",
+      "details": "Prohibited airspace. Missile launch risk.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "sudan",
+      "name": "Sudan",
+      "risk_level": "do_not_fly",
+      "countries": ["SD"],
+      "airports": [],
+      "source": "EASA CZIB 2023-003",
+      "details": "Civil war. Khartoum FIR dangerous.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "gulf_states",
+      "name": "Gulf States (Airspace Risk)",
+      "risk_level": "high_risk",
+      "countries": [],
+      "airports": ["DXB", "AUH", "SHJ", "RKT", "DOH", "BAH", "KWI", "MCT"],
+      "source": "safeairspace.net",
+      "details": "Adjacent to Iran/Iraq conflict. Overflights at risk during escalations. Gulf FIRs periodically restricted.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "eritrea",
+      "name": "Eritrea",
+      "risk_level": "high_risk",
+      "countries": ["ER"],
+      "airports": [],
+      "source": "EASA CZIB 2023-004",
+      "details": "Military activity near borders.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "ethiopia_partial",
+      "name": "Ethiopia (Northern Regions)",
+      "risk_level": "high_risk",
+      "countries": [],
+      "airports": ["MQX", "AXU", "GDQ", "SHC"],
+      "source": "EASA CZIB 2023-005",
+      "details": "Tigray region conflict aftermath. Northern airports affected.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "mali",
+      "name": "Mali",
+      "risk_level": "high_risk",
+      "countries": ["ML"],
+      "airports": [],
+      "source": "EASA CZIB 2023-006",
+      "details": "Armed groups active. Limited ATC in northern regions.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "niger",
+      "name": "Niger",
+      "risk_level": "high_risk",
+      "countries": ["NE"],
+      "airports": [],
+      "source": "EASA CZIB 2023-007",
+      "details": "Post-coup instability.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "pakistan_partial",
+      "name": "Pakistan (Border Regions)",
+      "risk_level": "caution",
+      "countries": [],
+      "airports": ["PEW", "UET", "TUK"],
+      "source": "safeairspace.net",
+      "details": "Western border areas near Afghanistan/Iran. Major airports (ISB, LHE, KHI) unaffected.",
+      "updated": "2026-02-28"
+    },
+    {
+      "id": "russia_partial",
+      "name": "Russia (Southern/Western Border)",
+      "risk_level": "caution",
+      "countries": [],
+      "airports": ["ROV", "KRR", "VOZ", "BEL"],
+      "source": "EASA CZIB 2022-01",
+      "details": "Southern Russian airports near Ukraine conflict. Moscow/St. Petersburg unaffected.",
+      "updated": "2026-02-28"
+    }
+  ]
+}