249 lines
8.7 KiB
Python
249 lines
8.7 KiB
Python
"""积分赠送系统游戏模块"""
|
||
import logging
|
||
from datetime import datetime
|
||
from games.base import BaseGame
|
||
from utils.parser import CommandParser
|
||
from core.database import get_db
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class GiftGame(BaseGame):
|
||
"""积分赠送系统游戏"""
|
||
|
||
def __init__(self):
|
||
"""初始化游戏"""
|
||
super().__init__()
|
||
self.db = get_db()
|
||
|
||
async def handle(self, command: str, chat_id: int, user_id: int) -> str:
|
||
"""处理积分赠送相关指令
|
||
|
||
Args:
|
||
command: 指令，如 ".gift 123 50 生日快乐", ".gift sent", ".gift received"
|
||
chat_id: 会话ID
|
||
user_id: 用户ID
|
||
|
||
Returns:
|
||
回复消息
|
||
"""
|
||
try:
|
||
# 提取参数
|
||
_, args = CommandParser.extract_command_args(command)
|
||
args = args.strip()
|
||
|
||
# 赠送统计
|
||
if args in ['stats', '统计']:
|
||
return self._get_gift_stats(user_id)
|
||
|
||
# 发送记录
|
||
elif args in ['sent', '发送', '送出']:
|
||
return self._get_gift_records_sent(user_id)
|
||
|
||
# 接收记录
|
||
elif args in ['received', '接收', '收到']:
|
||
return self._get_gift_records_received(user_id)
|
||
|
||
# 赠送帮助
|
||
elif args in ['help', '帮助']:
|
||
return self._get_gift_help()
|
||
|
||
# 默认：执行赠送
|
||
else:
|
||
return self._process_gift_command(args, user_id)
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理积分赠送指令错误: {e}", exc_info=True)
|
||
return f"❌ 处理指令出错: {str(e)}"
|
||
|
||
def _process_gift_command(self, args: str, sender_id: int) -> str:
|
||
"""处理赠送指令
|
||
|
||
Args:
|
||
args: 指令参数
|
||
sender_id: 发送者ID
|
||
|
||
Returns:
|
||
处理结果消息
|
||
"""
|
||
# 解析参数：.gift <receiver_id> <points> [message]
|
||
parts = args.split(maxsplit=2)
|
||
|
||
if len(parts) < 2:
|
||
return "❌ 指令格式错误！\n\n正确格式：`.gift <用户ID> <积分数量> [附赠消息]`\n\n示例：\n`.gift 123 50 生日快乐`\n`.gift 456 100`"
|
||
|
||
try:
|
||
receiver_id = int(parts[0])
|
||
points = int(parts[1])
|
||
message = parts[2] if len(parts) > 2 else None
|
||
except ValueError:
|
||
return "❌ 用户ID和积分数量必须是数字！"
|
||
|
||
# 验证参数
|
||
if points <= 0:
|
||
return "❌ 赠送积分数量必须大于0！"
|
||
|
||
if points > 1000:
|
||
return "❌ 单次赠送积分不能超过1000！"
|
||
|
||
if sender_id == receiver_id:
|
||
return "❌ 不能赠送积分给自己！"
|
||
|
||
# 执行赠送
|
||
if self.db.send_gift(sender_id, receiver_id, points, message):
|
||
# 获取更新后的积分信息
|
||
sender_points = self.db.get_user_points(sender_id)
|
||
receiver_points = self.db.get_user_points(receiver_id)
|
||
|
||
text = f"## 🎁 积分赠送成功！\n\n"
|
||
text += f"**赠送者**：用户{sender_id}\n\n"
|
||
text += f"**接收者**：用户{receiver_id}\n\n"
|
||
text += f"**赠送积分**：{points} 分\n\n"
|
||
|
||
if message:
|
||
text += f"**附赠消息**：{message}\n\n"
|
||
|
||
text += f"**赠送者剩余积分**：{sender_points['available_points']} 分\n\n"
|
||
text += f"**接收者当前积分**：{receiver_points['available_points']} 分\n\n"
|
||
text += "---\n\n"
|
||
text += "💝 感谢您的慷慨赠送！"
|
||
|
||
return text
|
||
else:
|
||
return "❌ 赠送失败！请检查积分是否足够或用户ID是否正确。"
|
||
|
||
def _get_gift_stats(self, user_id: int) -> str:
|
||
"""获取赠送统计信息
|
||
|
||
Args:
|
||
user_id: 用户ID
|
||
|
||
Returns:
|
||
统计信息消息
|
||
"""
|
||
stats = self.db.get_gift_stats(user_id)
|
||
|
||
if stats['sent_count'] == 0 and stats['received_count'] == 0:
|
||
return "📊 你还没有任何积分赠送记录哦～"
|
||
|
||
text = f"## 🎁 积分赠送统计\n\n"
|
||
text += f"**发送统计**：\n"
|
||
text += f"- 赠送次数：{stats['sent_count']} 次\n"
|
||
text += f"- 赠送积分：{stats['total_sent']} 分\n\n"
|
||
|
||
text += f"**接收统计**：\n"
|
||
text += f"- 接收次数：{stats['received_count']} 次\n"
|
||
text += f"- 接收积分：{stats['total_received']} 分\n\n"
|
||
|
||
net_gift = stats['net_gift']
|
||
if net_gift > 0:
|
||
text += f"**净收益**：+{net_gift} 分 🎉\n\n"
|
||
elif net_gift < 0:
|
||
text += f"**净收益**：{net_gift} 分 😅\n\n"
|
||
else:
|
||
text += f"**净收益**：0 分 ⚖️\n\n"
|
||
|
||
text += "---\n\n"
|
||
text += "💡 提示：使用 `.gift sent` 查看发送记录，`.gift received` 查看接收记录"
|
||
|
||
return text
|
||
|
||
def _get_gift_records_sent(self, user_id: int, limit: int = 10) -> str:
|
||
"""获取发送的赠送记录
|
||
|
||
Args:
|
||
user_id: 用户ID
|
||
limit: 限制数量
|
||
|
||
Returns:
|
||
记录信息消息
|
||
"""
|
||
records = self.db.get_gift_records_sent(user_id, limit)
|
||
|
||
if not records:
|
||
return "📝 暂无发送记录"
|
||
|
||
text = f"## 🎁 发送记录（最近 {len(records)} 条）\n\n"
|
||
|
||
for record in records:
|
||
timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M')
|
||
receiver_name = record.get('receiver_name', f"用户{record['receiver_id']}")
|
||
points = record['points']
|
||
message = record.get('message', '')
|
||
|
||
text += f"**{timestamp}** 赠送 {points} 分给 {receiver_name}\n"
|
||
if message:
|
||
text += f" 💬 {message}\n"
|
||
text += "\n"
|
||
|
||
return text
|
||
|
||
def _get_gift_records_received(self, user_id: int, limit: int = 10) -> str:
|
||
"""获取接收的赠送记录
|
||
|
||
Args:
|
||
user_id: 用户ID
|
||
limit: 限制数量
|
||
|
||
Returns:
|
||
记录信息消息
|
||
"""
|
||
records = self.db.get_gift_records_received(user_id, limit)
|
||
|
||
if not records:
|
||
return "📝 暂无接收记录"
|
||
|
||
text = f"## 🎁 接收记录（最近 {len(records)} 条）\n\n"
|
||
|
||
for record in records:
|
||
timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M')
|
||
sender_name = record.get('sender_name', f"用户{record['sender_id']}")
|
||
points = record['points']
|
||
message = record.get('message', '')
|
||
|
||
text += f"**{timestamp}** 收到 {sender_name} 的 {points} 分\n"
|
||
if message:
|
||
text += f" 💬 {message}\n"
|
||
text += "\n"
|
||
|
||
return text
|
||
|
||
def _get_gift_help(self) -> str:
|
||
"""获取赠送帮助信息
|
||
|
||
Returns:
|
||
帮助信息消息
|
||
"""
|
||
text = f"## 🎁 积分赠送系统\n\n"
|
||
text += f"### 基础用法\n"
|
||
text += f"- `.gift <用户ID> <积分数量> [附赠消息]` - 赠送积分\n"
|
||
text += f"- `.gift stats` - 查看赠送统计\n"
|
||
text += f"- `.gift sent` - 查看发送记录\n"
|
||
text += f"- `.gift received` - 查看接收记录\n"
|
||
text += f"- `.gift help` - 查看帮助\n\n"
|
||
|
||
text += f"### 赠送规则\n"
|
||
text += f"- **积分限制**：单次最多赠送1000积分\n"
|
||
text += f"- **自赠限制**：不能赠送给自己\n"
|
||
text += f"- **积分检查**：必须有足够积分才能赠送\n"
|
||
text += f"- **附赠消息**：可选，最多100字符\n\n"
|
||
|
||
text += f"### 示例\n"
|
||
text += f"```\n"
|
||
text += f".gift 123 50 生日快乐\n"
|
||
text += f".gift 456 100 感谢你的帮助\n"
|
||
text += f".gift 789 200\n"
|
||
text += f".gift stats\n"
|
||
text += f"```\n\n"
|
||
|
||
text += f"### 说明\n"
|
||
text += f"- 所有赠送都有完整记录\n"
|
||
text += f"- 赠送和接收都会在积分记录中显示\n"
|
||
text += f"- 赠送是单向的，无需对方确认\n"
|
||
|
||
return text
|
||
|
||
def get_help(self) -> str:
|
||
"""获取帮助信息"""
|
||
return self._get_gift_help()
|