fix: 赛马插件P0-P2问题修复
- P0: room_store sqlite3→aiosqlite异步化 - P0: points_service统一异常处理+轻量重试 - P0: _send_to_scope加warning日志 - P1: 积分历史记录补充source/reason字段 - P1: 赛马结算写入赔率快照(odds_snapshot) - P1: test_commands改为commands_mod间接引用(测试隔离) - P2: 马名去重统一casefold()比较
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -1,63 +1,86 @@
|
||||
from typing import Tuple
|
||||
from danding_bot.plugins.danding_points import points_api
|
||||
from .config import Config
|
||||
|
||||
|
||||
class PointsService:
|
||||
def __init__(self, config: Config):
|
||||
self.config = config
|
||||
|
||||
async def spend_bet_points(
|
||||
self, user_id: str, amount: int, reason: str = "赛马下注"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Deduct points for betting with retry."""
|
||||
success, balance = await points_api.spend_points(
|
||||
user_id, amount, "horse_race", reason
|
||||
)
|
||||
if not success:
|
||||
success, balance = await points_api.spend_points(
|
||||
user_id, amount, "horse_race", reason
|
||||
)
|
||||
return success, balance
|
||||
|
||||
async def refund_bet_points(
|
||||
self, user_id: str, amount: int, reason: str = "取消报名退还"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Refund bet points."""
|
||||
return await points_api.add_points(user_id, amount, "horse_race", reason)
|
||||
|
||||
async def payout_winnings(
|
||||
self, user_id: str, amount: int, odds: float
|
||||
) -> Tuple[bool, int]:
|
||||
"""Payout bet winnings."""
|
||||
payout = int(amount * odds)
|
||||
reason = f"下注获胜 ×{odds:.2f}"
|
||||
return await points_api.add_points(user_id, payout, "horse_race", reason)
|
||||
|
||||
async def reward_participant(self, user_id: str) -> Tuple[bool, int]:
|
||||
"""Reward race participant."""
|
||||
return await points_api.add_points(
|
||||
user_id,
|
||||
self.config.PARTICIPANT_REWARD,
|
||||
"horse_race",
|
||||
"参赛奖励",
|
||||
)
|
||||
|
||||
async def reward_champion(self, user_id: str) -> Tuple[bool, int]:
|
||||
"""Reward race champion."""
|
||||
return await points_api.add_points(
|
||||
user_id,
|
||||
self.config.CHAMPION_REWARD,
|
||||
"horse_race",
|
||||
"冠军奖励",
|
||||
)
|
||||
|
||||
async def set_points(
|
||||
self, user_id: str, amount: int, reason: str = "测试设置积分"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Set user points (for testing)."""
|
||||
return await points_api.set_points(user_id, amount, "horse_race", reason)
|
||||
|
||||
async def get_balance(self, user_id: str) -> int:
|
||||
"""Get user balance."""
|
||||
return await points_api.get_balance(user_id)
|
||||
import asyncio
|
||||
import logging
|
||||
from typing import Tuple
|
||||
from danding_bot.plugins.danding_points import points_api
|
||||
from .config import Config
|
||||
|
||||
logger = logging.getLogger("horse_racing.points")
|
||||
|
||||
MAX_RETRIES = 3
|
||||
RETRY_DELAY = 0.5
|
||||
|
||||
|
||||
class PointsService:
|
||||
def __init__(self, config: Config):
|
||||
self.config = config
|
||||
|
||||
async def _call_with_retry(self, func, *args, retries=MAX_RETRIES):
|
||||
"""Call API function with retry logic."""
|
||||
last_exc = None
|
||||
for attempt in range(retries):
|
||||
try:
|
||||
return await func(*args)
|
||||
except Exception as e:
|
||||
last_exc = e
|
||||
logger.warning(
|
||||
f"Points API call failed (attempt {attempt + 1}/{retries}): {e}"
|
||||
)
|
||||
if attempt < retries - 1:
|
||||
await asyncio.sleep(RETRY_DELAY)
|
||||
logger.error(f"Points API call failed after {retries} attempts: {last_exc}")
|
||||
raise last_exc
|
||||
|
||||
async def spend_bet_points(
|
||||
self, user_id: str, amount: int, reason: str = "赛马下注"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Deduct points for betting with retry."""
|
||||
try:
|
||||
return await self._call_with_retry(
|
||||
points_api.spend_points,
|
||||
user_id, amount, "horse_race", reason
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"spend_bet_points failed for user {user_id}: {e}")
|
||||
return False, 0
|
||||
|
||||
async def refund_bet_points(
|
||||
self, user_id: str, amount: int, reason: str = "取消报名退还"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Refund bet points."""
|
||||
return await points_api.add_points(user_id, amount, "horse_race", reason)
|
||||
|
||||
async def payout_winnings(
|
||||
self, user_id: str, amount: int, odds: float
|
||||
) -> Tuple[bool, int]:
|
||||
"""Payout bet winnings."""
|
||||
payout = int(amount * odds)
|
||||
reason = f"下注获胜 ×{odds:.2f}"
|
||||
return await points_api.add_points(user_id, payout, "horse_race", reason)
|
||||
|
||||
async def reward_participant(self, user_id: str) -> Tuple[bool, int]:
|
||||
"""Reward race participant."""
|
||||
return await points_api.add_points(
|
||||
user_id,
|
||||
self.config.PARTICIPANT_REWARD,
|
||||
"horse_race",
|
||||
"参赛奖励",
|
||||
)
|
||||
|
||||
async def reward_champion(self, user_id: str) -> Tuple[bool, int]:
|
||||
"""Reward race champion."""
|
||||
return await points_api.add_points(
|
||||
user_id,
|
||||
self.config.CHAMPION_REWARD,
|
||||
"horse_race",
|
||||
"冠军奖励",
|
||||
)
|
||||
|
||||
async def set_points(
|
||||
self, user_id: str, amount: int, reason: str = "测试设置积分"
|
||||
) -> Tuple[bool, int]:
|
||||
"""Set user points (for testing)."""
|
||||
return await points_api.set_points(user_id, amount, "horse_race", reason)
|
||||
|
||||
async def get_balance(self, user_id: str) -> int:
|
||||
"""Get user balance."""
|
||||
return await points_api.get_balance(user_id)
|
||||
|
||||
@@ -1,163 +1,192 @@
|
||||
import asyncio
|
||||
import sqlite3
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from .models import Room, RoomState, RaceResult
|
||||
from .config import Config
|
||||
|
||||
|
||||
class RoomStore:
|
||||
def __init__(self, config: Config):
|
||||
self.config = config
|
||||
self.rooms: dict[str, Room] = {}
|
||||
self._locks: dict[str, asyncio.Lock] = {}
|
||||
self.db_path = Path(config.RACE_DB_FILE)
|
||||
self.db_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
self._init_db()
|
||||
|
||||
def _init_db(self):
|
||||
"""Initialize database tables."""
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS room_snapshots (
|
||||
scope TEXT PRIMARY KEY,
|
||||
state TEXT NOT NULL,
|
||||
created_at TEXT NOT NULL,
|
||||
horses TEXT NOT NULL,
|
||||
bets TEXT NOT NULL,
|
||||
champion_name TEXT,
|
||||
tick_count INTEGER DEFAULT 0
|
||||
)
|
||||
""")
|
||||
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS race_history (
|
||||
race_id TEXT PRIMARY KEY,
|
||||
scope TEXT NOT NULL,
|
||||
champion_name TEXT NOT NULL,
|
||||
champion_owner TEXT NOT NULL,
|
||||
participants TEXT NOT NULL,
|
||||
bet_distribution TEXT NOT NULL,
|
||||
duration_ticks INTEGER NOT NULL,
|
||||
completed_at TEXT NOT NULL
|
||||
)
|
||||
""")
|
||||
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS user_horse_names (
|
||||
user_id TEXT PRIMARY KEY,
|
||||
horse_name TEXT NOT NULL
|
||||
)
|
||||
""")
|
||||
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
def get_lock(self, scope: str) -> asyncio.Lock:
|
||||
"""Get or create per-room lock."""
|
||||
if scope not in self._locks:
|
||||
self._locks[scope] = asyncio.Lock()
|
||||
return self._locks[scope]
|
||||
|
||||
def get_room(self, scope: str) -> Optional[Room]:
|
||||
"""Get room by scope."""
|
||||
return self.rooms.get(scope)
|
||||
|
||||
def create_room(self, scope: str) -> Room:
|
||||
"""Create new room."""
|
||||
room = Room(scope=scope)
|
||||
self.rooms[scope] = room
|
||||
self._save_snapshot(room)
|
||||
return room
|
||||
|
||||
def delete_room(self, scope: str):
|
||||
"""Delete room."""
|
||||
if scope in self.rooms:
|
||||
del self.rooms[scope]
|
||||
|
||||
def get_last_horse_name(self, user_id: str) -> Optional[str]:
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT horse_name FROM user_horse_names WHERE user_id = ?", (user_id,))
|
||||
row = cursor.fetchone()
|
||||
conn.close()
|
||||
return row[0] if row else None
|
||||
|
||||
def set_last_horse_name(self, user_id: str, horse_name: str):
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute(
|
||||
"INSERT OR REPLACE INTO user_horse_names (user_id, horse_name) VALUES (?, ?)",
|
||||
(user_id, horse_name),
|
||||
)
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
def _save_snapshot(self, room: Room):
|
||||
"""Save room snapshot to database."""
|
||||
import json
|
||||
|
||||
horses_json = json.dumps({
|
||||
name: {
|
||||
"owner_id": horse.owner_id,
|
||||
"name": horse.name,
|
||||
"index": horse.index,
|
||||
"position": horse.position,
|
||||
"state": horse.state.value,
|
||||
}
|
||||
for name, horse in room.horses.items()
|
||||
})
|
||||
|
||||
bets_json = json.dumps([
|
||||
{
|
||||
"user_id": bet.user_id,
|
||||
"horse_name": bet.horse_name,
|
||||
"amount": bet.amount,
|
||||
}
|
||||
for bet in room.bets
|
||||
])
|
||||
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
INSERT OR REPLACE INTO room_snapshots
|
||||
(scope, state, created_at, horses, bets, champion_name, tick_count)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
room.scope,
|
||||
room.state.value,
|
||||
room.created_at.isoformat(),
|
||||
horses_json,
|
||||
bets_json,
|
||||
room.champion_name,
|
||||
room.tick_count,
|
||||
))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
|
||||
def save_race_result(self, result: RaceResult):
|
||||
"""Save race result to history."""
|
||||
import json
|
||||
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
INSERT INTO race_history
|
||||
(race_id, scope, champion_name, champion_owner, participants, bet_distribution, duration_ticks, completed_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
result.race_id,
|
||||
result.scope,
|
||||
result.champion_name,
|
||||
result.champion_owner,
|
||||
json.dumps(result.participants),
|
||||
json.dumps(result.bet_distribution),
|
||||
result.duration_ticks,
|
||||
result.completed_at.isoformat(),
|
||||
))
|
||||
conn.commit()
|
||||
conn.close()
|
||||
import asyncio
|
||||
import aiosqlite
|
||||
import json
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from .models import Room, RoomState, RaceResult
|
||||
from .config import Config
|
||||
|
||||
|
||||
class RoomStore:
|
||||
def __init__(self, config: Config):
|
||||
self.config = config
|
||||
self.rooms: dict[str, Room] = {}
|
||||
self._locks: dict[str, asyncio.Lock] = {}
|
||||
self.db_path = Path(config.RACE_DB_FILE)
|
||||
self.db_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
self._initialized = False
|
||||
|
||||
async def _init_db(self):
|
||||
"""Initialize database tables asynchronously."""
|
||||
if self._initialized:
|
||||
return
|
||||
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS room_snapshots (
|
||||
scope TEXT PRIMARY KEY,
|
||||
state TEXT NOT NULL,
|
||||
created_at TEXT NOT NULL,
|
||||
horses TEXT NOT NULL,
|
||||
bets TEXT NOT NULL,
|
||||
champion_name TEXT,
|
||||
tick_count INTEGER DEFAULT 0
|
||||
)
|
||||
""")
|
||||
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS race_history (
|
||||
race_id TEXT PRIMARY KEY,
|
||||
scope TEXT NOT NULL,
|
||||
champion_name TEXT NOT NULL,
|
||||
champion_owner TEXT NOT NULL,
|
||||
participants TEXT NOT NULL,
|
||||
bet_distribution TEXT NOT NULL,
|
||||
duration_ticks INTEGER NOT NULL,
|
||||
completed_at TEXT NOT NULL,
|
||||
point_changes TEXT DEFAULT '{}',
|
||||
point_change_summaries TEXT DEFAULT '{}',
|
||||
odds_snapshot TEXT DEFAULT '{}'
|
||||
)
|
||||
""")
|
||||
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS user_horse_names (
|
||||
user_id TEXT PRIMARY KEY,
|
||||
horse_name TEXT NOT NULL
|
||||
)
|
||||
""")
|
||||
|
||||
# Add missing columns if they don't exist (for existing databases)
|
||||
try:
|
||||
await db.execute("SELECT point_changes FROM race_history LIMIT 1")
|
||||
except aiosqlite.OperationalError:
|
||||
await db.execute("ALTER TABLE race_history ADD COLUMN point_changes TEXT DEFAULT '{}'")
|
||||
|
||||
try:
|
||||
await db.execute("SELECT point_change_summaries FROM race_history LIMIT 1")
|
||||
except aiosqlite.OperationalError:
|
||||
await db.execute("ALTER TABLE race_history ADD COLUMN point_change_summaries TEXT DEFAULT '{}'")
|
||||
|
||||
try:
|
||||
await db.execute("SELECT odds_snapshot FROM race_history LIMIT 1")
|
||||
except aiosqlite.OperationalError:
|
||||
await db.execute("ALTER TABLE race_history ADD COLUMN odds_snapshot TEXT DEFAULT '{}'")
|
||||
|
||||
await db.commit()
|
||||
|
||||
self._initialized = True
|
||||
|
||||
async def ensure_initialized(self):
|
||||
"""Ensure database is initialized (call before any DB operation)."""
|
||||
if not self._initialized:
|
||||
await self._init_db()
|
||||
|
||||
def get_lock(self, scope: str) -> asyncio.Lock:
|
||||
"""Get or create per-room lock."""
|
||||
if scope not in self._locks:
|
||||
self._locks[scope] = asyncio.Lock()
|
||||
return self._locks[scope]
|
||||
|
||||
def get_room(self, scope: str) -> Optional[Room]:
|
||||
"""Get room by scope."""
|
||||
return self.rooms.get(scope)
|
||||
|
||||
async def create_room(self, scope: str) -> Room:
|
||||
"""Create new room."""
|
||||
room = Room(scope=scope)
|
||||
self.rooms[scope] = room
|
||||
await self._save_snapshot(room)
|
||||
return room
|
||||
|
||||
def delete_room(self, scope: str):
|
||||
"""Delete room."""
|
||||
if scope in self.rooms:
|
||||
del self.rooms[scope]
|
||||
|
||||
async def get_last_horse_name(self, user_id: str) -> Optional[str]:
|
||||
await self.ensure_initialized()
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
cursor = await db.execute(
|
||||
"SELECT horse_name FROM user_horse_names WHERE user_id = ?",
|
||||
(user_id,)
|
||||
)
|
||||
row = await cursor.fetchone()
|
||||
return row[0] if row else None
|
||||
|
||||
async def set_last_horse_name(self, user_id: str, horse_name: str):
|
||||
await self.ensure_initialized()
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute(
|
||||
"INSERT OR REPLACE INTO user_horse_names (user_id, horse_name) VALUES (?, ?)",
|
||||
(user_id, horse_name),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
async def _save_snapshot(self, room: Room):
|
||||
"""Save room snapshot to database."""
|
||||
await self.ensure_initialized()
|
||||
|
||||
horses_json = json.dumps({
|
||||
name: {
|
||||
"owner_id": horse.owner_id,
|
||||
"name": horse.name,
|
||||
"index": horse.index,
|
||||
"position": horse.position,
|
||||
"state": horse.state.value,
|
||||
}
|
||||
for name, horse in room.horses.items()
|
||||
})
|
||||
|
||||
bets_json = json.dumps([
|
||||
{
|
||||
"user_id": bet.user_id,
|
||||
"horse_name": bet.horse_name,
|
||||
"amount": bet.amount,
|
||||
}
|
||||
for bet in room.bets
|
||||
])
|
||||
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
INSERT OR REPLACE INTO room_snapshots
|
||||
(scope, state, created_at, horses, bets, champion_name, tick_count)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
room.scope,
|
||||
room.state.value,
|
||||
room.created_at.isoformat(),
|
||||
horses_json,
|
||||
bets_json,
|
||||
room.champion_name,
|
||||
room.tick_count,
|
||||
))
|
||||
await db.commit()
|
||||
|
||||
async def save_race_result(self, result: RaceResult):
|
||||
"""Save race result to history."""
|
||||
await self.ensure_initialized()
|
||||
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
INSERT INTO race_history
|
||||
(race_id, scope, champion_name, champion_owner, participants,
|
||||
bet_distribution, duration_ticks, completed_at,
|
||||
point_changes, point_change_summaries, odds_snapshot)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
""", (
|
||||
result.race_id,
|
||||
result.scope,
|
||||
result.champion_name,
|
||||
result.champion_owner,
|
||||
json.dumps(result.participants),
|
||||
json.dumps(result.bet_distribution),
|
||||
result.duration_ticks,
|
||||
result.completed_at.isoformat(),
|
||||
json.dumps(getattr(result, 'point_changes', {})),
|
||||
json.dumps(getattr(result, 'point_change_summaries', {})),
|
||||
json.dumps(getattr(result, 'odds_snapshot', {})),
|
||||
))
|
||||
await db.commit()
|
||||
|
||||
@@ -1,413 +1,413 @@
|
||||
from nonebot import on_command
|
||||
from nonebot.adapters.onebot.v11 import Bot, Event, GroupMessageEvent, PrivateMessageEvent
|
||||
|
||||
from . import plugin_config as config
|
||||
from .commands import get_scope, check_access, room_store, points_service, race_engine
|
||||
from .models import Horse, HorseState, RoomState, Bet, RaceResult
|
||||
|
||||
import asyncio
|
||||
import random
|
||||
from datetime import datetime
|
||||
import traceback
|
||||
|
||||
from . import commands as commands_mod
|
||||
|
||||
|
||||
async def check_tester(event: Event) -> bool:
|
||||
"""Check if user is a tester."""
|
||||
if not config.TEST_MODE:
|
||||
return False
|
||||
return event.user_id in config.TESTERS
|
||||
|
||||
|
||||
test_reset_points_cmd = on_command("测试重置积分", priority=5)
|
||||
|
||||
|
||||
@test_reset_points_cmd.handle()
|
||||
async def handle_test_reset_points(bot: Bot, event: Event):
|
||||
"""Reset user points to 1000 for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_reset_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
success, _ = await points_service.set_points(event.user_id, 1000, "测试重置积分")
|
||||
if success:
|
||||
await test_reset_points_cmd.finish("积分已重置为1000")
|
||||
else:
|
||||
await test_reset_points_cmd.finish("重置失败")
|
||||
|
||||
|
||||
test_set_points_cmd = on_command("测试设置积分", priority=5)
|
||||
|
||||
|
||||
@test_set_points_cmd.handle()
|
||||
async def handle_test_set_points(bot: Bot, event: Event):
|
||||
"""Set user points for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_set_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
# Get the message text and extract amount
|
||||
msg = str(event.get_message()).strip()
|
||||
# Remove command prefix
|
||||
parts = msg.split()
|
||||
|
||||
if len(parts) < 2:
|
||||
await test_set_points_cmd.finish("请使用: /测试设置积分 <金额>")
|
||||
return
|
||||
|
||||
try:
|
||||
amount = int(parts[1])
|
||||
if amount < 0:
|
||||
await test_set_points_cmd.finish("金额必须为非负数")
|
||||
return
|
||||
except ValueError:
|
||||
await test_set_points_cmd.finish("金额必须是整数")
|
||||
return
|
||||
|
||||
success, _ = await points_service.set_points(event.user_id, amount, f"测试设置积分为{amount}")
|
||||
if success:
|
||||
await test_set_points_cmd.finish(f"积分已设置为 {amount}")
|
||||
else:
|
||||
await test_set_points_cmd.finish("设置失败")
|
||||
|
||||
|
||||
test_query_points_cmd = on_command("测试查询积分", priority=5)
|
||||
|
||||
|
||||
@test_query_points_cmd.handle()
|
||||
async def handle_test_query_points(bot: Bot, event: Event):
|
||||
"""Query user points for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_query_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
balance = await points_service.get_balance(event.user_id)
|
||||
await test_query_points_cmd.finish(f"当前积分: {balance}")
|
||||
|
||||
|
||||
test_clear_room_cmd = on_command("测试清空房间", priority=5)
|
||||
|
||||
|
||||
@test_clear_room_cmd.handle()
|
||||
async def handle_test_clear_room(bot: Bot, event: Event):
|
||||
"""Clear test room."""
|
||||
if not await check_tester(event):
|
||||
await test_clear_room_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
scope = get_scope(event)
|
||||
room_store.delete_room(scope)
|
||||
await test_clear_room_cmd.finish("房间已清空")
|
||||
|
||||
|
||||
test_force_start_cmd = on_command("测试强制开赛", priority=5)
|
||||
|
||||
|
||||
@test_force_start_cmd.handle()
|
||||
async def handle_test_force_start(bot: Bot, event: Event):
|
||||
"""Force start race for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_force_start_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
await test_force_start_cmd.finish("测试强制开赛命令")
|
||||
|
||||
|
||||
def _generate_random_horse_names(count: int) -> list[str]:
|
||||
prefixes = ["赤焰", "踏雪", "追风", "流星", "疾电", "破晓", "青岚", "玄影", "星尘", "霜刃", "烈阳", "苍穹"]
|
||||
cores = ["奔", "跃", "影", "翼", "刃", "雷", "岚", "焰", "星", "雪", "风", "光"]
|
||||
suffixes = ["号", "骑", "王", "将", "卫", "客", "影", "者", "马", "军"]
|
||||
|
||||
names: set[str] = set()
|
||||
attempts = 0
|
||||
while len(names) < count and attempts < 500:
|
||||
attempts += 1
|
||||
name = f"{random.choice(prefixes)}{random.choice(cores)}{random.choice(suffixes)}"
|
||||
if len(name) > 10:
|
||||
name = name[:10]
|
||||
names.add(name)
|
||||
|
||||
while len(names) < count:
|
||||
names.add(f"测试马{len(names) + 1}")
|
||||
|
||||
return list(names)[:count]
|
||||
|
||||
|
||||
test_simulate_race_cmd = on_command("测试模拟赛马", aliases={"测试模拟"}, priority=1, block=True)
|
||||
|
||||
|
||||
class _FakeBot:
|
||||
def __init__(self):
|
||||
self.messages: list[dict] = []
|
||||
self._next_message_id = 1
|
||||
|
||||
async def send_msg(self, **kwargs):
|
||||
self.messages.append(dict(kwargs))
|
||||
message_id = self._next_message_id
|
||||
self._next_message_id += 1
|
||||
return {"message_id": message_id}
|
||||
|
||||
async def delete_msg(self, message_id: int):
|
||||
# Simply record the deletion if needed, or do nothing
|
||||
return
|
||||
|
||||
|
||||
class _InMemoryRoomStore:
|
||||
def __init__(self):
|
||||
self.rooms: dict[str, "commands_mod.Room"] = {}
|
||||
self.saved_results: list[RaceResult] = []
|
||||
|
||||
def get_room(self, scope: str):
|
||||
return self.rooms.get(scope)
|
||||
|
||||
def create_room(self, scope: str):
|
||||
room = commands_mod.Room(scope=scope)
|
||||
self.rooms[scope] = room
|
||||
return room
|
||||
|
||||
def delete_room(self, scope: str):
|
||||
if scope in self.rooms:
|
||||
del self.rooms[scope]
|
||||
|
||||
def save_race_result(self, result: RaceResult):
|
||||
self.saved_results.append(result)
|
||||
|
||||
|
||||
class _InMemoryPointsService:
|
||||
def __init__(self):
|
||||
self.calls: list[tuple[str, dict]] = []
|
||||
|
||||
async def reward_champion(self, user_id: str):
|
||||
self.calls.append(("reward_champion", {"user_id": user_id}))
|
||||
return True, 0
|
||||
|
||||
async def reward_participant(self, user_id: str):
|
||||
self.calls.append(("reward_participant", {"user_id": user_id}))
|
||||
return True, 0
|
||||
|
||||
async def payout_winnings(self, user_id: str, amount: int, odds: float):
|
||||
self.calls.append(("payout_winnings", {"user_id": user_id, "amount": amount, "odds": odds}))
|
||||
return True, 0
|
||||
|
||||
async def refund_bet_points(self, user_id: str, amount: int, reason: str = "比赛中断退还"):
|
||||
self.calls.append(("refund_bet_points", {"user_id": user_id, "amount": amount, "reason": reason}))
|
||||
return True, 0
|
||||
|
||||
async def get_balance(self, user_id: str) -> int:
|
||||
self.calls.append(("get_balance", {"user_id": user_id}))
|
||||
return 8888
|
||||
|
||||
|
||||
class _NoopMessageService:
|
||||
def __init__(self):
|
||||
self.last_messages: dict[str, dict[str, str]] = {}
|
||||
|
||||
def clear_pending_recalls(self, scope: str):
|
||||
if scope in self.last_messages:
|
||||
del self.last_messages[scope]
|
||||
|
||||
async def send_with_recall(self, bot, scope, message_type, message):
|
||||
# Support basic recall for race_update to avoid flooding during simulation
|
||||
if message_type == "race_update":
|
||||
await self.recall_previous_of_type(bot, scope, "race_update")
|
||||
|
||||
is_group = scope.startswith("group_")
|
||||
result = await bot.send_msg(
|
||||
message_type="group" if is_group else "private",
|
||||
group_id=int(scope.split("_", 1)[1]) if is_group else None,
|
||||
user_id=int(scope.split("_", 1)[1]) if not is_group else None,
|
||||
message=message,
|
||||
)
|
||||
|
||||
if scope not in self.last_messages:
|
||||
self.last_messages[scope] = {}
|
||||
|
||||
if isinstance(result, dict) and "message_id" in result:
|
||||
self.last_messages[scope][message_type] = result["message_id"]
|
||||
|
||||
return "fake_msg_id"
|
||||
|
||||
async def recall_previous_of_type(self, bot, scope, message_type):
|
||||
if scope in self.last_messages and message_type in self.last_messages[scope]:
|
||||
msg_id = self.last_messages[scope][message_type]
|
||||
try:
|
||||
await bot.delete_msg(message_id=msg_id)
|
||||
except Exception:
|
||||
pass
|
||||
del self.last_messages[scope][message_type]
|
||||
|
||||
|
||||
@test_simulate_race_cmd.handle()
|
||||
async def handle_test_simulate_race(bot: Bot, event: Event):
|
||||
if not await check_tester(event):
|
||||
await test_simulate_race_cmd.send("权限不足")
|
||||
return
|
||||
|
||||
await test_simulate_race_cmd.send("收到：测试模拟赛马，开始执行完全模拟（无真实积分/数据库副作用）")
|
||||
|
||||
raw_msg = str(event.get_message()).strip()
|
||||
stream_progress = ("展示" in raw_msg) or ("实时" in raw_msg) or ("慢" in raw_msg)
|
||||
|
||||
scope = get_scope(event)
|
||||
try:
|
||||
race_engine.stop_race(scope)
|
||||
room_store.delete_room(scope)
|
||||
except Exception:
|
||||
pass
|
||||
original_room_store = commands_mod.room_store
|
||||
original_points_service = commands_mod.points_service
|
||||
original_message_service = commands_mod.message_service
|
||||
original_tick_interval = commands_mod.config.RACE_TICK_INTERVAL
|
||||
original_stop_race = commands_mod.race_engine.stop_race
|
||||
original_send_to_scope = commands_mod._send_to_scope
|
||||
|
||||
|
||||
fake_room_store = _InMemoryRoomStore()
|
||||
fake_points_service = _InMemoryPointsService()
|
||||
fake_message_service = _NoopMessageService()
|
||||
fake_bot = _FakeBot()
|
||||
|
||||
start_task: asyncio.Task | None = None
|
||||
room = None
|
||||
|
||||
try:
|
||||
await test_simulate_race_cmd.send("阶段：构建沙盒环境与参赛数据")
|
||||
commands_mod.room_store = fake_room_store
|
||||
commands_mod.points_service = fake_points_service
|
||||
commands_mod.message_service = fake_message_service
|
||||
commands_mod.config.RACE_TICK_INTERVAL = 1 if stream_progress else 0
|
||||
commands_mod.race_engine.stop_race = lambda _scope: commands_mod.race_engine.active_tasks.pop(_scope, None)
|
||||
|
||||
progress_count = 0
|
||||
max_progress = 30
|
||||
|
||||
async def _test_send_to_scope(_bot: Bot, _scope: str, message: str, *args, **kwargs):
|
||||
nonlocal progress_count
|
||||
await fake_bot.send_msg(message_type="private", user_id=event.user_id, message=message)
|
||||
if not stream_progress:
|
||||
return
|
||||
|
||||
if message.startswith("【第") and "回合】" in message:
|
||||
progress_count += 1
|
||||
if progress_count > max_progress:
|
||||
return
|
||||
await original_send_to_scope(bot, scope, message, *args, **kwargs)
|
||||
|
||||
commands_mod._send_to_scope = _test_send_to_scope
|
||||
|
||||
room = fake_room_store.create_room(scope)
|
||||
horse_names = _generate_random_horse_names(8)
|
||||
for idx, horse_name in enumerate(horse_names, start=1):
|
||||
owner_id = f"sim_user_{idx}"
|
||||
room.horses[horse_name] = Horse(owner_id=owner_id, name=horse_name, index=idx, state=HorseState.RACING)
|
||||
room.next_horse_index = len(horse_names) + 1
|
||||
|
||||
bet_amount = max(commands_mod.config.MIN_BET, 10)
|
||||
room.bets.append(Bet(user_id="bettor_1", horse_name=horse_names[0], amount=bet_amount))
|
||||
room.bets.append(Bet(user_id="bettor_2", horse_name=horse_names[1], amount=bet_amount * 2))
|
||||
|
||||
room.state = RoomState.WAITING
|
||||
|
||||
for horse in room.horses.values():
|
||||
horse.state = HorseState.RACING
|
||||
|
||||
await test_simulate_race_cmd.send("阶段：执行赛程（后台任务）")
|
||||
start_task = asyncio.create_task(commands_mod.run_race_with_settlement(fake_bot, room, scope))
|
||||
commands_mod.race_engine.register_task(scope, start_task)
|
||||
await asyncio.wait_for(start_task, timeout=180 if stream_progress else 30)
|
||||
|
||||
messages = [str(m.get("message", "")) for m in fake_bot.messages]
|
||||
if not messages:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未捕获到任何消息")
|
||||
return
|
||||
|
||||
if not any("比赛开始！" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送开赛消息")
|
||||
return
|
||||
|
||||
# Look for the start message to verify horse names
|
||||
start_msg = next((msg for msg in messages if "比赛开始！" in msg), "")
|
||||
for idx, horse_name in enumerate(horse_names, start=1):
|
||||
if f"{idx:02d}号 {horse_name}" not in start_msg:
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：开赛名单中未找到 {idx:02d}号 {horse_name}")
|
||||
return
|
||||
|
||||
progress_messages = [msg for msg in messages if "【第" in msg and "回合】" in msg]
|
||||
if not progress_messages:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送回合进度消息")
|
||||
return
|
||||
|
||||
# Check first progress message format
|
||||
progress_lines = [line for line in progress_messages[0].splitlines() if "|" in line]
|
||||
if len(progress_lines) != len(horse_names):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：回合进度展示条目数量不匹配")
|
||||
return
|
||||
|
||||
if not any("比赛结束！冠军：" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送结束结算消息")
|
||||
return
|
||||
|
||||
if not any("积分变化：" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送积分变化总结")
|
||||
return
|
||||
|
||||
if not fake_room_store.saved_results:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未写入赛史结果（内存）")
|
||||
return
|
||||
|
||||
saved = fake_room_store.saved_results[-1]
|
||||
if saved.champion_name not in room.horses:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史冠军不在参赛马匹中")
|
||||
return
|
||||
if not saved.point_changes:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史未记录积分变化")
|
||||
return
|
||||
if not saved.point_change_summaries:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史未记录积分变化总结")
|
||||
return
|
||||
|
||||
champion_owner_id = room.horses[saved.champion_name].owner_id
|
||||
reward_champion_calls = [c for c in fake_points_service.calls if c[0] == "reward_champion"]
|
||||
if not reward_champion_calls or reward_champion_calls[0][1]["user_id"] != champion_owner_id:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未正确发放冠军奖励（内存记录）")
|
||||
return
|
||||
|
||||
participant_calls = [c for c in fake_points_service.calls if c[0] == "reward_participant"]
|
||||
if len(participant_calls) != len(room.horses):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：参赛奖励次数不匹配（内存记录）")
|
||||
return
|
||||
|
||||
await test_simulate_race_cmd.send(
|
||||
"\n".join(
|
||||
[
|
||||
"完全模拟赛马完成（无真实积分/数据库副作用）",
|
||||
f"参赛马匹：{', '.join(horse_names)}",
|
||||
f"冠军：{saved.champion_name}（马主：{saved.champion_owner}）",
|
||||
f"总回合：{saved.duration_ticks}",
|
||||
f"消息条数：{len(messages)}（开赛/进度/结算均已覆盖）",
|
||||
f"结算调用：{len(fake_points_service.calls)}（冠军/参赛/下注派奖）",
|
||||
f"积分变化用户数：{len(saved.point_changes)}",
|
||||
f"过程展示：{'开启' if stream_progress else '关闭'}",
|
||||
]
|
||||
)
|
||||
)
|
||||
return
|
||||
except asyncio.TimeoutError:
|
||||
ticks = room.tick_count if room else 0
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：超时未完成（当前回合：{ticks}）")
|
||||
except asyncio.CancelledError:
|
||||
ticks = room.tick_count if room else 0
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：任务被取消（当前回合：{ticks}）")
|
||||
except Exception as e:
|
||||
tail = "\n".join(traceback.format_exc().splitlines()[-8:])
|
||||
await test_simulate_race_cmd.send(f"完全模拟异常：{type(e).__name__}: {e}\n{tail}")
|
||||
finally:
|
||||
if start_task and not start_task.done():
|
||||
start_task.cancel()
|
||||
commands_mod.room_store = original_room_store
|
||||
commands_mod.points_service = original_points_service
|
||||
commands_mod.message_service = original_message_service
|
||||
commands_mod.config.RACE_TICK_INTERVAL = original_tick_interval
|
||||
commands_mod.race_engine.stop_race = original_stop_race
|
||||
commands_mod._send_to_scope = original_send_to_scope
|
||||
from nonebot import on_command
|
||||
from nonebot.adapters.onebot.v11 import Bot, Event, GroupMessageEvent, PrivateMessageEvent
|
||||
|
||||
from . import plugin_config as config
|
||||
from .commands import get_scope, check_access, race_engine
|
||||
from .models import Horse, HorseState, RoomState, Bet, RaceResult
|
||||
|
||||
import asyncio
|
||||
import random
|
||||
from datetime import datetime
|
||||
import traceback
|
||||
|
||||
from . import commands as commands_mod
|
||||
|
||||
|
||||
async def check_tester(event: Event) -> bool:
|
||||
"""Check if user is a tester."""
|
||||
if not config.TEST_MODE:
|
||||
return False
|
||||
return event.user_id in config.TESTERS
|
||||
|
||||
|
||||
test_reset_points_cmd = on_command("测试重置积分", priority=5)
|
||||
|
||||
|
||||
@test_reset_points_cmd.handle()
|
||||
async def handle_test_reset_points(bot: Bot, event: Event):
|
||||
"""Reset user points to 1000 for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_reset_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
success, _ = await commands_mod.points_service.set_points(event.user_id, 1000, "测试重置积分")
|
||||
if success:
|
||||
await test_reset_points_cmd.finish("积分已重置为1000")
|
||||
else:
|
||||
await test_reset_points_cmd.finish("重置失败")
|
||||
|
||||
|
||||
test_set_points_cmd = on_command("测试设置积分", priority=5)
|
||||
|
||||
|
||||
@test_set_points_cmd.handle()
|
||||
async def handle_test_set_points(bot: Bot, event: Event):
|
||||
"""Set user points for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_set_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
# Get the message text and extract amount
|
||||
msg = str(event.get_message()).strip()
|
||||
# Remove command prefix
|
||||
parts = msg.split()
|
||||
|
||||
if len(parts) < 2:
|
||||
await test_set_points_cmd.finish("请使用: /测试设置积分 <金额>")
|
||||
return
|
||||
|
||||
try:
|
||||
amount = int(parts[1])
|
||||
if amount < 0:
|
||||
await test_set_points_cmd.finish("金额必须为非负数")
|
||||
return
|
||||
except ValueError:
|
||||
await test_set_points_cmd.finish("金额必须是整数")
|
||||
return
|
||||
|
||||
success, _ = await commands_mod.points_service.set_points(event.user_id, amount, f"测试设置积分为{amount}")
|
||||
if success:
|
||||
await test_set_points_cmd.finish(f"积分已设置为 {amount}")
|
||||
else:
|
||||
await test_set_points_cmd.finish("设置失败")
|
||||
|
||||
|
||||
test_query_points_cmd = on_command("测试查询积分", priority=5)
|
||||
|
||||
|
||||
@test_query_points_cmd.handle()
|
||||
async def handle_test_query_points(bot: Bot, event: Event):
|
||||
"""Query user points for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_query_points_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
balance = await commands_mod.points_service.get_balance(event.user_id)
|
||||
await test_query_points_cmd.finish(f"当前积分: {balance}")
|
||||
|
||||
|
||||
test_clear_room_cmd = on_command("测试清空房间", priority=5)
|
||||
|
||||
|
||||
@test_clear_room_cmd.handle()
|
||||
async def handle_test_clear_room(bot: Bot, event: Event):
|
||||
"""Clear test room."""
|
||||
if not await check_tester(event):
|
||||
await test_clear_room_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
scope = get_scope(event)
|
||||
commands_mod.room_store.delete_room(scope)
|
||||
await test_clear_room_cmd.finish("房间已清空")
|
||||
|
||||
|
||||
test_force_start_cmd = on_command("测试强制开赛", priority=5)
|
||||
|
||||
|
||||
@test_force_start_cmd.handle()
|
||||
async def handle_test_force_start(bot: Bot, event: Event):
|
||||
"""Force start race for testing."""
|
||||
if not await check_tester(event):
|
||||
await test_force_start_cmd.finish("权限不足")
|
||||
return
|
||||
|
||||
await test_force_start_cmd.finish("测试强制开赛命令")
|
||||
|
||||
|
||||
def _generate_random_horse_names(count: int) -> list[str]:
|
||||
prefixes = ["赤焰", "踏雪", "追风", "流星", "疾电", "破晓", "青岚", "玄影", "星尘", "霜刃", "烈阳", "苍穹"]
|
||||
cores = ["奔", "跃", "影", "翼", "刃", "雷", "岚", "焰", "星", "雪", "风", "光"]
|
||||
suffixes = ["号", "骑", "王", "将", "卫", "客", "影", "者", "马", "军"]
|
||||
|
||||
names: set[str] = set()
|
||||
attempts = 0
|
||||
while len(names) < count and attempts < 500:
|
||||
attempts += 1
|
||||
name = f"{random.choice(prefixes)}{random.choice(cores)}{random.choice(suffixes)}"
|
||||
if len(name) > 10:
|
||||
name = name[:10]
|
||||
names.add(name)
|
||||
|
||||
while len(names) < count:
|
||||
names.add(f"测试马{len(names) + 1}")
|
||||
|
||||
return list(names)[:count]
|
||||
|
||||
|
||||
test_simulate_race_cmd = on_command("测试模拟赛马", aliases={"测试模拟"}, priority=1, block=True)
|
||||
|
||||
|
||||
class _FakeBot:
|
||||
def __init__(self):
|
||||
self.messages: list[dict] = []
|
||||
self._next_message_id = 1
|
||||
|
||||
async def send_msg(self, **kwargs):
|
||||
self.messages.append(dict(kwargs))
|
||||
message_id = self._next_message_id
|
||||
self._next_message_id += 1
|
||||
return {"message_id": message_id}
|
||||
|
||||
async def delete_msg(self, message_id: int):
|
||||
# Simply record the deletion if needed, or do nothing
|
||||
return
|
||||
|
||||
|
||||
class _InMemoryRoomStore:
|
||||
def __init__(self):
|
||||
self.rooms: dict[str, "commands_mod.Room"] = {}
|
||||
self.saved_results: list[RaceResult] = []
|
||||
|
||||
def get_room(self, scope: str):
|
||||
return self.rooms.get(scope)
|
||||
|
||||
def create_room(self, scope: str):
|
||||
room = commands_mod.Room(scope=scope)
|
||||
self.rooms[scope] = room
|
||||
return room
|
||||
|
||||
def delete_room(self, scope: str):
|
||||
if scope in self.rooms:
|
||||
del self.rooms[scope]
|
||||
|
||||
def save_race_result(self, result: RaceResult):
|
||||
self.saved_results.append(result)
|
||||
|
||||
|
||||
class _InMemoryPointsService:
|
||||
def __init__(self):
|
||||
self.calls: list[tuple[str, dict]] = []
|
||||
|
||||
async def reward_champion(self, user_id: str):
|
||||
self.calls.append(("reward_champion", {"user_id": user_id}))
|
||||
return True, 0
|
||||
|
||||
async def reward_participant(self, user_id: str):
|
||||
self.calls.append(("reward_participant", {"user_id": user_id}))
|
||||
return True, 0
|
||||
|
||||
async def payout_winnings(self, user_id: str, amount: int, odds: float):
|
||||
self.calls.append(("payout_winnings", {"user_id": user_id, "amount": amount, "odds": odds}))
|
||||
return True, 0
|
||||
|
||||
async def refund_bet_points(self, user_id: str, amount: int, reason: str = "比赛中断退还"):
|
||||
self.calls.append(("refund_bet_points", {"user_id": user_id, "amount": amount, "reason": reason}))
|
||||
return True, 0
|
||||
|
||||
async def get_balance(self, user_id: str) -> int:
|
||||
self.calls.append(("get_balance", {"user_id": user_id}))
|
||||
return 8888
|
||||
|
||||
|
||||
class _NoopMessageService:
|
||||
def __init__(self):
|
||||
self.last_messages: dict[str, dict[str, str]] = {}
|
||||
|
||||
def clear_pending_recalls(self, scope: str):
|
||||
if scope in self.last_messages:
|
||||
del self.last_messages[scope]
|
||||
|
||||
async def send_with_recall(self, bot, scope, message_type, message):
|
||||
# Support basic recall for race_update to avoid flooding during simulation
|
||||
if message_type == "race_update":
|
||||
await self.recall_previous_of_type(bot, scope, "race_update")
|
||||
|
||||
is_group = scope.startswith("group_")
|
||||
result = await bot.send_msg(
|
||||
message_type="group" if is_group else "private",
|
||||
group_id=int(scope.split("_", 1)[1]) if is_group else None,
|
||||
user_id=int(scope.split("_", 1)[1]) if not is_group else None,
|
||||
message=message,
|
||||
)
|
||||
|
||||
if scope not in self.last_messages:
|
||||
self.last_messages[scope] = {}
|
||||
|
||||
if isinstance(result, dict) and "message_id" in result:
|
||||
self.last_messages[scope][message_type] = result["message_id"]
|
||||
|
||||
return "fake_msg_id"
|
||||
|
||||
async def recall_previous_of_type(self, bot, scope, message_type):
|
||||
if scope in self.last_messages and message_type in self.last_messages[scope]:
|
||||
msg_id = self.last_messages[scope][message_type]
|
||||
try:
|
||||
await bot.delete_msg(message_id=msg_id)
|
||||
except Exception:
|
||||
pass
|
||||
del self.last_messages[scope][message_type]
|
||||
|
||||
|
||||
@test_simulate_race_cmd.handle()
|
||||
async def handle_test_simulate_race(bot: Bot, event: Event):
|
||||
if not await check_tester(event):
|
||||
await test_simulate_race_cmd.send("权限不足")
|
||||
return
|
||||
|
||||
await test_simulate_race_cmd.send("收到：测试模拟赛马，开始执行完全模拟（无真实积分/数据库副作用）")
|
||||
|
||||
raw_msg = str(event.get_message()).strip()
|
||||
stream_progress = ("展示" in raw_msg) or ("实时" in raw_msg) or ("慢" in raw_msg)
|
||||
|
||||
scope = get_scope(event)
|
||||
try:
|
||||
race_engine.stop_race(scope)
|
||||
await commands_mod.room_store.delete_room(scope)
|
||||
except Exception:
|
||||
pass
|
||||
original_room_store = commands_mod.room_store
|
||||
original_points_service = commands_mod.points_service
|
||||
original_message_service = commands_mod.message_service
|
||||
original_tick_interval = commands_mod.config.RACE_TICK_INTERVAL
|
||||
original_stop_race = commands_mod.race_engine.stop_race
|
||||
original_send_to_scope = commands_mod._send_to_scope
|
||||
|
||||
|
||||
fake_room_store = _InMemoryRoomStore()
|
||||
fake_points_service = _InMemoryPointsService()
|
||||
fake_message_service = _NoopMessageService()
|
||||
fake_bot = _FakeBot()
|
||||
|
||||
start_task: asyncio.Task | None = None
|
||||
room = None
|
||||
|
||||
try:
|
||||
await test_simulate_race_cmd.send("阶段：构建沙盒环境与参赛数据")
|
||||
commands_mod.room_store = fake_room_store
|
||||
commands_mod.points_service = fake_points_service
|
||||
commands_mod.message_service = fake_message_service
|
||||
commands_mod.config.RACE_TICK_INTERVAL = 1 if stream_progress else 0
|
||||
commands_mod.race_engine.stop_race = lambda _scope: commands_mod.race_engine.active_tasks.pop(_scope, None)
|
||||
|
||||
progress_count = 0
|
||||
max_progress = 30
|
||||
|
||||
async def _test_send_to_scope(_bot: Bot, _scope: str, message: str, *args, **kwargs):
|
||||
nonlocal progress_count
|
||||
await fake_bot.send_msg(message_type="private", user_id=event.user_id, message=message)
|
||||
if not stream_progress:
|
||||
return
|
||||
|
||||
if message.startswith("【第") and "回合】" in message:
|
||||
progress_count += 1
|
||||
if progress_count > max_progress:
|
||||
return
|
||||
await original_send_to_scope(bot, scope, message, *args, **kwargs)
|
||||
|
||||
commands_mod._send_to_scope = _test_send_to_scope
|
||||
|
||||
room = fake_room_store.create_room(scope)
|
||||
horse_names = _generate_random_horse_names(8)
|
||||
for idx, horse_name in enumerate(horse_names, start=1):
|
||||
owner_id = f"sim_user_{idx}"
|
||||
room.horses[horse_name] = Horse(owner_id=owner_id, name=horse_name, index=idx, state=HorseState.RACING)
|
||||
room.next_horse_index = len(horse_names) + 1
|
||||
|
||||
bet_amount = max(commands_mod.config.MIN_BET, 10)
|
||||
room.bets.append(Bet(user_id="bettor_1", horse_name=horse_names[0], amount=bet_amount))
|
||||
room.bets.append(Bet(user_id="bettor_2", horse_name=horse_names[1], amount=bet_amount * 2))
|
||||
|
||||
room.state = RoomState.WAITING
|
||||
|
||||
for horse in room.horses.values():
|
||||
horse.state = HorseState.RACING
|
||||
|
||||
await test_simulate_race_cmd.send("阶段：执行赛程（后台任务）")
|
||||
start_task = asyncio.create_task(commands_mod.run_race_with_settlement(fake_bot, room, scope))
|
||||
commands_mod.race_engine.register_task(scope, start_task)
|
||||
await asyncio.wait_for(start_task, timeout=180 if stream_progress else 30)
|
||||
|
||||
messages = [str(m.get("message", "")) for m in fake_bot.messages]
|
||||
if not messages:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未捕获到任何消息")
|
||||
return
|
||||
|
||||
if not any("比赛开始！" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送开赛消息")
|
||||
return
|
||||
|
||||
# Look for the start message to verify horse names
|
||||
start_msg = next((msg for msg in messages if "比赛开始！" in msg), "")
|
||||
for idx, horse_name in enumerate(horse_names, start=1):
|
||||
if f"{idx:02d}号 {horse_name}" not in start_msg:
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：开赛名单中未找到 {idx:02d}号 {horse_name}")
|
||||
return
|
||||
|
||||
progress_messages = [msg for msg in messages if "【第" in msg and "回合】" in msg]
|
||||
if not progress_messages:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送回合进度消息")
|
||||
return
|
||||
|
||||
# Check first progress message format
|
||||
progress_lines = [line for line in progress_messages[0].splitlines() if "|" in line]
|
||||
if len(progress_lines) != len(horse_names):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：回合进度展示条目数量不匹配")
|
||||
return
|
||||
|
||||
if not any("比赛结束！冠军：" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送结束结算消息")
|
||||
return
|
||||
|
||||
if not any("积分变化：" in msg for msg in messages):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未发送积分变化总结")
|
||||
return
|
||||
|
||||
if not fake_room_store.saved_results:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未写入赛史结果（内存）")
|
||||
return
|
||||
|
||||
saved = fake_room_store.saved_results[-1]
|
||||
if saved.champion_name not in room.horses:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史冠军不在参赛马匹中")
|
||||
return
|
||||
if not saved.point_changes:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史未记录积分变化")
|
||||
return
|
||||
if not saved.point_change_summaries:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：赛史未记录积分变化总结")
|
||||
return
|
||||
|
||||
champion_owner_id = room.horses[saved.champion_name].owner_id
|
||||
reward_champion_calls = [c for c in fake_points_service.calls if c[0] == "reward_champion"]
|
||||
if not reward_champion_calls or reward_champion_calls[0][1]["user_id"] != champion_owner_id:
|
||||
await test_simulate_race_cmd.send("完全模拟失败：未正确发放冠军奖励（内存记录）")
|
||||
return
|
||||
|
||||
participant_calls = [c for c in fake_points_service.calls if c[0] == "reward_participant"]
|
||||
if len(participant_calls) != len(room.horses):
|
||||
await test_simulate_race_cmd.send("完全模拟失败：参赛奖励次数不匹配（内存记录）")
|
||||
return
|
||||
|
||||
await test_simulate_race_cmd.send(
|
||||
"\n".join(
|
||||
[
|
||||
"完全模拟赛马完成（无真实积分/数据库副作用）",
|
||||
f"参赛马匹：{', '.join(horse_names)}",
|
||||
f"冠军：{saved.champion_name}（马主：{saved.champion_owner}）",
|
||||
f"总回合：{saved.duration_ticks}",
|
||||
f"消息条数：{len(messages)}（开赛/进度/结算均已覆盖）",
|
||||
f"结算调用：{len(fake_points_service.calls)}（冠军/参赛/下注派奖）",
|
||||
f"积分变化用户数：{len(saved.point_changes)}",
|
||||
f"过程展示：{'开启' if stream_progress else '关闭'}",
|
||||
]
|
||||
)
|
||||
)
|
||||
return
|
||||
except asyncio.TimeoutError:
|
||||
ticks = room.tick_count if room else 0
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：超时未完成（当前回合：{ticks}）")
|
||||
except asyncio.CancelledError:
|
||||
ticks = room.tick_count if room else 0
|
||||
await test_simulate_race_cmd.send(f"完全模拟失败：任务被取消（当前回合：{ticks}）")
|
||||
except Exception as e:
|
||||
tail = "\n".join(traceback.format_exc().splitlines()[-8:])
|
||||
await test_simulate_race_cmd.send(f"完全模拟异常：{type(e).__name__}: {e}\n{tail}")
|
||||
finally:
|
||||
if start_task and not start_task.done():
|
||||
start_task.cancel()
|
||||
commands_mod.room_store = original_room_store
|
||||
commands_mod.points_service = original_points_service
|
||||
commands_mod.message_service = original_message_service
|
||||
commands_mod.config.RACE_TICK_INTERVAL = original_tick_interval
|
||||
commands_mod.race_engine.stop_race = original_stop_race
|
||||
commands_mod._send_to_scope = original_send_to_scope
|
||||
|
||||
Reference in New Issue
Block a user