nedviga-cli 0.1.0__py3-none-any.whl

nedviga/metro_vestibules.py

@@ -0,0 +1,134 @@
+"""Directory of Moscow metro vestibules with real WGS-84 coordinates.
+
+The coordinates are taken from public cartographic sources
+(OpenStreetMap / Wikipedia) and represent the actual street-level exits.
+"""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+
+
+def _normalize(name: str) -> str:
+    """Normalize a station name for directory lookup."""
+    return name.strip().casefold().replace("ё", "е")
+
+
+#: Mapping station name -> list of (vestibule_name, lat, lon).
+VESTIBULES: dict[str, list[tuple[str, float, float]]] = {
+    # 1. Комсомольская (Sokolnicheskaya + Koltsevaya lines, Komsomolskaya square)
+    _normalize("Комсомольская"): [
+        ("Вестибюль на Комсомольской площади (Сокольническая линия)", 55.7753, 37.6555),
+        ("Вестибюль у Ленинградского вокзала (Кольцевая линия)", 55.7747, 37.6548),
+    ],
+    # 2. Чертановская
+    _normalize("Чертановская"): [
+        ("Вестибюль 1 (южный)", 55.6405, 37.6067),
+        ("Вестибюль 2 (северный)", 55.6412, 37.6073),
+    ],
+    # 3. Тверская
+    _normalize("Тверская"): [
+        ("Вестибюль на Пушкинской площади", 55.7648, 37.6042),
+    ],
+    # 4. Арбатская (Arbatsko-Pokrovskaya + Filyovskaya)
+    _normalize("Арбатская"): [
+        ("Вестибюль на улице Воздвиженка (Арбатско-Покровская линия)", 55.7523, 37.6039),
+        ("Вестибюль у Старого Арбата (Филёвская линия)", 55.7521, 37.6019),
+    ],
+    # 5. Курская (Arbatsko-Pokrovskaya + Koltsevaya)
+    _normalize("Курская"): [
+        ("Вестибюль у Курского вокзала (Арбатско-Покровская линия)", 55.7572, 37.6591),
+        ("Вестибюль на Земляном валу (Кольцевая линия)", 55.7566, 37.6597),
+    ],
+    # 6. Киевская (Filyovskaya + Arbatsko-Pokrovskaya + Koltsevaya)
+    _normalize("Киевская"): [
+        ("Вестибюль у Киевского вокзала (Филёвская линия)", 55.7440, 37.5666),
+        ("Вестибюль на площади Киевского вокзала (Арбатско-Покровская линия)", 55.7437, 37.5645),
+        ("Вестибюль на Киевской площади (Кольцевая линия)", 55.7436, 37.5663),
+    ],
+    # 7. Таганская (Tagansko-Krasnopresnenskaya + Koltsevaya)
+    _normalize("Таганская"): [
+        ("Вестибюль на Таганской площади (Таганско-Краснопресненская линия)", 55.7393, 37.6538),
+        ("Вестибюль у Таганской радиальной (Кольцевая линия)", 55.7390, 37.6534),
+    ],
+    # 8. Баррикадная
+    _normalize("Баррикадная"): [
+        ("Вестибюль на Баррикадной улице", 55.7612, 37.5812),
+    ],
+    # 9. ВДНХ
+    _normalize("ВДНХ"): [
+        ("Вестибюль у главного входа ВДНХ", 55.8203, 37.6411),
+    ],
+    # 10. Сокольники
+    _normalize("Сокольники"): [
+        ("Вестибюль у Сокольнического круга", 55.7891, 37.6803),
+    ],
+    # 11. Автозаводская
+    _normalize("Автозаводская"): [
+        ("Вестибюль на Автозаводской улице", 55.7066, 37.6571),
+    ],
+    # 12. Пролетарская
+    _normalize("Пролетарская"): [
+        ("Вестибюль на Крутицком валу", 55.7319, 37.6658),
+    ],
+    # 13. Тульская
+    _normalize("Тульская"): [
+        ("Вестибюль на Большой Тульской улице", 55.7095, 37.6229),
+    ],
+    # 14. Нагатинская
+    _normalize("Нагатинская"): [
+        ("Вестибюль на Варшавском шоссе", 55.6830, 37.6224),
+    ],
+    # 15. Павелецкая (Zamoskvoretskaya + Koltsevaya)
+    _normalize("Павелецкая"): [
+        ("Вестибюль у Павелецкого вокзала (Замоскворецкая линия)", 55.7304, 37.6364),
+        ("Вестибюль на Павелецкой площади (Кольцевая линия)", 55.7317, 37.6386),
+    ],
+}
+
+
+def get_vestibules(station: str) -> list[tuple[str, float, float]] | None:
+    """Return vestibule list for a station, or ``None`` if unknown.
+
+    v0.2.0: сначала сгенерированный справочник из OSM
+    (metro_directory.json, 237 станций метро + МЦД), затем ручной хардкод MVP.
+    """
+    k = _normalize(station)
+    generated = _load_generated()
+    if generated is not None and k in generated:
+        entry = generated[k]
+        return [(v["name"], v["lat"], v["lon"]) for v in entry["vestibules"]]
+    return VESTIBULES.get(k)
+
+
+_GENERATED_CACHE = None
+
+
+def _load_generated() -> dict | None:
+    """Load the generated OSM directory (metro_directory.json), cached."""
+    global _GENERATED_CACHE
+    if _GENERATED_CACHE is not None:
+        return _GENERATED_CACHE
+    candidates = [
+        Path(__file__).parent / "metro_directory.json",
+        Path(__file__).parent.parent / "metro_directory.json",
+    ]
+    import os
+    env = os.environ.get("NEDVIGA_METRO_DIR")
+    if env:
+        candidates.insert(0, Path(env))
+    for c in candidates:
+        try:
+            if c.exists():
+                _GENERATED_CACHE = json.loads(c.read_text(encoding="utf-8"))
+                return _GENERATED_CACHE
+        except Exception:
+            continue
+    return None
+
+
+# Директория станций из сгенерированного справочника (для CLI "list-stations" и тестов)
+def all_stations() -> dict[str, dict]:
+    """Return {norm_key: {name, transit_type, vestibules}} из сгенерированного справочника."""
+    return _load_generated() or {}

nedviga/models.py

@@ -0,0 +1,164 @@
+"""Pydantic data models for the nedviga-cli JSON contract."""
+
+from __future__ import annotations
+
+from datetime import datetime, timezone
+from typing import Any, Literal, Optional
+
+from pydantic import BaseModel, Field, computed_field
+
+
+class ApiOutcomeEnvelope(BaseModel):
+    """Strict single-JSON contract for every CLI command."""
+
+    ok: bool
+    status: Literal["ok", "degraded", "error"]
+    reason: Optional[str] = None
+    source: Optional[str] = None
+    parsed_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    data: Optional[Any] = None
+    metadata: dict = Field(default_factory=dict)
+
+
+class PriceHistoryPoint(BaseModel):
+    date: str
+    price: float
+
+
+class Pricing(BaseModel):
+    price_total_rub: Optional[float] = None
+    price_per_m2_rub: Optional[float] = None
+    price_history: list[PriceHistoryPoint] = Field(default_factory=list)
+    # v0.3.1 (реалтор-аудит §1): юридические условия сделки
+    sale_type: Optional[str] = None          # free / alternative
+    mortgage_allowed: Optional[bool] = None
+    price_usd: Optional[float] = None
+    price_eur: Optional[float] = None
+    # v0.3.4 (ТЗ реалтора): торги/аукционы ДГИ/ЦУГИ
+    is_auction: Optional[bool] = None
+    trading_link: Optional[str] = None
+
+
+class Parameters(BaseModel):
+    rooms: Optional[int] = None
+    rooms_type: Optional[str] = None
+    area_total_m2: Optional[float] = None
+    area_living_m2: Optional[float] = None
+    area_kitchen_m2: Optional[float] = None
+    floor: Optional[int] = None
+    floors_total: Optional[int] = None
+    ceiling_height_m: Optional[float] = None
+    balcony: Optional[str] = None
+    bathroom: Optional[str] = None
+    renovation: Optional[str] = None
+    window_view: Optional[str] = None
+    # v0.3.1 (реалтор-аудит §3): физические параметры раздельно
+    loggias_count: Optional[int] = None
+    balconies_count: Optional[int] = None
+    separate_wcs_count: Optional[int] = None
+    combined_wcs_count: Optional[int] = None
+    furniture: Optional[str] = None          # furnished / partially / no
+    layouts: list[str] = Field(default_factory=list)  # URL схем БТИ
+    # v0.4.0 (ТЗ реалтора): доля в квартире
+    share_amount: Optional[float] = None
+
+
+class Building(BaseModel):
+    address_raw: Optional[str] = None
+    # v0.4.3 (ТЗ the task brief, realtor): ЖК/комплекс из сниппета; только если явно помечен.
+    complex: Optional[str] = None
+    house_series: Optional[str] = None
+    year_built: Optional[int] = None
+    building_type: Optional[str] = None
+    floor_material: Optional[str] = None
+    renovation_program: Optional[bool] = None
+    elevators_passenger: Optional[int] = None
+    elevators_cargo: Optional[int] = None
+    gas: Optional[bool] = None
+    # v0.3.1 (реалтор-аудит §2, §4): здание + флаги
+    has_garbage_chute: Optional[bool] = None
+    parking_type: Optional[str] = None       # ground / underground
+    is_apartments: Optional[bool] = None
+    is_illegal_construction: Optional[bool] = None
+
+
+class CianClaimedMetro(BaseModel):
+    station: Optional[str] = None
+    claimed_time_minutes: Optional[int] = None
+    transport_type: Optional[str] = None
+
+
+class HonestWalkRoute(BaseModel):
+    target_station: Optional[str] = None
+    target_vestibule: Optional[str] = None
+    walk_distance_meters: Optional[int] = None
+    walk_duration_minutes: Optional[float] = None
+    routing_provider: Optional[str] = None
+    route_verified: bool = False
+
+
+class Geo(BaseModel):
+    lat: Optional[float] = None
+    lon: Optional[float] = None
+    district: Optional[str] = None
+    area_name: Optional[str] = None
+    cian_claimed_metro: Optional[CianClaimedMetro] = None
+    honest_walk_route: Optional[HonestWalkRoute] = None
+    honest_overclaim: Optional[dict] = None  # v0.3.2: {'overclaim_minutes': N, 'flag': bool}
+
+
+class Seller(BaseModel):
+    seller_type: Optional[str] = None
+    name: Optional[str] = None
+    agent_phone_masked: Optional[str] = None
+    # v0.3.1 (реалтор-аудит §4, §5)
+    is_by_homeowner: Optional[bool] = None
+    phones: list[str] = Field(default_factory=list)
+    created_at: Optional[str] = None          # дата публикации
+    edit_date: Optional[str] = None           # дата последнего изменения
+
+
+class Image(BaseModel):
+    url: str
+    category: Optional[str] = None
+
+
+class ListingCard(BaseModel):
+    """Full listing card schema used by import/get-card/search."""
+
+    id: str
+    source: str
+    url: str
+    status: Literal["active", "inactive", "sold", "removed"] = "active"
+    title: Optional[str] = None
+
+    pricing: Pricing = Field(default_factory=Pricing)
+    parameters: Parameters = Field(default_factory=Parameters)
+    building: Building = Field(default_factory=Building)
+    geo: Geo = Field(default_factory=Geo)
+
+    description_text: Optional[str] = None
+    # v0.4.0 (ТЗ реалтора): категория объявления (flatShare и т.д.)
+    category: Optional[str] = None
+    seller: Seller = Field(default_factory=Seller)
+    images: list[Image] = Field(default_factory=list)
+
+    @computed_field
+    @property
+    def missing_fields(self) -> list[str]:
+        missing: list[str] = []
+        if self.pricing.price_total_rub is None:
+            missing.append("price_total_rub")
+        if self.parameters.area_total_m2 is None:
+            missing.append("area_total_m2")
+        if self.parameters.rooms is None:
+            missing.append("rooms")
+        if self.parameters.floor is None:
+            missing.append("floor")
+        if self.parameters.floors_total is None:
+            missing.append("floors_total")
+        if not (self.building.address_raw or "").strip():
+            missing.append("address_raw")
+        if not (self.geo.district or "").strip():
+            missing.append("geo.district")
+        return missing

nedviga/mosdata.py

@@ -0,0 +1,95 @@
+"""v0.5.8 (the task brief) mosdata client."""
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+from typing import Any, Optional
+
+import requests
+
+BASE_URL = "https://apidata.mos.ru/v1"
+
+
+def _token() -> Optional[str]:
+    p = os.environ.get("NEDVIGA_MOS_TOKEN_FILE")
+    if not p:
+        return None
+    try:
+        return Path(p).read_text(encoding="utf-8").strip()
+    except Exception:
+        return None
+
+
+def _proxy() -> Optional[str]:
+    return os.environ.get("NEDVIGA_EGRESS_PROXY") or None
+
+
+
+
+def _redact(s: str) -> str:
+    import re
+    return re.sub(r"api_key=[^&\s]+", "api_key=REDACTED", str(s))
+
+
+class _SafeError(RuntimeError):
+    def __str__(self):
+        return _redact(super().__str__())
+
+def _request(method: str, path: str, params=None, timeout=30.0) -> Any:
+    from .net_guard import guarded_get
+    url = BASE_URL.rstrip("/") + "/" + path.lstrip("/")
+    headers = {}
+    tok = _token()
+    if tok is None:
+        raise RuntimeError("token_missing")
+    # v0.5.8 live: apidata v1 принимает ключ КАК QUERY-ПАРАМЕТР api_key (заголовок не канон).
+    # Ключ в URL = риск утечки в exception/лог: все строки ошибок прогоняются через redact.
+    params = dict(params or {})
+    params["api_key"] = tok
+    # v0.5.8 контроль качества: ЕДИНСТВЕННЫЙ транспорт — guarded_get (allowlist + RU-egress
+    # внутри netguard по env NEDVIGA_EGRESS_PROXY). Прежний двойной путь (прямые requests
+    # c прокси в обход пина) вырезан: он нарушал инвариант транспорта и молча глотал ошибки.
+    import urllib.parse
+    if params:
+        url = url + "?" + urllib.parse.urlencode(params)
+    try:
+        resp = guarded_get(url, timeout=timeout, headers=headers)
+        resp.raise_for_status()
+        return resp.json()
+    except Exception as exc:
+        msg = _redact(str(exc))
+        err = _SafeError(f"{type(exc).__name__}: {msg}")
+        raise err from None
+
+
+def search_datasets(query: str) -> dict:
+    params = {"$filter": f"contains(Name,'{query}')"}
+    return _request("GET", "datasets", params=params)
+
+
+def dataset_versions(dataset_id: str) -> Any:
+    return _request("GET", f"datasets/{dataset_id}/versions")
+
+
+def rows(dataset_id: str, version: Optional[str] = None, top: int = 1000, skip: int = 0, filter_expr: Optional[str] = None) -> Any:
+    path = f"datasets/{dataset_id}/rows"
+    params = {"$top": top, "$skip": skip}
+    if version:
+        params["versionNumber"] = version
+    if filter_expr:
+        params["$filter"] = filter_expr
+    return _request("GET", path, params=params)
+
+
+def save_discovery(data: dict) -> None:
+    base = Path(os.environ.get("NEDVIGA_STATE_DIR", str(Path.home() / ".local" / "share" / "nedviga")))
+    base.mkdir(parents=True, exist_ok=True)
+    p = base / "mos_datasets.json"
+    # store only ids from live payload
+    ids = []
+    for d in (data.get("datasets") or data.get("value") or []):
+        if isinstance(d, dict) and d.get("Id") is not None:
+            ids.append(d["Id"])
+    payload = {"datasets": data.get("datasets") or data.get("value") or [], "ids": ids}
+    p.write_text(json.dumps(payload, ensure_ascii=False, indent=2), encoding="utf-8")