kundali-chart-mcp 0.2.1

package/python/kundali_lib/vedic/zodiac.py

@@ -0,0 +1,42 @@
+"""Zodiac helpers: rashi and nakshatra from longitude."""
+
+from kundali_lib.vedic.constants import NAKSHATRA_LORDS, NAKSHATRAS, RASHI_LORDS, RASHIS
+
+NAKSHATRA_SPAN = 360 / 27
+
+
+def get_rashi(longitude: float) -> str:
+    i = int(longitude / 30) % 12
+    return RASHIS[i]
+
+
+def get_rashi_lord(longitude: float) -> str:
+    i = int(longitude / 30) % 12
+    return RASHI_LORDS[i]
+
+
+def get_nakshatra(longitude: float) -> str:
+    i = int(longitude / NAKSHATRA_SPAN) % 27
+    return NAKSHATRAS[i]
+
+
+def get_nakshatra_lord(longitude: float) -> str:
+    i = int(longitude / NAKSHATRA_SPAN) % 27
+    return NAKSHATRA_LORDS[i]
+
+
+def get_pada(longitude: float) -> int:
+    within = longitude % NAKSHATRA_SPAN
+    return int(within / (NAKSHATRA_SPAN / 4)) + 1
+
+
+def position_dict(longitude: float) -> dict:
+    """Build rashi/nakshatra dict for a longitude."""
+    return {
+        "rashi": get_rashi(longitude),
+        "rashi_lord": get_rashi_lord(longitude),
+        "degree": longitude % 30,
+        "nakshatra": get_nakshatra(longitude),
+        "nakshatra_lord": get_nakshatra_lord(longitude),
+        "pada": get_pada(longitude),
+    }

package/python/kundali_lib/vedicastro_bridge.py

@@ -0,0 +1,198 @@
+"""Thin wrapper around vedicastro.VedicAstro for use in the MCP bridge.
+
+vedicastro provides BPHS-grade features:
+  - Full planetary data with SubLord/SubSubLord (KP system)
+  - House cusps with Nakshatra/SubLord
+  - Vimshottari Dasha (mahadasha + antardasha)
+  - Planetary aspects
+  - Planet-wise and house-wise significators
+
+Supported ayanamsas by vedicastro:
+  Lahiri, Lahiri_1940, Lahiri_VP285, Lahiri_ICRC, Raman,
+  Krishnamurti, Krishnamurti_Senthilathiban
+
+Supported house systems by vedicastro:
+  Placidus, Equal, Equal 2, Whole Sign
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+# vedicastro.VedicAstro is imported lazily so the module can be imported
+# even if vedicastro is not installed (our own pyswisseph-based path is the default).
+_VEDICASTRO_AVAILABLE = False
+
+try:
+    from vedicastro.VedicAstro import VedicHoroscopeData as _VedicHoroscopeData
+
+    _VEDICASTRO_AVAILABLE = True
+except ImportError:
+    _VedicHoroscopeData = None
+
+SUPPORTED_AYANAMSAS = [
+    "Lahiri",
+    "Lahiri_1940",
+    "Lahiri_VP285",
+    "Lahiri_ICRC",
+    "Raman",
+    "Krishnamurti",
+    "Krishnamurti_Senthilathiban",
+]
+
+SUPPORTED_HOUSE_SYSTEMS = ["Placidus", "Equal", "Equal 2", "Whole Sign"]
+
+DEFAULT_AYANAMSA = "Lahiri"
+DEFAULT_HOUSE_SYSTEM = "Placidus"
+
+
+def is_available() -> bool:
+    return _VEDICASTRO_AVAILABLE
+
+
+def _make_horoscope(payload: dict[str, Any]) -> "_VedicHoroscopeData":  # type: ignore[name-defined]
+    if not _VEDICASTRO_AVAILABLE:
+        raise RuntimeError(
+            "vedicastro is not installed in this environment. "
+            "Run: pip install vedicastro flatlib --no-deps"
+        )
+
+    year = int(payload["year"])
+    month = int(payload["month"])
+    day = int(payload["day"])
+    hour = int(payload["hour"])
+    minute = int(payload["minute"])
+    second = int(payload.get("second") or 0)
+    utc = str(payload["utc_offset"])  # e.g. "+05:30"
+    latitude = float(payload["latitude"])
+    longitude = float(payload["longitude"])
+    ayanamsa = str(payload.get("ayanamsa") or DEFAULT_AYANAMSA)
+    house_system = str(payload.get("house_system") or DEFAULT_HOUSE_SYSTEM)
+
+    return _VedicHoroscopeData(
+        year,
+        month,
+        day,
+        hour,
+        minute,
+        second,
+        utc,
+        latitude,
+        longitude,
+        ayanamsa=ayanamsa,
+        house_system=house_system,
+    )
+
+
+def _namedtuple_to_dict(obj: Any) -> Any:
+    """Recursively convert namedtuples and lists to JSON-friendly dicts."""
+    if obj is None:
+        return None
+    if hasattr(obj, "_asdict"):
+        return {k: _namedtuple_to_dict(v) for k, v in obj._asdict().items()}
+    if isinstance(obj, list):
+        return [_namedtuple_to_dict(i) for i in obj]
+    if isinstance(obj, dict):
+        return {k: _namedtuple_to_dict(v) for k, v in obj.items()}
+    return obj
+
+
+def get_full_chart(payload: dict[str, Any]) -> dict[str, Any]:
+    """Return planets + house cusps (full enhanced chart with SubLord/KP data)."""
+    h = _make_horoscope(payload)
+    chart = h.generate_chart()
+    planets = h.get_planets_data_from_chart(chart)
+    houses = h.get_houses_data_from_chart(chart)
+    return {
+        "ayanamsa": h.get_ayanamsa(),
+        "house_system": h.get_house_system(),
+        "planetary_positions": _namedtuple_to_dict(planets),
+        "house_cusps": _namedtuple_to_dict(houses),
+    }
+
+
+def get_vimshottari_dasha(payload: dict[str, Any]) -> dict[str, Any]:
+    """Return full Vimshottari Dasha (mahadasha + antardasha) for the chart."""
+    h = _make_horoscope(payload)
+    chart = h.generate_chart()
+    dasha = h.compute_vimshottari_dasa(chart)
+    return {
+        "ayanamsa": h.get_ayanamsa(),
+        "house_system": h.get_house_system(),
+        "vimshottari_dasha": dasha,
+    }
+
+
+def get_planetary_aspects(payload: dict[str, Any]) -> dict[str, Any]:
+    """Return all planetary aspects (conjunction, sextile, square, trine, opposition, etc.)."""
+    h = _make_horoscope(payload)
+    chart = h.generate_chart()
+    aspects = h.get_planetary_aspects(chart)
+    return {
+        "ayanamsa": h.get_ayanamsa(),
+        "house_system": h.get_house_system(),
+        "aspects": [
+            {k: v for k, v in a.items()}
+            if isinstance(a, dict)
+            else _namedtuple_to_dict(a)
+            for a in aspects
+        ],
+    }
+
+
+def get_significators(payload: dict[str, Any]) -> dict[str, Any]:
+    """Return planet-wise and house-wise significators (KP system)."""
+    h = _make_horoscope(payload)
+    chart = h.generate_chart()
+    # Use the helper methods to get properly formatted data
+    planets_data = h.get_planets_data_from_chart(chart)
+    houses_data = h.get_houses_data_from_chart(chart)
+    planet_sig = h.get_planet_wise_significators(planets_data, houses_data)
+    house_sig = h.get_house_wise_significators(planets_data, houses_data)
+
+    def _planet_sig_row(row: list) -> dict:
+        return {
+            "planet": row[0] if len(row) > 0 else None,
+            "rasi_lord_house": row[1] if len(row) > 1 else None,
+            "nakshatra_lord_house": row[2] if len(row) > 2 else None,
+            "houses_in_same_rasi": row[3] if len(row) > 3 else [],
+            "houses_lordship": row[4] if len(row) > 4 else [],
+        }
+
+    def _house_sig_row(row: list) -> dict:
+        return {
+            "house": row[0] if len(row) > 0 else None,
+            "planets_in_house": row[1] if len(row) > 1 else [],
+            "planets_aspecting_house": row[2] if len(row) > 2 else [],
+            "planets_whose_lord_is_here": row[3] if len(row) > 3 else [],
+            "rasi_lord": row[4] if len(row) > 4 else None,
+        }
+
+    return {
+        "ayanamsa": h.get_ayanamsa(),
+        "house_system": h.get_house_system(),
+        "planet_wise_significators": [_planet_sig_row(r) for r in planet_sig],
+        "house_wise_significators": [_house_sig_row(r) for r in house_sig],
+    }
+
+
+def list_vedicastro_options(payload: dict[str, Any]) -> dict[str, Any]:
+    """List supported ayanamsas and house systems for vedicastro-powered tools."""
+    return {
+        "available": _VEDICASTRO_AVAILABLE,
+        "supported_ayanamsas": SUPPORTED_AYANAMSAS,
+        "supported_house_systems": SUPPORTED_HOUSE_SYSTEMS,
+        "default_ayanamsa": DEFAULT_AYANAMSA,
+        "default_house_system": DEFAULT_HOUSE_SYSTEM,
+        "tools": [
+            "get_extended_chart",
+            "get_vimshottari_dasha",
+            "get_planetary_aspects",
+            "get_significators",
+        ],
+        "note": (
+            "These tools use vedicastro (diliprk/VedicAstro) with flatlib. "
+            "They provide SubLord/KP data, Vimshottari Dasha, aspects, and significators. "
+            "generate_kundali uses a different engine (pyswisseph directly) and supports more ayanamsas and chart types."
+        ),
+    }