ziwei-chart 0.1.0__py3-none-any.whl

ziwei_chart/__init__.py

@@ -0,0 +1,103 @@
+"""ziwei-chart: deterministic Zi Wei Dou Shu natal chart casting.
+
+>>> from datetime import datetime
+>>> from zoneinfo import ZoneInfo
+>>> from ziwei_chart import BirthInput, cast_chart
+>>> chart = cast_chart(BirthInput(
+...     datetime=datetime(2004, 4, 25, 9, 20, tzinfo=ZoneInfo("Asia/Shanghai")),
+...     gender="女"))
+>>> chart.to_dict()["schema"]
+'ziwei_chart.v1'
+"""
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from datetime import datetime
+from typing import Literal
+
+from .calendar import CalendarRules, resolve_calendar
+from .models import Chart, HoroscopeChart
+from .natal import build_natal_chart
+from .rules import RuleSet
+
+Gender = Literal["男", "女"]
+
+
+@dataclass
+class BirthInput:
+    """Birth data for casting.
+
+    - ``datetime``: aware civil birth datetime with an IANA ZoneInfo.
+    - ``gender``: "男" or "女".
+    - ``longitude``/``latitude``: optional; when both are given the civil
+      time is converted to true solar time (requires the ``[solar]`` extra).
+      Place names are resolved by the caller.
+    - ``bazi``/``lunar_date_str``: optional precomputed calendar.  When both
+      are provided they are used as-is and ``datetime`` may be None — this is
+      the path used by applications that keep their own calendar service.
+    """
+
+    datetime: datetime | None = None
+    gender: Gender = "男"
+    longitude: float | None = None
+    latitude: float | None = None
+    bazi: str | None = None
+    lunar_date_str: str | None = None
+    calendar_rules: CalendarRules = field(default_factory=CalendarRules)
+
+    def __post_init__(self):
+        if self.gender not in ("男", "女"):
+            raise ValueError("紫微斗数暂不支持非二元性别")
+        has_precomputed = self.bazi is not None or self.lunar_date_str is not None
+        if has_precomputed and not (self.bazi and self.lunar_date_str):
+            raise ValueError("bazi and lunar_date_str must be provided together")
+        if not has_precomputed and self.datetime is None:
+            raise ValueError("either datetime or a precomputed bazi/lunar_date_str is required")
+
+
+def cast_chart(birth_input: BirthInput, rules: RuleSet | None = None) -> Chart:
+    """Cast the natal chart (and unlock horoscope views) from a BirthInput."""
+    rules = rules or RuleSet.default()
+    if birth_input.bazi and birth_input.lunar_date_str:
+        chart = cast_chart_from_calendar(
+            bazi=birth_input.bazi, lunar_date_str=birth_input.lunar_date_str,
+            gender=birth_input.gender, rules=rules,
+        )
+        if birth_input.datetime is not None:
+            chart.input_receipt["datetime"] = birth_input.datetime.isoformat()
+        return chart
+
+    calendar_info = resolve_calendar(
+        birth_input.datetime, birth_input.longitude, birth_input.latitude,
+        birth_input.calendar_rules,
+    )
+    chart = cast_chart_from_calendar(
+        bazi=calendar_info.bazi, lunar_date_str=calendar_info.lunar_date_str,
+        gender=birth_input.gender, rules=rules,
+    )
+    chart.calendar = calendar_info
+    chart.input_receipt.update(calendar_info.as_receipt())
+    if birth_input.datetime is not None:
+        chart.input_receipt["datetime"] = birth_input.datetime.isoformat()
+    if birth_input.longitude is not None and birth_input.latitude is not None:
+        chart.input_receipt["coordinates"] = [birth_input.latitude, birth_input.longitude]
+    return chart
+
+
+def cast_chart_from_calendar(bazi: str, lunar_date_str: str, gender: str,
+                             rules: RuleSet | None = None) -> Chart:
+    """Cast the natal chart from a precomputed bazi + lunar_date_str pair."""
+    return build_natal_chart(bazi=bazi, lunar_date_str=lunar_date_str,
+                             gender=gender, rules=rules)
+
+
+__all__ = [
+    "BirthInput",
+    "CalendarRules",
+    "Chart",
+    "HoroscopeChart",
+    "RuleSet",
+    "cast_chart",
+    "cast_chart_from_calendar",
+]
+__version__ = "0.1.0"

ziwei_chart/calendar.py

@@ -0,0 +1,160 @@
+"""Calendar resolution: true solar time, day/time-switch rules, bazi and
+lunar date computation via lunar-python.
+
+Default rules match the legacy suanqiu behaviour exactly:
+- ``day_rule="zi-early"``: 23:00 (子初) starts the next day pillar.
+- ``time_rule="next-day"``: the hour pillar for 23:00-24:00 is computed at
+  00:30 of the following civil day.
+- the lunar date used by the chart itself follows civil midnight, NOT the
+  day-pillar switch.
+"""
+from __future__ import annotations
+
+import re
+from dataclasses import dataclass
+from datetime import datetime, timedelta
+from typing import Literal, NamedTuple
+from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
+
+DayRule = Literal["midnight", "zi-early", "solar-midnight"]
+TimeRule = Literal["midnight", "prev-day", "next-day"]
+
+
+class LunarDateTime(NamedTuple):
+    year: int
+    month: int
+    day: int
+    hour: int
+    minute: int
+    second: int
+
+
+_LUNAR_DATETIME_RE = re.compile(r"^\s*(\d{1,4})-(\d{1,2})-(\d{1,2}) (\d{1,2}):(\d{1,2}):(\d{1,2})\s*$")
+
+
+def parse_lunar_datetime(text: str) -> LunarDateTime:
+    """解析 lunar_date_str。农历二月有廿九/三十，不能用 datetime.strptime。"""
+    match = _LUNAR_DATETIME_RE.match(text or "")
+    if not match:
+        raise ValueError(f"Invalid lunar date string: {text!r}")
+    value = LunarDateTime(*(int(part) for part in match.groups()))
+    if not (1 <= value.month <= 12 and 1 <= value.day <= 30
+            and value.hour < 24 and value.minute < 60 and value.second < 60):
+        raise ValueError(f"Invalid lunar date string: {text!r}")
+    return value
+
+
+@dataclass(frozen=True)
+class CalendarRules:
+    day_rule: DayRule = "zi-early"
+    time_rule: TimeRule = "next-day"
+
+
+@dataclass
+class CalendarInfo:
+    bazi: str
+    lunar_date_str: str
+    true_solar_time: datetime | None   # from_calendar 路径时为 None
+    timezone: str
+    day_rule: DayRule
+    time_rule: TimeRule
+    civil_datetime: datetime | None = None
+
+    def as_receipt(self) -> dict:
+        return {
+            "bazi": self.bazi,
+            "lunar_date_str": self.lunar_date_str,
+            "true_solar_time": (
+                self.true_solar_time.strftime("%Y-%m-%d %H:%M:%S")
+                if self.true_solar_time else None
+            ),
+            "timezone": self.timezone,
+            "calendar_rules": {"day_rule": self.day_rule, "time_rule": self.time_rule},
+        }
+
+
+def apply_day_switch(dt_ast: datetime, ast_hour_float: float, tz: str,
+                     day_rule: DayRule = "zi-early") -> datetime:
+    """day_rule: "midnight" 子正换日 | "zi-early" 子初换日 | "solar-midnight" 真太阳午夜."""
+    dt_ast = dt_ast.astimezone(ZoneInfo(tz))
+    if day_rule == "zi-early":
+        if 23.0 <= ast_hour_float < 24.0:
+            return dt_ast + timedelta(days=1)
+    elif day_rule == "solar-midnight":
+        if ast_hour_float < 0.5:
+            return dt_ast + timedelta(days=1)
+    # "midnight" 不动
+    return dt_ast
+
+
+def apply_time_switch(dt_for_day: datetime, ast_hour_float: float, tz: str,
+                      time_rule: TimeRule = "next-day") -> datetime:
+    """time_rule: "midnight" 民用日界 | "next-day" 夜子归次日 | "prev-day" 早子归前日."""
+    dt_for_day = dt_for_day.astimezone(ZoneInfo(tz))
+    if time_rule == "next-day" and 23.0 <= ast_hour_float < 24.0:
+        # 不要简单 +1 小时（会跨到次次日）；直接把时间“对齐”到当日 00:30
+        return dt_for_day.replace(hour=0, minute=30, second=0, microsecond=0)
+    if time_rule == "prev-day" and 0.0 <= ast_hour_float < 1.0:
+        prev = dt_for_day - timedelta(days=1)
+        return prev.replace(hour=23, minute=30, second=0, microsecond=0)
+    return dt_for_day
+
+
+def _timezone_name(dt: datetime) -> str:
+    tz_name = getattr(dt.tzinfo, "key", None)
+    if not tz_name:
+        raise ValueError("BirthInput.datetime must carry an IANA ZoneInfo timezone")
+    return tz_name
+
+
+def resolve_calendar(dt: datetime, longitude: float | None, latitude: float | None,
+                     rules: CalendarRules) -> CalendarInfo:
+    """Compute bazi + chart lunar date from an aware civil datetime.
+
+    With ``longitude``/``latitude`` the civil time is first converted to true
+    solar time (requires the ``[solar]`` extra).  Without them the datetime is
+    used as-is — no longitude or DST assumptions.
+    """
+    if dt.tzinfo is None or dt.utcoffset() is None:
+        raise ValueError("BirthInput.datetime must be timezone-aware")
+    tz = _timezone_name(dt)
+    if longitude is not None and latitude is not None:
+        from .solar_time import resolve_true_solar_time
+
+        dt_ast = resolve_true_solar_time(dt, lat=latitude, lon=longitude, tz=tz)
+    else:
+        dt_ast = dt.astimezone(ZoneInfo(tz))
+    h_ast = dt_ast.hour + dt_ast.minute / 60.0 + dt_ast.second / 3600.0
+
+    from lunar_python import Solar
+
+    dt_for_day = apply_day_switch(dt_ast, h_ast, tz, day_rule=rules.day_rule)
+    dt_for_bazi = apply_time_switch(dt_for_day, h_ast, tz, time_rule=rules.time_rule)
+    solar_obj = Solar.fromYmdHms(
+        dt_for_bazi.year, dt_for_bazi.month, dt_for_bazi.day,
+        dt_for_bazi.hour, dt_for_bazi.minute, dt_for_bazi.second,
+    )
+    ec = solar_obj.getLunar().getEightChar()
+
+    # 紫微使用实际排盘时刻的农历日期，不能继承八字日柱的子初换日。
+    chart_lunar = Solar.fromYmdHms(
+        dt_ast.year, dt_ast.month, dt_ast.day,
+        dt_ast.hour, dt_ast.minute, dt_ast.second,
+    ).getLunar()
+    lunar_year = chart_lunar.getYear()
+    lunar_month = abs(chart_lunar.getMonth())  # leap months come back negative
+    lunar_day = chart_lunar.getDay()
+    # 农历日期不能塞进公历 datetime：二月廿九/三十在公历二月不存在。只拼字符串。
+    lunar_date_str = (
+        f"{lunar_year:04d}-{lunar_month:02d}-{lunar_day:02d} "
+        f"{dt_ast.hour:02d}:{dt_ast.minute:02d}:{dt_ast.second:02d}"
+    )
+    return CalendarInfo(
+        bazi=f"{ec.getYear()}{ec.getMonth()}{ec.getDay()}{ec.getTime()}",
+        lunar_date_str=lunar_date_str,
+        true_solar_time=dt_ast,
+        timezone=tz,
+        day_rule=rules.day_rule,
+        time_rule=rules.time_rule,
+        civil_datetime=dt,
+    )

ziwei_chart/data/60jiazi.json

@@ -0,0 +1,62 @@
+{
+  "甲子": "金",
+  "乙丑": "金",
+  "丙寅": "火",
+  "丁卯": "火",
+  "戊辰": "木",
+  "己巳": "木",
+  "庚午": "土",
+  "辛未": "土",
+  "壬申": "金",
+  "癸酉": "金",
+  "甲戌": "火",
+  "乙亥": "火",
+  "丙子": "水",
+  "丁丑": "水",
+  "戊寅": "土",
+  "己卯": "土",
+  "庚辰": "金",
+  "辛巳": "金",
+  "壬午": "木",
+  "癸未": "木",
+  "甲申": "水",
+  "乙酉": "水",
+  "丙戌": "土",
+  "丁亥": "土",
+  "戊子": "火",
+  "己丑": "火",
+  "庚寅": "木",
+  "辛卯": "木",
+  "壬辰": "水",
+  "癸巳": "水",
+  "甲午": "金",
+  "乙未": "金",
+  "丙申": "火",
+  "丁酉": "火",
+  "戊戌": "木",
+  "己亥": "木",
+  "庚子": "土",
+  "辛丑": "土",
+  "壬寅": "金",
+  "癸卯": "金",
+  "甲辰": "火",
+  "乙巳": "火",
+  "丙午": "水",
+  "丁未": "水",
+  "戊申": "土",
+  "己酉": "土",
+  "庚戌": "金",
+  "辛亥": "金",
+  "壬子": "木",
+  "癸丑": "木",
+  "甲寅": "水",
+  "乙卯": "水",
+  "丙辰": "土",
+  "丁巳": "土",
+  "戊午": "火",
+  "己未": "火",
+  "庚申": "木",
+  "辛酉": "木",
+  "壬戌": "水",
+  "癸亥": "水"
+}

ziwei_chart/data/6base_mainstar_location.json

@@ -0,0 +1,171 @@
+{
+  "紫微在子": {
+    "子": ["紫微"],
+    "丑": [],
+    "寅": ["破军"],
+    "卯": [],
+    "辰": ["天府", "廉贞"],
+    "巳": ["太阴"],
+    "午": ["贪狼"],
+    "未": ["天同", "巨门"],
+    "申": ["武曲", "天相"],
+    "酉": ["天梁", "太阳"],
+    "戌": ["七杀"],
+    "亥": ["天机"]
+  },
+  "紫微在丑": {
+    "子": ["天机"],
+    "丑": ["紫微", "破军"],
+    "寅": [],
+    "卯": ["天府"],
+    "辰": ["太阴"],
+    "巳": ["廉贞","贪狼"],
+    "午": ["巨门"],
+    "未": ["天相"],
+    "申": ["天同", "天梁"],
+    "酉": ["武曲", "七杀"],
+    "戌": ["太阳"],
+    "亥": []
+  },
+  "紫微在寅": {
+    "子": ["破军"],
+    "丑": ["天机"],
+    "寅": ["紫微", "天府"],
+    "卯": ["太阴"],
+    "辰": ["贪狼"],
+    "巳": ["巨门"],
+    "午": ["廉贞", "天相"],
+    "未": ["天梁"],
+    "申": ["七杀"],
+    "酉": ["天同"],
+    "戌": ["武曲"],
+    "亥": ["太阳"]
+  },
+  "紫微在卯": {
+    "子": ["太阳"],
+    "丑": ["天府"],
+    "寅": ["天机", "太阴"],
+    "卯": ["紫微", "贪狼"],
+    "辰": ["巨门"],
+    "巳": ["天相"],
+    "午": ["天梁"],
+    "未": ["廉贞", "七杀"],
+    "申": [],
+    "酉": [],
+    "戌": ["天同"],
+    "亥": ["武曲", "破军"]
+  },
+  "紫微在辰": {
+    "子": ["武曲", "天府"],
+    "丑": ["太阳", "太阴"],
+    "寅": ["贪狼"],
+    "卯": ["天机", "巨门"],
+    "辰": ["紫微", "天相"],
+    "巳": ["天梁"],
+    "午": ["七杀"],
+    "未": [],
+    "申": ["廉贞"],
+    "酉": [],
+    "戌": ["破军"],
+    "亥": ["天同"]
+  },
+  "紫微在巳": {
+    "子": ["天同", "太阴"],
+    "丑": ["武曲", "贪狼"],
+    "寅": ["巨门", "太阳"],
+    "卯": ["天相"],
+    "辰": ["天机", "天梁"],
+    "巳": ["紫微", "七杀"],
+    "午": [],
+    "未": [],
+    "申": [],
+    "酉": ["廉贞", "破军"],
+    "戌": [],
+    "亥": ["天府"]
+  },
+  "紫微在午": {
+    "子": ["贪狼"],
+    "丑": ["天同","巨门"],
+    "寅": ["武曲","天相"],
+    "卯": ["天梁","太阳"],
+    "辰": ["七杀"],
+    "巳": ["天机"],
+    "午": ["紫微"],
+    "未": [],
+    "申": ["破军"],
+    "酉": [],
+    "戌": ["天府","廉贞"],
+    "亥": ["太阴"]
+  },
+  "紫微在未": {
+    "子": ["巨门"],
+    "丑": ["天相"],
+    "寅": ["天同", "天梁"],
+    "卯": ["武曲", "七杀"],
+    "辰": ["太阳"],
+    "巳": [],
+    "午": ["天机"],
+    "未": ["紫微", "破军"],
+    "申": [],
+    "酉": ["天府"],
+    "戌": ["太阴"],
+    "亥": ["廉贞", "贪狼"]
+  },
+  "紫微在申": {
+    "子": ["廉贞", "天相"],
+    "丑": ["天梁"],
+    "寅": ["七杀"],
+    "卯": ["天同"],
+    "辰": ["武曲"],
+    "巳": ["太阳"],
+    "午": ["破军"],
+    "未": ["天机"],
+    "申": ["紫微", "天府"],
+    "酉": ["太阴"],
+    "戌": ["贪狼"],
+    "亥": ["巨门"]
+  },
+  "紫微在酉": {
+    "子": ["天梁"],
+    "丑": ["廉贞", "七杀"],
+    "寅": [],
+    "卯": [],
+    "辰": ["天同"],
+    "巳": ["武曲", "破军"],
+    "午": ["太阳"],
+    "未": ["天府"],
+    "申": ["天机", "太阴"],
+    "酉": ["紫微", "贪狼"],
+    "戌": ["巨门"],
+    "亥": ["天相"]
+  },
+  "紫微在戌": {
+    "子": ["七杀"],
+    "丑": [],
+    "寅": ["廉贞"],
+    "卯": [],
+    "辰": ["破军"],
+    "巳": ["天同"],
+    "午": ["武曲", "天府"],
+    "未": ["太阳", "太阴"],
+    "申": ["贪狼"],
+    "酉": ["天机", "巨门"],
+    "戌": ["紫微", "天相"],
+    "亥": ["天梁"]
+  },
+  "紫微在亥": {
+    "子": [],
+    "丑": [],
+    "寅": [],
+    "卯": ["廉贞", "破军"],
+    "辰": [],
+    "巳": ["天府"],
+    "午": ["天同", "太阴"],
+    "未": ["武曲", "贪狼"],
+    "申": ["巨门", "太阳"],
+    "酉": ["天相"],
+    "戌": ["天机", "天梁"],
+    "亥": ["紫微", "七杀"]
+  }
+}
+

ziwei_chart/data/sihua_config.json

@@ -0,0 +1,62 @@
+{
+  "甲": {
+    "廉贞": "禄",
+    "破军": "权",
+    "武曲": "科",
+    "太阳": "忌"
+  },
+  "乙": {
+    "天机": "禄",
+    "天梁": "权",
+    "紫微": "科",
+    "太阴": "忌"
+  },
+  "丙": {
+    "天同": "禄",
+    "天机": "权",
+    "文昌": "科",
+    "廉贞": "忌"
+  },
+  "丁": {
+    "太阴": "禄",
+    "天同": "权",
+    "天机": "科",
+    "巨门": "忌"
+  },
+  "戊": {
+    "贪狼": "禄",
+    "太阴": "权",
+    "右弼": "科",
+    "天机": "忌"
+  },
+  "己": {
+    "武曲": "禄",
+    "贪狼": "权",
+    "天梁": "科",
+    "文曲": "忌"
+  },
+  "庚": {
+    "太阳": "禄",
+    "武曲": "权",
+    "太阴": "科",
+    "天同": "忌"
+  },
+  "辛": {
+    "巨门": "禄",
+    "太阳": "权",
+    "文曲": "科",
+    "文昌": "忌"
+  },
+  "壬": {
+    "天梁": "禄",
+    "紫微": "权",
+    "左辅": "科",
+    "武曲": "忌"
+  },
+  "癸": {
+    "破军": "禄",
+    "巨门": "权",
+    "太阴": "科",
+    "贪狼": "忌"
+  }
+}

ziwei_chart/data/wuhu_config.json

@@ -0,0 +1,12 @@
+{
+  "甲": "丙",
+  "乙": "戊",
+  "丙": "庚",
+  "丁": "壬",
+  "戊": "甲",
+  "己": "丙",
+  "庚": "戊",
+  "辛": "庚",
+  "壬": "壬",
+  "癸": "甲"
+}