239 lines
6.5 KiB
Python
239 lines
6.5 KiB
Python
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
|
||
"""选择第几个挑战者"""
|
||
|
||
when_no_set: Literal['remind', 'wait', 'auto_set', 'auto_set_silent'] = 'remind'
|
||
"""竞赛队伍未编成时应该：remind=通知我并跳过竞赛，wait=提醒我并等待手动编成，auto_set=使用自动编成并提醒，auto_set_silent=使用自动编成不提醒"""
|
||
|
||
|
||
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)
|