Files
DanDingNoneBot/danding_bot/plugins/group_horse_racing/config.py
Mr.Xia e773a7a1ef feat: 赛马游戏调整 - 允许自马下注 & 取消非冠军参赛奖励
- 移除下注时不能给自己马匹下注的限制
- 非冠军马主不再获得参赛奖励积分(PARTICIPANT_REWARD=0)
- 同步更新帮助文本、README文档和config默认值
2026-04-05 21:25:10 +08:00

75 lines
2.2 KiB
Python

from pydantic import Field, field_validator
from pydantic_settings import BaseSettings, SettingsConfigDict
import json
class Config(BaseSettings):
model_config = SettingsConfigDict(
extra="ignore",
env_prefix="GROUP_HORSE_RACING_",
)
# 测试模式配置
TEST_MODE: bool = False
TESTERS: set[int] = Field(default_factory=set)
TEST_GROUPS: set[int] = Field(default_factory=set)
ALLOWED_GROUPS: set[int] = Field(default_factory=set)
# 奖励配置
PARTICIPANT_REWARD: int = 0
CHAMPION_REWARD: int = 200
MIN_BET: int = 10
MIN_ODDS: float = 1.2
RACE_DISTANCE: int = 100
RACE_TICK_INTERVAL: int = 5
RACE_RENDER_AS_IMAGE: bool = True
RACE_IMAGE_WIDTH: int = 900
RACE_IMAGE_FONT_SIZE: int = 26
RACE_IMAGE_PADDING: int = 28
RACE_IMAGE_LINE_SPACING: float = 1.35
# 消息撤回配置
MESSAGE_RECALL: dict[str, int] = Field(
default_factory=lambda: {
"race_update": 30,
"registration": 180,
"bet_confirm": 180,
"cancel_confirm": 60,
"error": 60,
"race_result": 0,
"leaderboard": 0,
"help": 0,
"odds_display": 0,
}
)
# 数据库配置
RACE_DB_FILE: str = "data/group_horse_racing/race.db"
@field_validator("TESTERS", "TEST_GROUPS", "ALLOWED_GROUPS", mode="before")
@classmethod
def parse_id_sets(cls, v):
"""Parse ID sets from various formats."""
if isinstance(v, set):
return v
if isinstance(v, str):
return cls._parse_id_set(v)
if isinstance(v, (list, tuple)):
return set(int(x) for x in v)
return v if isinstance(v, set) else set()
@staticmethod
def _parse_id_set(v: str) -> set[int]:
"""Parse ID sets from various formats."""
try:
parsed = json.loads(v)
if isinstance(parsed, list):
return set(int(x) for x in parsed)
except (json.JSONDecodeError, ValueError, TypeError):
pass
try:
return set(int(x.strip()) for x in v.split(",") if x.strip())
except ValueError:
pass
return set()