ksaa 2025.7.5.0__py3-none-any.whl → 2025.7.9.0__py3-none-any.whl

kotonebot/kaa/config/migrations/_v2_to_v3.py

@@ -0,0 +1,126 @@
+"""v2 → v3 迁移脚本
+
+引入游戏解包数据后，`produce.idols` 不再使用 `PIdol` 枚举，而是直接使用
+游戏内的 idol skin id (字符串)。这里负责完成枚举到字符串的转换。
+"""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from ._idol import PIdol
+
+logger = logging.getLogger(__name__)
+
+
+# 枚举 → skin_id 映射表（复制自旧实现）。
+_PIDOL_TO_SKIN: dict[PIdol, str] = {
+    PIdol.倉本千奈_Campusmode: "i_card-skin-kcna-3-007",
+    PIdol.倉本千奈_WonderScale: "i_card-skin-kcna-3-000",
+    PIdol.倉本千奈_ようこそ初星温泉: "i_card-skin-kcna-3-005",
+    PIdol.倉本千奈_仮装狂騒曲: "i_card-skin-kcna-3-002",
+    PIdol.倉本千奈_初心: "i_card-skin-kcna-1-001",
+    PIdol.倉本千奈_学園生活: "i_card-skin-kcna-1-000",
+    PIdol.倉本千奈_日々_発見的ステップ: "i_card-skin-kcna-3-001",
+    PIdol.倉本千奈_胸を張って一歩ずつ: "i_card-skin-kcna-2-000",
+    PIdol.十王星南_Campusmode: "i_card-skin-jsna-3-002",
+    PIdol.十王星南_一番星: "i_card-skin-jsna-2-000",
+    PIdol.十王星南_学園生活: "i_card-skin-jsna-1-000",
+    PIdol.十王星南_小さな野望: "i_card-skin-jsna-3-000",
+    PIdol.姫崎莉波_clumsytrick: "i_card-skin-hrnm-3-000",
+    PIdol.姫崎莉波_私らしさのはじまり: "i_card-skin-hrnm-2-000",
+    PIdol.姫崎莉波_キミとセミブルー: "i_card-skin-hrnm-3-001",
+    PIdol.姫崎莉波_Campusmode: "i_card-skin-hrnm-3-007",
+    PIdol.姫崎莉波_LUV: "i_card-skin-hrnm-3-002",
+    PIdol.姫崎莉波_ようこそ初星温泉: "i_card-skin-hrnm-3-004",
+    PIdol.姫崎莉波_ハッピーミルフィーユ: "i_card-skin-hrnm-3-008",
+    PIdol.姫崎莉波_初心: "i_card-skin-hrnm-1-001",
+    PIdol.姫崎莉波_学園生活: "i_card-skin-hrnm-1-000",
+    PIdol.月村手毬_Lunasaymaybe: "i_card-skin-ttmr-3-000",
+    PIdol.月村手毬_一匹狼: "i_card-skin-ttmr-2-000",
+    PIdol.月村手毬_Campusmode: "i_card-skin-ttmr-3-007",
+    PIdol.月村手毬_アイヴイ: "i_card-skin-ttmr-3-001",
+    PIdol.月村手毬_初声: "i_card-skin-ttmr-1-001",
+    PIdol.月村手毬_学園生活: "i_card-skin-ttmr-1-000",
+    PIdol.月村手毬_仮装狂騒曲: "i_card-skin-ttmr-3-002",
+    PIdol.有村麻央_Fluorite: "i_card-skin-amao-3-000",
+    PIdol.有村麻央_はじまりはカッコよく: "i_card-skin-amao-2-000",
+    PIdol.有村麻央_Campusmode: "i_card-skin-amao-3-007",
+    PIdol.有村麻央_FeelJewelDream: "i_card-skin-amao-3-002",
+    PIdol.有村麻央_キミとセミブルー: "i_card-skin-amao-3-001",
+    PIdol.有村麻央_初恋: "i_card-skin-amao-1-001",
+    PIdol.有村麻央_学園生活: "i_card-skin-amao-1-000",
+    PIdol.篠泽广_コントラスト: "i_card-skin-shro-3-001",
+    PIdol.篠泽广_一番向いていないこと: "i_card-skin-shro-2-000",
+    PIdol.篠泽广_光景: "i_card-skin-shro-3-000",
+    PIdol.篠泽广_Campusmode: "i_card-skin-shro-3-007",
+    PIdol.篠泽广_仮装狂騒曲: "i_card-skin-shro-3-002",
+    PIdol.篠泽广_ハッピーミルフィーユ: "i_card-skin-shro-3-008",
+    PIdol.篠泽广_初恋: "i_card-skin-shro-1-001",
+    PIdol.篠泽广_学園生活: "i_card-skin-shro-1-000",
+    PIdol.紫云清夏_TameLieOneStep: "i_card-skin-ssmk-3-000",
+    PIdol.紫云清夏_カクシタワタシ: "i_card-skin-ssmk-3-002",
+    PIdol.紫云清夏_夢へのリスタート: "i_card-skin-ssmk-2-000",
+    PIdol.紫云清夏_Campusmode: "i_card-skin-ssmk-3-007",
+    PIdol.紫云清夏_キミとセミブルー: "i_card-skin-ssmk-3-001",
+    PIdol.紫云清夏_初恋: "i_card-skin-ssmk-1-001",
+    PIdol.紫云清夏_学園生活: "i_card-skin-ssmk-1-000",
+    PIdol.花海佑芽_WhiteNightWhiteWish: "i_card-skin-hume-3-005",
+    PIdol.花海佑芽_学園生活: "i_card-skin-hume-1-000",
+    PIdol.花海佑芽_Campusmode: "i_card-skin-hume-3-006",
+    PIdol.花海佑芽_TheRollingRiceball: "i_card-skin-hume-3-000",
+    PIdol.花海佑芽_アイドル_はじめっ: "i_card-skin-hume-2-000",
+    PIdol.花海咲季_BoomBoomPow: "i_card-skin-hski-3-001",
+    PIdol.花海咲季_Campusmode: "i_card-skin-hski-3-008",
+    PIdol.花海咲季_FightingMyWay: "i_card-skin-hski-3-000",
+    PIdol.花海咲季_わたしが一番: "i_card-skin-hski-2-000",
+    PIdol.花海咲季_冠菊: "i_card-skin-hski-3-001",
+    PIdol.花海咲季_初声: "i_card-skin-hski-1-001",
+    PIdol.花海咲季_古今東西ちょちょいのちょい: "i_card-skin-hski-3-006",
+    PIdol.花海咲季_学園生活: "i_card-skin-hski-1-000",
+    PIdol.葛城リーリヤ_一つ踏み出した先に: "i_card-skin-kllj-2-000",
+    PIdol.葛城リーリヤ_白線: "i_card-skin-kllj-3-000",
+    PIdol.葛城リーリヤ_Campusmode: "i_card-skin-kllj-3-006",
+    PIdol.葛城リーリヤ_WhiteNightWhiteWish: "i_card-skin-kllj-3-005",
+    PIdol.葛城リーリヤ_冠菊: "i_card-skin-kllj-3-001",
+    PIdol.葛城リーリヤ_初心: "i_card-skin-kllj-1-001",
+    PIdol.葛城リーリヤ_学園生活: "i_card-skin-kllj-1-000",
+    PIdol.藤田ことね_カワイイ_はじめました: "i_card-skin-fktn-2-000",
+    PIdol.藤田ことね_世界一可愛い私: "i_card-skin-fktn-3-000",
+    PIdol.藤田ことね_Campusmode: "i_card-skin-fktn-3-007",
+    PIdol.藤田ことね_YellowBigBang: "i_card-skin-fktn-3-001",
+    PIdol.藤田ことね_WhiteNightWhiteWish: "i_card-skin-fktn-3-006",
+    PIdol.藤田ことね_冠菊: "i_card-skin-fktn-3-002",
+    PIdol.藤田ことね_初声: "i_card-skin-fktn-1-001",
+    PIdol.藤田ことね_学園生活: "i_card-skin-fktn-1-000",
+}
+
+
+def migrate(user_config: dict[str, Any]) -> str | None:  # noqa: D401
+    """执行 v2→v3 迁移。"""
+    options = user_config.get("options")
+    if options is None:
+        logger.debug("No 'options' in user_config, skip v2→v3 migration.")
+        return None
+
+    produce_conf = options.get("produce", {})
+    old_idols = produce_conf.get("idols", [])
+    msg = ""
+
+    new_idols: list[str] = []
+    for idol in old_idols:
+        if isinstance(idol, int):  # 原本已是 int(PIdol)
+            try:
+                skin = _PIDOL_TO_SKIN[PIdol(idol)]
+                new_idols.append(skin)
+            except (ValueError, KeyError):
+                msg += f"未知 PIdol: {idol}\n"
+        else:
+            msg += f"旧 idol 数据格式异常: {idol}\n"
+
+    produce_conf["idols"] = new_idols
+    options["produce"] = produce_conf
+    user_config["options"] = options
+
+    return msg or None 

kotonebot/kaa/config/migrations/_v3_to_v4.py

@@ -0,0 +1,29 @@
+"""v3 -> v4 迁移脚本
+
+修正游戏包名错误。
+"""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+
+def migrate(user_config: dict[str, Any]) -> str | None:  # noqa: D401
+    """执行 v3→v4 迁移：修正错误的游戏包名。"""
+    options = user_config.get("options")
+    if options is None:
+        logger.debug("No 'options' in user_config, skip v3→v4 migration.")
+        return None
+
+    start_conf = options.get("start_game", {})
+    old_pkg = start_conf.get("game_package_name")
+    if old_pkg == "com.bandinamcoent.idolmaster_gakuen":
+        start_conf["game_package_name"] = "com.bandainamcoent.idolmaster_gakuen"
+        logger.info("Corrected game package name to com.bandainamcoent.idolmaster_gakuen")
+
+    options["start_game"] = start_conf
+    user_config["options"] = options
+
+    return None 

kotonebot/kaa/config/migrations/_v4_to_v5.py

@@ -0,0 +1,26 @@
+"""v4 -> v5 迁移脚本
+
+为 Windows 截图方式的配置统一设置 backend.type = 'dmm'。
+"""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+
+def migrate(user_config: dict[str, Any]) -> str | None:  # noqa: D401
+    """执行 v4→v5 迁移：
+
+    当截图方式为 windows / remote_windows 时，将 backend.type 统一设置为 'dmm'。
+    """
+    backend = user_config.get("backend", {})
+    impl = backend.get("screenshot_impl")
+    if impl in {"windows", "remote_windows"}:
+        logger.info("Set backend type to dmm for screenshot_impl=%s", impl)
+        backend["type"] = "dmm"
+        user_config["backend"] = backend
+
+    # v4→v5 无 options 结构更改，直接返回
+    return None 

kotonebot/kaa/config/migrations/_v5_to_v6.py

@@ -0,0 +1,134 @@
+"""v5 -> v6 迁移脚本
+
+重构培育配置：将原有的 ProduceConfig 中的培育参数迁移到新的 ProduceSolution 结构中。
+"""
+from __future__ import annotations
+
+import logging
+import os
+import json
+import uuid
+import re
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+
+def _sanitize_filename(name: str) -> str:
+    """
+    清理文件名中的非法字符
+
+    :param name: 原始名称
+    :return: 清理后的文件名
+    """
+    # 替换 \/:*?"<>| 为下划线
+    return re.sub(r'[\\/:*?"<>|]', '_', name)
+
+
+def _create_default_solution(old_produce_config: dict[str, Any]) -> tuple[dict[str, Any], str]:
+    """
+    根据旧的培育配置创建默认的培育方案
+
+    :param old_produce_config: 旧的培育配置
+    :return: (新的培育方案数据, 方案ID)
+    """
+    # 生成唯一ID
+    solution_id = uuid.uuid4().hex
+
+    # 构建培育数据
+    produce_data = {
+        "mode": old_produce_config.get("mode", "regular"),
+        "idol": old_produce_config.get("idols", [None])[0] if old_produce_config.get("idols") else None,
+        "memory_set": old_produce_config.get("memory_sets", [None])[0] if old_produce_config.get("memory_sets") else None,
+        "support_card_set": old_produce_config.get("support_card_sets", [None])[0] if old_produce_config.get("support_card_sets") else None,
+        "auto_set_memory": old_produce_config.get("auto_set_memory", False),
+        "auto_set_support_card": old_produce_config.get("auto_set_support_card", False),
+        "use_pt_boost": old_produce_config.get("use_pt_boost", False),
+        "use_note_boost": old_produce_config.get("use_note_boost", False),
+        "follow_producer": old_produce_config.get("follow_producer", False),
+        "self_study_lesson": old_produce_config.get("self_study_lesson", "dance"),
+        "prefer_lesson_ap": old_produce_config.get("prefer_lesson_ap", False),
+        "actions_order": old_produce_config.get("actions_order", [
+            "recommended", "visual", "vocal", "dance",
+            "allowance", "outing", "study", "consult", "rest"
+        ]),
+        "recommend_card_detection_mode": old_produce_config.get("recommend_card_detection_mode", "normal"),
+        "use_ap_drink": old_produce_config.get("use_ap_drink", False),
+        "skip_commu": old_produce_config.get("skip_commu", True)
+    }
+
+    # 构建方案对象
+    solution = {
+        "type": "produce_solution",
+        "id": solution_id,
+        "name": "默认方案",
+        "description": "从旧配置迁移的默认培育方案",
+        "data": produce_data
+    }
+
+    return solution, solution_id
+
+
+def _save_solution_to_file(solution: dict[str, Any]) -> None:
+    """
+    将培育方案保存到文件
+
+    :param solution: 培育方案数据
+    """
+    solutions_dir = "conf/produce"
+    os.makedirs(solutions_dir, exist_ok=True)
+
+    safe_name = _sanitize_filename(solution["name"])
+    file_path = os.path.join(solutions_dir, f"{safe_name}.json")
+
+    with open(file_path, 'w', encoding='utf-8') as f:
+        json.dump(solution, f, ensure_ascii=False, indent=4)
+
+
+def migrate(user_config: dict[str, Any]) -> str | None:  # noqa: D401
+    """执行 v5→v6 迁移：重构培育配置结构。
+
+    将原有的 ProduceConfig 中的培育参数迁移到新的 ProduceSolution 结构中。
+    """
+    options = user_config.get("options")
+    if options is None:
+        logger.debug("No 'options' in user_config, skip v5→v6 migration.")
+        return None
+
+    produce_conf = options.get("produce", {})
+    if not produce_conf:
+        logger.debug("No 'produce' config found, skip v5→v6 migration.")
+        return None
+
+    # 检查是否已经是新格式（有 selected_solution_id 字段）
+    if "selected_solution_id" in produce_conf:
+        logger.debug("Produce config already in v6 format, skip migration.")
+        return None
+
+    msg = ""
+
+    try:
+        # 创建默认培育方案
+        solution, solution_id = _create_default_solution(produce_conf)
+
+        # 保存方案到文件
+        _save_solution_to_file(solution)
+
+        # 更新配置为新格式
+        new_produce_conf = {
+            "enabled": produce_conf.get("enabled", False),
+            "selected_solution_id": solution_id,
+            "produce_count": produce_conf.get("produce_count", 1)
+        }
+
+        options["produce"] = new_produce_conf
+        user_config["options"] = options
+
+        msg = f"已将培育配置迁移到新的方案系统。默认方案已创建并保存为 '{solution['name']}'。"
+        logger.info("Successfully migrated produce config to v6 format with solution ID: %s", solution_id)
+
+    except Exception as e:
+        logger.error("Failed to migrate produce config: %s", e)
+        msg = f"培育配置迁移失败：{e}"
+
+    return msg or None

kotonebot/kaa/config/produce.py

@@ -0,0 +1,255 @@
+import os
+import json
+import uuid
+import re
+import logging
+from typing import Literal
+from pydantic import BaseModel, ConfigDict, field_serializer, field_validator
+
+from .const import ProduceAction, RecommendCardDetectionMode
+
+logger = logging.getLogger(__name__)
+
+class ConfigBaseModel(BaseModel):
+    model_config = ConfigDict(use_attribute_docstrings=True)
+
+
+class ProduceData(ConfigBaseModel):
+    mode: Literal['regular', 'pro', 'master'] = 'regular'
+    """
+    培育模式。
+    进行一次 REGULAR 培育需要 ~30min，进行一次 PRO 培育需要 ~1h（具体视设备性能而定）。
+    """
+    idol: str | None = None
+    """
+    要培育偶像的 IdolCardSkin.id。
+    """
+    memory_set: int | None = None
+    """要使用的回忆编成编号，从 1 开始。"""
+    support_card_set: int | None = None
+    """要使用的支援卡编成编号，从 1 开始。"""
+    auto_set_memory: bool = False
+    """是否自动编成回忆。此选项优先级高于回忆编成编号。"""
+    auto_set_support_card: bool = False
+    """是否自动编成支援卡。此选项优先级高于支援卡编成编号。"""
+    use_pt_boost: bool = False
+    """是否使用支援强化 Pt 提升。"""
+    use_note_boost: bool = False
+    """是否使用笔记数提升。"""
+    follow_producer: bool = False
+    """是否关注租借了支援卡的制作人。"""
+    self_study_lesson: Literal['dance', 'visual', 'vocal'] = 'dance'
+    """自习课类型。"""
+    prefer_lesson_ap: bool = False
+    """
+    优先 SP 课程。
+
+    启用后，若出现 SP 课程，则会优先执行 SP 课程，而不是推荐课程。
+    若出现多个 SP 课程，随机选择一个。
+    """
+    actions_order: list[ProduceAction] = [
+        ProduceAction.RECOMMENDED,
+        ProduceAction.VISUAL,
+        ProduceAction.VOCAL,
+        ProduceAction.DANCE,
+        ProduceAction.ALLOWANCE,
+        ProduceAction.OUTING,
+        ProduceAction.STUDY,
+        ProduceAction.CONSULT,
+        ProduceAction.REST,
+    ]
+    """
+    行动优先级
+
+    每一周的行动将会按这里设置的优先级执行。
+    """
+    recommend_card_detection_mode: RecommendCardDetectionMode = RecommendCardDetectionMode.NORMAL
+    """
+    推荐卡检测模式
+
+    严格模式下，识别速度会降低，但识别准确率会提高。
+    """
+    use_ap_drink: bool = False
+    """
+    AP 不足时自动使用 AP 饮料
+    """
+    skip_commu: bool = True
+    """检测并跳过交流"""
+
+class ProduceSolution(ConfigBaseModel):
+    """培育方案"""
+    type: Literal['produce_solution'] = 'produce_solution'
+    """方案类型标识"""
+    id: str
+    """方案唯一标识符"""
+    name: str
+    """方案名称"""
+    description: str | None = None
+    """方案描述"""
+    data: ProduceData
+    """培育数据"""
+
+
+class ProduceSolutionManager:
+    """培育方案管理器"""
+
+    SOLUTIONS_DIR = "conf/produce"
+
+    def __init__(self):
+        """初始化管理器，确保目录存在"""
+        os.makedirs(self.SOLUTIONS_DIR, exist_ok=True)
+
+    def _sanitize_filename(self, name: str) -> str:
+        """
+        清理文件名中的非法字符
+
+        :param name: 原始名称
+        :return: 清理后的文件名
+        """
+        # 替换 \/:*?"<>| 为下划线
+        return re.sub(r'[\\/:*?"<>|]', '_', name)
+
+    def _get_file_path(self, name: str) -> str:
+        """
+        根据方案名称获取文件路径
+
+        :param name: 方案名称
+        :return: 文件路径
+        """
+        safe_name = self._sanitize_filename(name)
+        return os.path.join(self.SOLUTIONS_DIR, f"{safe_name}.json")
+
+    def _find_file_path_by_id(self, id: str) -> str | None:
+        """
+        根据方案ID查找文件路径
+
+        :param id: 方案ID
+        :return: 文件路径，如果未找到则返回 None
+        """
+        if not os.path.exists(self.SOLUTIONS_DIR):
+            return None
+
+        for filename in os.listdir(self.SOLUTIONS_DIR):
+            if filename.endswith('.json'):
+                try:
+                    file_path = os.path.join(self.SOLUTIONS_DIR, filename)
+                    with open(file_path, 'r', encoding='utf-8') as f:
+                        data = json.load(f)
+                    if data.get('id') == id:
+                        return file_path
+                except Exception:
+                    continue
+        return None
+
+    def new(self, name: str) -> ProduceSolution:
+        """
+        创建新的培育方案
+
+        :param name: 方案名称
+        :return: 新创建的方案
+        """
+        solution = ProduceSolution(
+            id=uuid.uuid4().hex,
+            name=name,
+            data=ProduceData()
+        )
+        return solution
+
+    def list(self) -> list[ProduceSolution]:
+        """
+        列出所有培育方案
+
+        :return: 方案列表
+        """
+        solutions = []
+        if not os.path.exists(self.SOLUTIONS_DIR):
+            return solutions
+
+        for filename in os.listdir(self.SOLUTIONS_DIR):
+            if filename.endswith('.json'):
+                try:
+                    file_path = os.path.join(self.SOLUTIONS_DIR, filename)
+                    with open(file_path, 'r', encoding='utf-8') as f:
+                        solution = ProduceSolution.model_validate_json(f.read())
+                    solutions.append(solution)
+                    logger.info(f"Loaded produce solution from {file_path}")
+                except Exception:
+                    logger.warning(f"Failed to load produce solution from {file_path}")
+                    continue
+
+        return solutions
+
+    def delete(self, id: str) -> None:
+        """
+        删除指定ID的培育方案
+
+        :param id: 方案ID
+        """
+        file_path = self._find_file_path_by_id(id)
+        if file_path:
+            os.remove(file_path)
+
+    def save(self, id: str, solution: ProduceSolution) -> None:
+        """
+        保存培育方案
+
+        :param id: 方案ID
+        :param solution: 方案对象
+        """
+        # 确保ID一致
+        solution.id = id
+
+        # 先删除具有相同ID的旧文件（如果存在），避免名称变更时产生重复文件
+        old_file_path = self._find_file_path_by_id(id)
+        if old_file_path:
+            os.remove(old_file_path)
+
+        # 保存新文件
+        file_path = self._get_file_path(solution.name)
+        with open(file_path, 'w', encoding='utf-8') as f:
+            # 使用 model_dump 并指定 mode='json' 来正确序列化枚举
+            data = solution.model_dump(mode='json')
+            json.dump(data, f, ensure_ascii=False, indent=4)
+
+    def read(self, id: str) -> ProduceSolution:
+        """
+        读取指定ID的培育方案
+
+        :param id: 方案ID
+        :return: 方案对象
+        :raises FileNotFoundError: 当方案不存在时
+        """
+        file_path = self._find_file_path_by_id(id)
+        if not file_path:
+            raise FileNotFoundError(f"Solution with id '{id}' not found")
+
+        try:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                return ProduceSolution.model_validate_json(f.read())
+        except Exception as e:
+            raise FileNotFoundError(f"Failed to read solution with id '{id}': {e}")
+
+    def duplicate(self, id: str) -> ProduceSolution:
+        """
+        复制指定ID的培育方案
+
+        :param id: 要复制的方案ID
+        :return: 新的方案对象（具有新的ID和名称）
+        :raises FileNotFoundError: 当原方案不存在时
+        """
+        original = self.read(id)
+
+        # 生成新的ID和名称
+        new_id = uuid.uuid4().hex
+        new_name = f"{original.name} - 副本"
+
+        # 创建新的方案对象
+        new_solution = ProduceSolution(
+            type=original.type,
+            id=new_id,
+            name=new_name,
+            description=original.description,
+            data=original.data.model_copy()  # 深拷贝数据
+        )
+
+        return new_solution