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

kotonebot/kaa/config/schema.py

@@ -0,0 +1,236 @@
+from typing import TypeVar, Literal, Sequence
+from pydantic import BaseModel, ConfigDict
+
+from kotonebot import config
+from kotonebot.kaa.config.produce import ProduceSolution, ProduceSolutionManager
+from .const import (
+    ConfigEnum,
+    Priority,
+    APShopItems,
+    DailyMoneyShopItems,
+)
+
+T = TypeVar('T')
+
+class ConfigBaseModel(BaseModel):
+    model_config = ConfigDict(use_attribute_docstrings=True)
+
+class PurchaseConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用商店购买"""
+    money_enabled: bool = False
+    """是否启用金币购买"""
+    money_items: list[DailyMoneyShopItems] = []
+    """金币商店要购买的物品"""
+    money_refresh: bool = True
+    """
+    是否使用每日一次免费刷新金币商店。
+    """
+    ap_enabled: bool = False
+    """是否启用AP购买"""
+    ap_items: Sequence[Literal[0, 1, 2, 3]] = []
+    """AP商店要购买的物品"""
+
+
+class ActivityFundsConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用收取活动费"""
+
+
+class PresentsConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用收取礼物"""
+
+
+class AssignmentConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用工作"""
+
+    mini_live_reassign_enabled: bool = False
+    """是否启用重新分配 MiniLive"""
+    mini_live_duration: Literal[4, 6, 12] = 12
+    """MiniLive 工作时长"""
+
+    online_live_reassign_enabled: bool = False
+    """是否启用重新分配 OnlineLive"""
+    online_live_duration: Literal[4, 6, 12] = 12
+    """OnlineLive 工作时长"""
+
+
+class ContestConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用竞赛"""
+
+    select_which_contestant: Literal[1, 2, 3] = 1
+    """选择第几个挑战者"""
+
+
+
+class ProduceConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用培育"""
+    selected_solution_id: str | None = None
+    """选中的培育方案ID"""
+    produce_count: int = 1
+    """培育的次数。"""
+
+class MissionRewardConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用领取任务奖励"""
+
+class ClubRewardConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用领取社团奖励"""
+
+    selected_note: DailyMoneyShopItems = DailyMoneyShopItems.AnomalyNoteVisual
+    """想在社团奖励中获取到的笔记"""
+
+class UpgradeSupportCardConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用支援卡升级"""
+
+class CapsuleToysConfig(ConfigBaseModel):
+    enabled: bool = False
+    """是否启用扭蛋机"""
+
+    friend_capsule_toys_count: int = 0
+    """好友扭蛋机次数"""
+
+    sense_capsule_toys_count: int = 0
+    """感性扭蛋机次数"""
+
+    logic_capsule_toys_count: int = 0
+    """理性扭蛋机次数"""
+
+    anomaly_capsule_toys_count: int = 0
+    """非凡扭蛋机次数"""
+
+class TraceConfig(ConfigBaseModel):
+    recommend_card_detection: bool = False
+    """跟踪推荐卡检测"""
+
+class StartGameConfig(ConfigBaseModel):
+    enabled: bool = True
+    """是否启用自动启动游戏。默认为True"""
+
+    start_through_kuyo: bool = False
+    """是否通过Kuyo来启动游戏"""
+
+    game_package_name: str = 'com.bandainamcoent.idolmaster_gakuen'
+    """游戏包名"""
+
+    kuyo_package_name: str = 'org.kuyo.game'
+    """Kuyo包名"""
+    
+    disable_gakumas_localify: bool = False
+    """
+    自动检测并禁用 Gakumas Localify 汉化插件。
+    
+    （目前仅对 DMM 版有效。）
+    """
+    
+    dmm_game_path: str | None = None
+    """
+    DMM 版游戏路径。若不填写，会自动检测。
+    
+    例：`F:\\Games\\gakumas\\gakumas.exe`
+    """
+
+class EndGameConfig(ConfigBaseModel):
+    exit_kaa: bool = False
+    """退出 kaa"""
+    kill_game: bool = False
+    """关闭游戏"""
+    kill_dmm: bool = False
+    """关闭 DMMGamePlayer"""
+    kill_emulator: bool = False
+    """关闭模拟器"""
+    shutdown: bool = False
+    """关闭系统"""
+    hibernate: bool = False
+    """休眠系统"""
+    restore_gakumas_localify: bool = False
+    """
+    恢复 Gakumas Localify 汉化插件状态至启动前。通常与
+    `disable_gakumas_localify` 配对使用。
+    
+    （目前仅对 DMM 版有效。）
+    """
+
+class MiscConfig(ConfigBaseModel):
+    check_update: Literal['never', 'startup'] = 'startup'
+    """
+    检查更新时机。
+
+    * never: 从不检查更新。
+    * startup: 启动时检查更新。
+    """
+    auto_install_update: bool = True
+    """
+    是否自动安装更新。
+
+    若启用，则每次自动检查更新时若有新版本会自动安装，否则只是会提示。
+    """
+    expose_to_lan: bool = False
+    """
+    是否允许局域网访问 Web 界面。
+
+    启用后，局域网内的其他设备可以通过本机 IP 地址访问 Web 界面。
+    """
+
+class BaseConfig(ConfigBaseModel):
+    purchase: PurchaseConfig = PurchaseConfig()
+    """商店购买配置"""
+
+    activity_funds: ActivityFundsConfig = ActivityFundsConfig()
+    """活动费配置"""
+
+    presents: PresentsConfig = PresentsConfig()
+    """收取礼物配置"""
+
+    assignment: AssignmentConfig = AssignmentConfig()
+    """工作配置"""
+
+    contest: ContestConfig = ContestConfig()
+    """竞赛配置"""
+
+    produce: ProduceConfig = ProduceConfig()
+    """培育配置"""
+
+    mission_reward: MissionRewardConfig = MissionRewardConfig()
+    """领取任务奖励配置"""
+
+    club_reward: ClubRewardConfig = ClubRewardConfig()
+    """领取社团奖励配置"""
+
+    upgrade_support_card: UpgradeSupportCardConfig = UpgradeSupportCardConfig()
+    """支援卡升级配置"""
+
+    capsule_toys: CapsuleToysConfig = CapsuleToysConfig()
+    """扭蛋机配置"""
+
+    trace: TraceConfig = TraceConfig()
+    """跟踪配置"""
+
+    start_game: StartGameConfig = StartGameConfig()
+    """启动游戏配置"""
+
+    end_game: EndGameConfig = EndGameConfig()
+    """关闭游戏配置"""
+
+    misc: MiscConfig = MiscConfig()
+    """杂项配置"""
+
+
+def conf() -> BaseConfig:
+    """获取当前配置数据"""
+    c = config.to(BaseConfig).current
+    return c.options
+
+def produce_solution() -> ProduceSolution:
+    """获取当前培育方案"""
+    id = conf().produce.selected_solution_id
+    if id is None:
+        raise ValueError("No produce solution selected")
+    # TODO: 这里需要缓存，不能每次都从磁盘读取
+    return ProduceSolutionManager().read(id)

kotonebot/kaa/config/upgrade.py

@@ -0,0 +1,63 @@
+import os
+import json
+import logging
+import shutil
+from typing import Any
+
+logger = logging.getLogger(__name__)
+
+def upgrade_config() -> str | None:
+    """检查并升级 `config.json` 到最新版本。
+
+    若配置已是最新版本，则返回 ``None``；否则返回合并后的迁移提示信息。
+    """
+    # 避免循环依赖，这里再进行本地导入
+    from .migrations import MIGRATION_REGISTRY, LATEST_VERSION  # pylint: disable=import-outside-toplevel
+
+    logger.setLevel(logging.DEBUG)
+    print('1212121212')
+    config_path = "config.json"
+    if not os.path.exists(config_path):
+        logger.debug("config.json not found. Skip upgrade.")
+        return None
+
+    # 读取配置
+    with open(config_path, "r", encoding="utf-8") as f:
+        root: dict[str, Any] = json.load(f)
+
+    version: int = root.get("version", 1)
+    if version >= LATEST_VERSION:
+        logger.info("Config already at latest version (v%s).", version)
+        return None
+
+    logger.info("Start upgrading config: current v%s → target v%s", version, LATEST_VERSION)
+
+    messages: list[str] = []
+
+    # 循环依次升级
+    while version < LATEST_VERSION:
+        migrator = MIGRATION_REGISTRY.get(version)
+        if migrator is None:
+            logger.warning("No migrator registered for version v%s. Abort upgrade.", version)
+            break
+
+        # 备份文件
+        backup_path = f"config.v{version}.json"
+        shutil.copy(config_path, backup_path)
+        logger.info("Backup saved: %s", backup_path)
+
+        # 对每个 user_config 应用迁移
+        for user_cfg in root.get("user_configs", []):
+            msg = migrator(user_cfg)
+            if msg:
+                messages.append(f"v{version} → v{version+1}:\n{msg}")
+
+        # 更新版本号并写回
+        version += 1
+        root["version"] = version
+        with open(config_path, "w", encoding="utf-8") as f:
+            json.dump(root, f, ensure_ascii=False, indent=4)
+
+    logger.info("Config upgrade finished. Now at v%s", version)
+
+    return "\n---\n".join(messages) if messages else None

kotonebot/kaa/game_ui/commu_event_buttons.py

@@ -32,10 +32,13 @@ YELLOW_TARGET = (39, 81, 97)
 YELLOW_LOW = (30, 70, 90)
 YELLOW_HIGH = (45, 90, 100)
 
+ORANGE_RANGE = ((14, 178, 229), (16, 229, 255))
+
 DEFAULT_COLORS = [
     (web2cv(PINK_LOW), web2cv(PINK_HIGH)),
     (web2cv(YELLOW_LOW), web2cv(YELLOW_HIGH)),
     (web2cv(BLUE_LOW), web2cv(BLUE_HIGH)),
+    ORANGE_RANGE
 ]
 
 # 参考图片：

kotonebot/kaa/game_ui/schedule.py

@@ -6,7 +6,7 @@ from cv2.typing import MatLike
 from kotonebot.primitives import Rect
 from kotonebot import ocr, device, image, action
 from kotonebot.backend.core import HintBox
-from kotonebot.kaa.common import ProduceAction
+from kotonebot.kaa.config import ProduceAction
 from kotonebot.kaa.tasks import R
 
 logger = logging.getLogger(__name__)