nonebot-plugin-werewolf 1.1.5__py3-none-any.whl → 1.1.7__py3-none-any.whl

nonebot_plugin_werewolf/__init__.py

@@ -2,6 +2,7 @@ from nonebot import require
 from nonebot.plugin import PluginMetadata, inherit_supported_adapters
 
 require("nonebot_plugin_alconna")
+require("nonebot_plugin_localstore")
 require("nonebot_plugin_uninfo")
 require("nonebot_plugin_waiter")
 
@@ -9,7 +10,7 @@ from . import matchers as matchers
 from . import players as players
 from .config import Config
 
-__version__ = "1.1.5"
+__version__ = "1.1.7"
 __plugin_meta__ = PluginMetadata(
     name="狼人杀",
     description="适用于 Nonebot2 的狼人杀插件",

nonebot_plugin_werewolf/config.py

@@ -1,82 +1,45 @@
-from typing import Any, Literal, overload
+import json
 
-from nonebot import get_plugin_config, logger
-from nonebot.compat import PYDANTIC_V2
+import nonebot
+from nonebot.compat import model_dump, type_validate_json
+from nonebot_plugin_localstore import get_plugin_data_file
 from pydantic import BaseModel, Field
 from typing_extensions import Self
 
 from .constant import (
-    Role,
-    RolePresetConfig,
-    RolePresetDict,
     default_priesthood_proirity,
     default_role_preset,
     default_werewolf_priority,
 )
+from .models import Role
 
-if PYDANTIC_V2:
-    from pydantic import model_validator as model_validator
-else:
-    from pydantic import root_validator
 
-    @overload
-    def model_validator(*, mode: Literal["before"]) -> Any: ...  # noqa: ANN401
-
-    @overload
-    def model_validator(*, mode: Literal["after"]) -> Any: ...  # noqa: ANN401
-
-    def model_validator(*, mode: Literal["before", "after"]) -> Any:
-        return root_validator(
-            pre=mode == "before",  # pyright: ignore[reportArgumentType]
-            allow_reuse=True,
-        )
-
-
-class PluginConfig(BaseModel):
-    enable_poke: bool = True
-    role_preset: RolePresetConfig = default_role_preset.copy()
+class PresetData(BaseModel):
+    role_preset: dict[int, tuple[int, int, int]] = default_role_preset.copy()
     werewolf_priority: list[Role] = default_werewolf_priority.copy()
     priesthood_proirity: list[Role] = default_priesthood_proirity.copy()
     joker_probability: float = Field(default=0.0, ge=0.0, le=1.0)
 
-    @model_validator(mode="after")
-    def _validate(self) -> Self:
-        if isinstance(self.role_preset, list):
-            for preset in self.role_preset:
-                if preset[0] != sum(preset[1:]):
-                    raise ValueError(
-                        "配置项 `role_preset` 错误: "
-                        f"预设总人数为 {preset[0]}, 实际总人数为 {sum(preset[1:])} "
-                        f"({', '.join(map(str, preset[1:]))})"
-                    )
-            self.role_preset = default_role_preset | {
-                i[0]: i[1:] for i in self.role_preset
-            }
-            logger.debug(f"覆写配置 role_preset: {self.role_preset}")
-
-        min_length = max(i[0] for i in self.role_preset.values())
-        if len(self.werewolf_priority) < min_length:
-            raise ValueError(
-                f"配置项 `werewolf_priority` 错误: 应至少为 {min_length} 项"
-            )
+    @classmethod
+    def load(cls) -> Self:
+        return type_validate_json(cls, preset_data_file.read_text())
 
-        min_length = max(i[1] for i in self.role_preset.values())
-        if len(self.priesthood_proirity) < min_length:
-            raise ValueError(
-                f"配置项 `priesthood_proirity` 错误: 应至少为 {min_length} 项"
-            )
+    def save(self) -> None:
+        preset_data_file.write_text(json.dumps(model_dump(self)))
 
-        return self
 
-    def get_role_preset(self) -> RolePresetDict:
-        if isinstance(self.role_preset, list):
-            self.role_preset = {i[0]: i[1:] for i in self.role_preset}
-        return self.role_preset
+class PluginConfig(BaseModel):
+    enable_poke: bool = True
+    enable_button: bool = False
 
 
 class Config(BaseModel):
     werewolf: PluginConfig = PluginConfig()
 
 
-config = get_plugin_config(Config).werewolf
-logger.debug(f"加载插件配置: {config}")
+preset_data_file = get_plugin_data_file("preset.json")
+if not preset_data_file.exists():
+    PresetData().save()
+
+config = nonebot.get_plugin_config(Config).werewolf
+nonebot.logger.debug(f"加载插件配置: {config}")

nonebot_plugin_werewolf/constant.py

@@ -1,14 +1,6 @@
-from __future__ import annotations
-
-import dataclasses
-from enum import Enum, auto
-from typing import TYPE_CHECKING
-
 import nonebot
 
-if TYPE_CHECKING:
-    from .players import Player
-
+from .models import GameStatus, KillReason, Role, RoleGroup
 
 COMMAND_START = next(
     iter(sorted(nonebot.get_driver().config.command_start, key=len)), ""
@@ -17,67 +9,6 @@ STOP_COMMAND_PROMPT = f"{COMMAND_START}stop"
 STOP_COMMAND = "{{stop}}"
 
 
-class Role(Enum):
-    # 狼人
-    Werewolf = 1
-    WolfKing = 2
-
-    # 神职
-    Prophet = 11
-    Witch = 12
-    Hunter = 13
-    Guard = 14
-    Idiot = 15
-
-    # 其他
-    Joker = 51
-
-    # 平民
-    Civilian = 0
-
-
-class RoleGroup(Enum):
-    Werewolf = auto()
-    GoodGuy = auto()
-    Others = auto()
-
-
-class KillReason(Enum):
-    Werewolf = auto()
-    Poison = auto()
-    Shoot = auto()
-    Vote = auto()
-
-
-class GameStatus(Enum):
-    GoodGuy = auto()
-    Werewolf = auto()
-    Joker = auto()
-
-
-@dataclasses.dataclass
-class GameState:
-    day: int
-    """当前天数记录, 不会被 `reset()` 重置"""
-    killed: Player | None = None
-    """当晚狼人击杀目标, `None` 则为空刀"""
-    shoot: Player | None = None
-    """当前执行射杀操作的玩家"""
-    antidote: set[Player] = dataclasses.field(default_factory=set)
-    """当晚女巫使用解药的目标"""
-    poison: set[Player] = dataclasses.field(default_factory=set)
-    """当晚使用了毒药的女巫"""
-    protected: set[Player] = dataclasses.field(default_factory=set)
-    """当晚守卫保护的目标"""
-
-    def reset(self) -> None:
-        self.killed = None
-        self.shoot = None
-        self.antidote = set()
-        self.poison = set()
-        self.protected = set()
-
-
 role_name_conv: dict[Role | RoleGroup, str] = {
     Role.Werewolf: "狼人",
     Role.WolfKing: "狼王",
@@ -105,10 +36,20 @@ role_emoji: dict[Role, str] = {
     Role.Civilian: "👨🏻‍🌾",
 }
 
-RolePresetDict = dict[int, tuple[int, int, int]]
-RolePresetConfig = RolePresetDict | list[tuple[int, int, int, int]]
+game_status_conv: dict[GameStatus, str] = {
+    GameStatus.GoodGuy: "好人",
+    GameStatus.Werewolf: "狼人",
+    GameStatus.Joker: "小丑",
+}
 
-default_role_preset: RolePresetDict = {
+report_text: dict[KillReason, tuple[str, str]] = {
+    KillReason.Werewolf: ("🔪", "刀了"),
+    KillReason.Poison: ("🧪", "毒死"),
+    KillReason.Shoot: ("🔫", "射杀"),
+    KillReason.Vote: ("🗳️", "票出"),
+}
+
+default_role_preset: dict[int, tuple[int, int, int]] = {
     # 总人数: (狼, 神, 民)
     6: (1, 2, 3),
     7: (2, 2, 3),
@@ -118,7 +59,6 @@ default_role_preset: RolePresetDict = {
     11: (3, 5, 3),
     12: (4, 5, 3),
 }
-
 default_werewolf_priority: list[Role] = [
     Role.Werewolf,
     Role.Werewolf,

nonebot_plugin_werewolf/exception.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from .constant import GameStatus
+    from .models import GameStatus
 
 
 class Error(Exception):