Create navitia.py

services/providers/navitia.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+from typing import Optional, Dict, Any
+import os
+import requests
+from datetime import datetime
+
+UA = {"User-Agent": "HF-Space-Trip-Planner/1.0 (+navitia)"}
+
+def _navitia_token() -> Optional[str]:
+    # NavitiaのAPIキーを .env / Secrets に設定してください（例: NAVITIA_TOKEN=xxxxxxxx）
+    return os.getenv("NAVITIA_TOKEN")
+
+def _fmt_datetime(when_iso: str) -> str:
+    """
+    入力: "YYYY-MM-DDTHH:MM:SS" or "YYYY-MM-DD"
+    出力: "YYYYMMDDTHHMMSS"（Navitia形式）
+    """
+    try:
+        if "T" in when_iso:
+            dt = datetime.fromisoformat(when_iso)
+        else:
+            dt = datetime.fromisoformat(when_iso + "T09:00:00")
+        return dt.strftime("%Y%m%dT%H%M%S")
+    except Exception:
+        return datetime.utcnow().strftime("%Y%m%dT%H%M%S")
+
+def get_journey(
+    lat_from: float,
+    lon_from: float,
+    lat_to: float,
+    lon_to: float,
+    when_iso: str,
+) -> Optional[Dict[str, Any]]:
+    """
+    Navitiaで経路を取得（トークンが無い/失敗時は None を返すフェイルセーフ）。
+    戻り:
+      {"duration_s": int, "fare": {...}}  など（存在するもののみ）
+    注意:
+      Navitiaは地域カバレッジに制約があります。対象外エリアではNoneになります。
+    """
+    token = _navitia_token()
+    if not token:
+        return None
+
+    url = "https://api.navitia.io/v1/journeys"
+    params = {
+        "from": f"{lon_from};{lat_from}",
+        "to": f"{lon_to};{lat_to}",
+        "datetime": _fmt_datetime(when_iso),
+        "count": 1,
+    }
+
+    try:
+        r = requests.get(url, params=params, headers=UA, auth=(token, ""), timeout=20)
+        if r.status_code >= 400:
+            return None
+        j = r.json()
+        js = j.get("journeys") or []
+        if not js:
+            return None
+        best = js[0]
+        out: Dict[str, Any] = {}
+        if "duration" in best:
+            out["duration_s"] = int(best["duration"])
+        fare = (best.get("fare") or {}).get("total") or best.get("fare")
+        if fare:
+            out["fare"] = fare
+        return out or None
+    except Exception:
+        return None