"""
祝福语模板 - 场合对应的祝福语
"""
from typing import Dict, List
import random

class BlessingGenerator:
    """祝福语生成器"""

    # 场合祝福语库
    BLESSINGS = {
        "生日": [
            "生日快乐！愿你年年有今日，岁岁有今朝！",
            "祝生日快乐，心想事成，万事如意！",
            "生日快乐！愿你每一天都充满阳光和欢笑！",
        ],
        "情人节": [
            "情人节快乐！愿我们的爱情永远甜蜜！",
            "有你的每一天都是情人节，爱你！",
            "愿得一人心，白首不分离。情人节快乐！",
        ],
        "纪念日": [
            "纪念日快乐！感谢有你相伴！",
            "时光飞逝，爱意永存。纪念日快乐！",
            "每一个纪念日都是我们爱的见证！",
        ],
        "母亲节": [
            "妈妈，您辛苦了！母亲节快乐！",
            "世上只有妈妈好，母亲节快乐！",
            "感谢您的养育之恩，妈妈节日快乐！",
        ],
        "父亲节": [
            "爸爸，您是我的榜样！父亲节快乐！",
            "父爱如山，感恩有您！节日快乐！",
            "祝爸爸身体健康，节日快乐！",
        ],
        "教师节": [
            "老师，您辛苦了！教师节快乐！",
            "桃李满天下，师恩如海深。教师节快乐！",
            "感谢您的教导，祝教师节快乐！",
        ],
        "春节": [
            "新春快乐，万事如意！",
            "恭喜发财，红包拿来！新年快乐！",
            "愿新年带来好运和幸福！春节快乐！",
        ],
        "中秋": [
            "月圆人团圆，中秋节快乐！",
            "但愿人长久，千里共婵娟。中秋快乐！",
            "花好月圆，中秋快乐！",
        ],
        "乔迁": [
            "乔迁之喜，新家新气象！",
            "恭喜搬新家，愿你生活更美好！",
            "新居落成，乔迁大吉！",
        ],
        "结婚": [
            "百年好合，永结同心！",
            "祝新婚快乐，早生贵子！",
            "愿你们的爱情甜蜜如初，幸福美满！",
        ],
        "默认": [
            "祝你开心每一天！",
            "愿你心想事成，万事顺意！",
            "送上最真挚的祝福！",
        ],
    }

    def generate(self, occasion: str = None) -> str:
        """
        生成祝福语

        Args:
            occasion: 场合

        Returns:
            祝福语文本
        """
        # 查找场合对应的祝福语
        for key, blessings in self.BLESSINGS.items():
            if occasion and key in occasion:
                return random.choice(blessings)

        # 默认祝福语
        return random.choice(self.BLESSINGS["默认"])
