nonebot-plugin-werewolf 1.1.6__py3-none-any.whl → 1.1.8__py3-none-any.whl

nonebot_plugin_werewolf/__init__.py

@@ -10,7 +10,7 @@ from . import matchers as matchers
 from . import players as players
 from .config import Config
 
-__version__ = "1.1.6"
+__version__ = "1.1.8"
 __plugin_meta__ = PluginMetadata(
     name="狼人杀",
     description="适用于 Nonebot2 的狼人杀插件",

nonebot_plugin_werewolf/config.py

@@ -1,45 +1,103 @@
 import json
+from pathlib import Path
+from typing import Any, ClassVar, Final
+from typing_extensions import Self
 
-from nonebot import get_plugin_config, logger
+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 (
-    default_priesthood_proirity,
-    default_role_preset,
-    default_werewolf_priority,
+    DEFAULT_PRIESTHOOD_PRIORITY,
+    DEFAULT_ROLE_PRESET,
+    DEFAULT_WEREWOLF_PRIORITY,
+    stop_command_prompt,
 )
 from .models import Role
 
 
-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)
+class ConfigFile(BaseModel):
+    FILE: ClassVar[Path]
+    _cache: ClassVar[Self | None] = None
+
+    def __init_subclass__(cls, **kwargs: Any) -> None:  # noqa: ANN401
+        super().__init_subclass__(**kwargs)
+        if not cls.FILE.exists():
+            cls().save()
 
     @classmethod
     def load(cls) -> Self:
-        return type_validate_json(cls, preset_data_file.read_text())
+        return type_validate_json(cls, cls.FILE.read_text())
+
+    @classmethod
+    def get(cls, *, use_cache: bool = True) -> Self:
+        if cls._cache is None or not use_cache:
+            cls._cache = cls.load()
+        return cls._cache
 
     def save(self) -> None:
-        preset_data_file.write_text(json.dumps(model_dump(self)))
+        self.FILE.write_text(json.dumps(model_dump(self)))
+        type(self)._cache = self  # noqa: SLF001
+
+
+class PresetData(ConfigFile):
+    FILE: ClassVar[Path] = get_plugin_data_file("preset.json")
+
+    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_PRIORITY.copy()
+    jester_probability: float = Field(default=0.0, ge=0.0, le=1.0)
+
+
+class GameBehavior(ConfigFile):
+    FILE: ClassVar[Path] = get_plugin_data_file("behavior.json")
+
+    show_roles_list_on_start: bool = False
+    speak_in_turn: bool = False
+    dead_channel_rate_limit: int = 8  # per minute
+
+    class _Timeout(BaseModel):
+        prepare: int = Field(default=5 * 60, ge=5 * 60)
+        speak: int = Field(default=60, ge=60)
+        group_speak: int = Field(default=120, ge=120)
+        interact: int = Field(default=60, ge=60)
+        vote: int = Field(default=60, ge=60)
+        werewolf: int = Field(default=120, ge=120)
+
+        @property
+        def speak_timeout_prompt(self) -> str:
+            return (
+                f"限时{self.speak / 60:.1f}分钟, "
+                f"发送 “{stop_command_prompt()}” 结束发言"
+            )
+
+        @property
+        def group_speak_timeout_prompt(self) -> str:
+            return (
+                f"限时{self.group_speak / 60:.1f}分钟, "
+                f"全员发送 “{stop_command_prompt()}” 结束发言"
+            )
+
+    timeout: Final[_Timeout] = _Timeout()
 
 
 class PluginConfig(BaseModel):
     enable_poke: bool = True
     enable_button: bool = False
+    stop_command: str | set[str] = "stop"
+
+    def get_stop_command(self) -> list[str]:
+        return (
+            [self.stop_command]
+            if isinstance(self.stop_command, str)
+            else sorted(self.stop_command, key=len)
+        )
 
 
 class Config(BaseModel):
     werewolf: PluginConfig = PluginConfig()
 
 
-preset_data_file = get_plugin_data_file("preset.json")
-if not preset_data_file.exists():
-    PresetData().save()
-
-config = get_plugin_config(Config).werewolf
-logger.debug(f"加载插件配置: {config}")
+config = nonebot.get_plugin_config(Config).werewolf
+nonebot.logger.debug(f"加载插件配置: {config}")

nonebot_plugin_werewolf/constant.py

@@ -1,55 +1,63 @@
+import functools
+
 import nonebot
 
 from .models import GameStatus, KillReason, Role, RoleGroup
 
+STOP_COMMAND = "{{stop}}"
 COMMAND_START = next(
     iter(sorted(nonebot.get_driver().config.command_start, key=len)), ""
 )
-STOP_COMMAND_PROMPT = f"{COMMAND_START}stop"
-STOP_COMMAND = "{{stop}}"
 
 
-role_name_conv: dict[Role | RoleGroup, str] = {
-    Role.Werewolf: "狼人",
-    Role.WolfKing: "狼王",
-    Role.Prophet: "预言家",
-    Role.Witch: "女巫",
-    Role.Hunter: "猎人",
-    Role.Guard: "守卫",
-    Role.Idiot: "白痴",
-    Role.Joker: "小丑",
-    Role.Civilian: "平民",
-    RoleGroup.Werewolf: "狼人",
-    RoleGroup.GoodGuy: "好人",
-    RoleGroup.Others: "其他",
+@functools.cache
+def stop_command_prompt() -> str:
+    from .config import config  # circular import
+
+    return COMMAND_START + config.get_stop_command()[0]
+
+
+ROLE_NAME_CONV: dict[Role | RoleGroup, str] = {
+    Role.WEREWOLF: "狼人",
+    Role.WOLFKING: "狼王",
+    Role.PROPHET: "预言家",
+    Role.WITCH: "女巫",
+    Role.HUNTER: "猎人",
+    Role.GUARD: "守卫",
+    Role.IDIOT: "白痴",
+    Role.JESTER: "小丑",
+    Role.CIVILIAN: "平民",
+    RoleGroup.WEREWOLF: "狼人",
+    RoleGroup.GOODGUY: "好人",
+    RoleGroup.OTHERS: "其他",
 }
 
-role_emoji: dict[Role, str] = {
-    Role.Werewolf: "🐺",
-    Role.WolfKing: "🐺👑",
-    Role.Prophet: "🔮",
-    Role.Witch: "🧙‍♀️",
-    Role.Hunter: "🕵️",
-    Role.Guard: "🛡️",
-    Role.Idiot: "👨🏻‍🦲",
-    Role.Joker: "🤡",
-    Role.Civilian: "👨🏻‍🌾",
+ROLE_EMOJI: dict[Role, str] = {
+    Role.WEREWOLF: "🐺",
+    Role.WOLFKING: "🐺👑",
+    Role.PROPHET: "🔮",
+    Role.WITCH: "🧙‍♀️",
+    Role.HUNTER: "🕵️",
+    Role.GUARD: "🛡️",
+    Role.IDIOT: "👨🏻‍🦲",
+    Role.JESTER: "🤡",
+    Role.CIVILIAN: "👨🏻‍🌾",
 }
 
-game_status_conv: dict[GameStatus, str] = {
-    GameStatus.GoodGuy: "好人",
-    GameStatus.Werewolf: "狼人",
-    GameStatus.Joker: "小丑",
+GAME_STATUS_CONV: dict[GameStatus, str] = {
+    GameStatus.GOODGUY: "好人",
+    GameStatus.WEREWOLF: "狼人",
+    GameStatus.JESTER: "小丑",
 }
 
-report_text: dict[KillReason, tuple[str, str]] = {
-    KillReason.Werewolf: ("🔪", "刀了"),
-    KillReason.Poison: ("🧪", "毒死"),
-    KillReason.Shoot: ("🔫", "射杀"),
-    KillReason.Vote: ("🗳️", "票出"),
+REPORT_TEXT: dict[KillReason, tuple[str, str]] = {
+    KillReason.WEREWOLF: ("🔪", "刀了"),
+    KillReason.POISON: ("🧪", "毒死"),
+    KillReason.SHOOT: ("🔫", "射杀"),
+    KillReason.VOTE: ("🗳️", "票出"),
 }
 
-default_role_preset: dict[int, tuple[int, int, int]] = {
+DEFAULT_ROLE_PRESET: dict[int, tuple[int, int, int]] = {
     # 总人数: (狼, 神, 民)
     6: (1, 2, 3),
     7: (2, 2, 3),
@@ -59,16 +67,16 @@ default_role_preset: dict[int, tuple[int, int, int]] = {
     11: (3, 5, 3),
     12: (4, 5, 3),
 }
-default_werewolf_priority: list[Role] = [
-    Role.Werewolf,
-    Role.Werewolf,
-    Role.WolfKing,
-    Role.Werewolf,
+DEFAULT_WEREWOLF_PRIORITY: list[Role] = [
+    Role.WEREWOLF,
+    Role.WEREWOLF,
+    Role.WOLFKING,
+    Role.WEREWOLF,
 ]
-default_priesthood_proirity: list[Role] = [
-    Role.Witch,
-    Role.Prophet,
-    Role.Hunter,
-    Role.Guard,
-    Role.Idiot,
+DEFAULT_PRIESTHOOD_PRIORITY: list[Role] = [
+    Role.WITCH,
+    Role.PROPHET,
+    Role.HUNTER,
+    Role.GUARD,
+    Role.IDIOT,
 ]

nonebot_plugin_werewolf/exception.py

@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
@@ -13,7 +11,7 @@ class Error(Exception):
 class GameFinished(Error):  # noqa: N818
     """游戏结束时抛出，无视游戏进程进入结算"""
 
-    status: GameStatus
+    status: "GameStatus"
 
-    def __init__(self, status: GameStatus) -> None:
+    def __init__(self, status: "GameStatus") -> None:
         self.status = status