Files
web-page-backend/saju-lab/app/calculator/lucky.py

85 lines
2.8 KiB
Python

"""오늘의 럭키 컬러/숫자/방향 + 행운/위험 알림."""
from datetime import date as _date
from typing import Dict, List
import sxtwl
from .constants import HEAVENLY_STEMS, EARTHLY_BRANCHES, FIVE_ELEMENTS, KE_CYCLE
LUCKY_COLOR_BY_ELEMENT: Dict[str, List[str]] = {
"": ["청록", "녹색"],
"": ["빨강", "주황"],
"": ["황색", "베이지"],
"": ["흰색", "은색"],
"": ["파랑", "검정"],
}
LUCKY_DIRECTION_BY_ELEMENT: Dict[str, str] = {
"": "동쪽",
"": "남쪽",
"": "중앙",
"": "서쪽",
"": "북쪽",
}
def _day_stem_branch_idx(target: _date) -> tuple[int, int]:
"""양력 → sxtwl 일주 천간/지지 인덱스."""
day_obj = sxtwl.fromSolar(target.year, target.month, target.day)
try:
gz = day_obj.getDayGZ()
return gz.tg, gz.dz
except AttributeError:
return day_obj.getDayTG(), day_obj.getDayDZ()
def calculate_lucky(saju: dict, analysis: dict, target_date: _date) -> dict:
"""오늘의 럭키 정보."""
yongshin = analysis["yong_shin"]["yong_shin"]
color = LUCKY_COLOR_BY_ELEMENT.get(yongshin, ["흰색"])
direction = LUCKY_DIRECTION_BY_ELEMENT.get(yongshin, "중앙")
# 럭키 숫자: 오늘 일진 천간+지지 인덱스 합 % 9 + 1 → 1~9
day_stem_idx, day_branch_idx = _day_stem_branch_idx(target_date)
number = (day_stem_idx + day_branch_idx) % 9 + 1
# 행운/위험
good_signs: List[str] = []
warnings: List[str] = []
today_stem = HEAVENLY_STEMS[day_stem_idx]
today_branch = EARTHLY_BRANCHES[day_branch_idx]
day_master = saju["day_stem"]
day_elem = FIVE_ELEMENTS[day_master]
today_elem = FIVE_ELEMENTS[today_stem]
if KE_CYCLE.get(day_elem) == today_elem:
good_signs.append("재물 기회가 다가옵니다")
if KE_CYCLE.get(today_elem) == day_elem:
warnings.append("강한 압박이 있을 수 있어요")
LIU_CHONG = {
frozenset(["", ""]), frozenset(["", ""]),
frozenset(["", ""]), frozenset(["", ""]),
frozenset(["", ""]), frozenset(["", ""]),
}
LIU_HE = {
frozenset(["", ""]), frozenset(["", ""]),
frozenset(["", ""]), frozenset(["", ""]),
frozenset(["", ""]), frozenset(["", ""]),
}
day_branch = saju["day"]["branch"]
if frozenset([day_branch, today_branch]) in LIU_CHONG:
warnings.append("대인 갈등에 주의하세요")
if frozenset([day_branch, today_branch]) in LIU_HE:
good_signs.append("좋은 인연을 만날 수 있어요")
return {
"color": color,
"number": number,
"direction": direction,
"good_signs": good_signs,
"warnings": warnings,
}