liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

855 协议版本-调整完毕内容

Browse Source
This commit is contained in:
liuwei
2025-04-30 13:22:33 +08:00
parent 869bce8a18
commit 454d084715
88 changed files with 1565 additions and 7816 deletions
Download Patch File Download Diff File Expand all files Collapse all files

144
plugins/point_trade/main.py
View File

@@ -1,10 +1,9 @@
import logging
from loguru import logger
import re
from datetime import datetime
from typing import Dict, Any, List, Optional, Tuple
import xml.etree.ElementTree as ET
from db.connection import DBConnectionManager
from db.points_db import PointsDBOperator, PointSource
from message_util import MessageUtil
@@ -15,6 +14,8 @@ from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotMan
import mysql.connector.pooling
from wechat_ipad import WechatAPIClient
class PointTradePlugin(MessagePluginInterface):
"""积分交易插件"""
@@ -46,15 +47,15 @@ class PointTradePlugin(MessagePluginInterface):
def __init__(self):
super().__init__()
self.db_pool = None
self.bot: WechatAPIClient = None
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
self.LOG = logging.getLogger(f"Plugin.{self.name}")
self.LOG = logger
self.LOG.info(f"正在初始化 {self.name} 插件...")
# 保存上下文对象
self.event_system = context.get("event_system")
self.message_util: MessageUtil = context.get("message_util")
self.gbm = context.get("gbm")
self.db_manager = DBConnectionManager.get_instance()
@@ -120,7 +121,7 @@ class PointTradePlugin(MessagePluginInterface):
return command in self._commands
@plugin_stats_decorator(plugin_name="积分交易")
def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()
self.LOG.info(f"插件执行： {self.name}：{content}")
@@ -129,6 +130,7 @@ class PointTradePlugin(MessagePluginInterface):
roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm")
xml = message.get("xml", "")
self.bot: WechatAPIClient = message.get("bot")
# 检查权限
if roomid and gbm.get_group_permission(roomid, Feature.POINT_TRADE) == PermissionStatus.DISABLED:
@@ -136,21 +138,21 @@ class PointTradePlugin(MessagePluginInterface):
# 处理不同的命令
if command[0] == "我的积分":
return self._handle_my_points(message)
return await self._handle_my_points(message)
elif command[0] == "积分排行":
return self._handle_points_ranking(message)
return await self._handle_points_ranking(message)
elif command[0] == "打劫":
return self._handle_rob_points(message)
return await self._handle_rob_points(message)
elif command[0] == "保释":
return self._handle_bailout(message)
return await self._handle_bailout(message)
elif command[0] in self._commands:
return self._handle_transfer_points(message)
return await self._handle_transfer_points(message)
else:
self.message_util.send_text(f"❌未知命令！{self.command_format}",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), f"❌未知命令！{self.command_format}",
sender)
return True, "未知命令"
def _handle_transfer_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def _handle_transfer_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理积分转账命令"""
content = str(message.get("content", "")).strip()
command = content.split(" ")
@@ -160,21 +162,23 @@ class PointTradePlugin(MessagePluginInterface):
# 检查命令格式
if len(command) < 3:
self.message_util.send_text(f"❌命令格式错误！积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), f"❌命令格式错误！积分转账 积分数 @用户",
sender)
return True, "命令格式错误"
# 检查积分数是否为正整数
if not command[1].isdigit():
self.message_util.send_text(f"🈚️转账积分无效(必须为正整数!) \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender),
f"🈚️转账积分无效(必须为正整数!) \n积分转账 积分数 @用户",
sender)
return True, "积分无效"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text(f"转账失败❌\n🈚️转账人无效！ \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender),
f"转账失败❌\n🈚️转账人无效！ \n积分转账 积分数 @用户",
sender)
return True, "转账人无效"
reward_points = int(command[1])
@@ -193,13 +197,13 @@ class PointTradePlugin(MessagePluginInterface):
error_msg = result.get("error", "未知错误")
if "积分不足" in error_msg:
current_points = result.get("current_points", 0)
self.message_util.send_text(
f"❌转账失败!\n你的积分不足以进行转账！当前积分：{current_points}，你需要 {reward_points} 积分。",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender),
f"❌转账失败!\n你的积分不足以进行转账！当前积分：{current_points}，你需要 {reward_points} 积分。",
sender)
else:
self.message_util.send_text(
f"❌转账失败!\n{error_msg}",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender),
f"❌转账失败!\n{error_msg}",
sender)
return True, f"转账失败: {error_msg}"
# 获取转账后的积分信息
@@ -220,16 +224,16 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{to_user_name} 当前积分: {to_user.get('total_points', 0)}"
)
self.message_util.send_text(output, (roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), output, sender)
return True, "转账成功"
except Exception as e:
self.LOG.error(f"积分交易出错: {e}")
self.message_util.send_text(f"❌积分交易失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), f"❌积分交易失败！请稍后重试。错误: {str(e)}",
sender)
return True, f"处理出错: {str(e)}"
def _handle_my_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def _handle_my_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理查询个人积分命令"""
sender = message.get("sender")
roomid = message.get("roomid", "")
@@ -239,8 +243,9 @@ class PointTradePlugin(MessagePluginInterface):
user_points = self.points_db.get_user_points(sender, roomid)
if not user_points:
self.message_util.send_text(f"❌未找到你的积分记录！请先参与积分活动[签到，答题/t]。",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender),
f"❌未找到你的积分记录！请先参与积分活动[签到，答题/t]。",
sender)
return False, "未找到积分记录"
# 获取用户昵称
@@ -295,22 +300,22 @@ class PointTradePlugin(MessagePluginInterface):
f"{recent_txs}"
)
self.message_util.send_text(output, (roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), output, sender)
return True, "查询积分成功"
except Exception as e:
self.LOG.error(f"查询积分出错: {e}")
self.message_util.send_text(f"❌查询积分失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
await self.bot.send_text_message((roomid if roomid else sender), f"❌查询积分失败！请稍后重试。错误: {str(e)}",
sender)
return True, f"处理出错: {str(e)}"
def _handle_points_ranking(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def _handle_points_ranking(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理积分排行榜命令"""
sender = message.get("sender")
roomid = message.get("roomid", "")
if not roomid:
self.message_util.send_text("❌积分排行榜仅在群聊中可用！", sender, "")
await self.bot.send_text_message(sender, "❌积分排行榜仅在群聊中可用！", "")
return True, "非群聊环境"
try:
@@ -318,7 +323,7 @@ class PointTradePlugin(MessagePluginInterface):
ranking = self.points_db.get_points_ranking(roomid, 10)
if not ranking:
self.message_util.send_text("❌暂无积分排行数据！请先参与积分活动。", roomid, sender)
await self.bot.send_text_message(roomid, "❌暂无积分排行数据！请先参与积分活动。", sender)
return True, "无排行数据"
# 构建排行榜消息
@@ -343,12 +348,12 @@ class PointTradePlugin(MessagePluginInterface):
f"更新时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
)
self.message_util.send_text(output, roomid, sender)
await self.bot.send_text_message(roomid, output, sender)
return True, "查询排行榜成功"
except Exception as e:
self.LOG.error(f"查询积分排行榜出错: {e}")
self.message_util.send_text(f"❌查询积分排行榜失败！请稍后重试。错误: {str(e)}", roomid, sender)
await self.bot.send_text_message(roomid, f"❌查询积分排行榜失败！请稍后重试。错误: {str(e)}", sender)
return True, f"处理出错: {str(e)}"
def at_list(self, xml):
@@ -438,7 +443,7 @@ class PointTradePlugin(MessagePluginInterface):
self.LOG.error(f"更新用户积分失败: {e}")
raise
def _handle_rob_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def _handle_rob_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理打劫积分命令"""
import random
import time
@@ -451,13 +456,13 @@ class PointTradePlugin(MessagePluginInterface):
# 检查是否在群聊中
if not roomid:
self.message_util.send_text("❌打劫功能仅在群聊中可用！", sender, "")
await self.bot.send_text_message(sender, "❌打劫功能仅在群聊中可用！", "")
return True, "非群聊环境"
# 检查时间限制 - 只允许在18:00-21:00之间打劫
current_hour = datetime.now().hour
if current_hour < 18 or current_hour >= 21:
self.message_util.send_text("❌打劫功能仅在晚上18:00-21:00之间开放！请在开放时间再来。", roomid, sender)
await self.bot.send_text_message(roomid, "❌打劫功能仅在晚上18:00-21:00之间开放！请在开放时间再来。", sender)
return True, "时间限制"
# 检查是否在押
@@ -466,8 +471,10 @@ class PointTradePlugin(MessagePluginInterface):
remaining_time = prison_status['end_time'] - datetime.now()
hours = int(remaining_time.total_seconds() / 3600)
minutes = int((remaining_time.total_seconds() % 3600) / 60)
self.message_util.send_text(f"❌你正在服刑！\n剩余时间: {hours}小时{minutes}分钟\n可请求他人花费30积分保释。", roomid,
sender)
await self.bot.send_text_message(
f"❌你正在服刑！\n剩余时间: {hours}小时{minutes}分钟\n可请求他人花费30积分保释。",
roomid,
sender)
return True, "在押状态"
# 检查冷却时间
@@ -478,15 +485,16 @@ class PointTradePlugin(MessagePluginInterface):
if time_passed < self.rob_cooldown:
remaining_time = int(self.rob_cooldown - time_passed)
minutes, seconds = divmod(remaining_time, 60)
self.message_util.send_text(f"❌你最近已经打劫过了，需要冷却 {minutes}分{seconds}秒 后才能再次打劫！",
roomid, sender)
await self.bot.send_text_message(roomid,
f"❌你最近已经打劫过了，需要冷却 {minutes}分{seconds}秒 后才能再次打劫！",
sender)
return True, "冷却中"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text(f"打劫失败❌\n请指定一个打劫目标！\n打劫 @用户",
roomid, sender)
await self.bot.send_text_message(roomid, f"打劫失败❌\n请指定一个打劫目标！\n打劫 @用户",
sender)
return True, "目标无效"
target_wxid = next(iter(at_users))
@@ -494,7 +502,7 @@ class PointTradePlugin(MessagePluginInterface):
# 不能打劫自己
if target_wxid == robber_wxid:
self.message_util.send_text("❌你不能打劫自己！", roomid, sender)
await self.bot.send_text_message(roomid, "❌你不能打劫自己！", sender)
return True, "不能打劫自己"
try:
@@ -503,11 +511,11 @@ class PointTradePlugin(MessagePluginInterface):
target_points = self.points_db.get_user_points(target_wxid, roomid)
if not robber_points:
self.message_util.send_text("❌你没有积分记录，无法进行打劫！请先参与积分活动。", roomid, sender)
await self.bot.send_text_message(roomid, "❌你没有积分记录，无法进行打劫！请先参与积分活动。", sender)
return True, "打劫者无积分"
if not target_points:
self.message_util.send_text("❌目标没有积分记录，无法进行打劫！", roomid, sender)
await self.bot.send_text_message(roomid, "❌目标没有积分记录，无法进行打劫！", sender)
return True, "目标无积分"
robber_total = robber_points.get('total_points', 0)
@@ -515,11 +523,12 @@ class PointTradePlugin(MessagePluginInterface):
# 检查最低积分要求
if robber_total < self.rob_min_points:
self.message_util.send_text(f"❌你的积分不足 {self.rob_min_points} 点，无法进行打劫！", roomid, sender)
await self.bot.send_text_message(roomid, f"❌你的积分不足 {self.rob_min_points} 点，无法进行打劫！",
sender)
return True, "打劫者积分不足"
if target_total < self.rob_min_points:
self.message_util.send_text(f"❌目标积分不足 {self.rob_min_points} 点，不值得打劫！", roomid, sender)
await self.bot.send_text_message(roomid, f"❌目标积分不足 {self.rob_min_points} 点，不值得打劫！", sender)
return True, "目标积分不足"
# 获取用户昵称
@@ -575,10 +584,11 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{target_name} 当前积分: {from_user.get('total_points', 0)}"
)
self.message_util.send_text(output, roomid, sender)
await self.bot.send_text_message(output, roomid, sender)
return True, "打劫成功"
else:
self.message_util.send_text(f"❌打劫过程中出现问题：{result.get('error', '未知错误')}", roomid, sender)
await self.bot.send_text_message(roomid, f"❌打劫过程中出现问题：{result.get('error', '未知错误')}",
sender)
return True, "打劫失败"
else:
@@ -612,18 +622,20 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{robber_name} 当前积分: {from_user.get('total_points', 0)}"
)
self.message_util.send_text(output, roomid, sender)
await self.bot.send_text_message(roomid, output, sender)
return True, "打劫失败"
else:
self.message_util.send_text(f"❌处理打劫惩罚时出现问题：{result.get('error', '未知错误')}", roomid, sender)
await self.bot.send_text_message(roomid,
f"❌处理打劫惩罚时出现问题：{result.get('error', '未知错误')}",
sender)
return True, "处理惩罚失败"
except Exception as e:
self.LOG.error(f"处理打劫请求出错: {e}")
self.message_util.send_text(f"❌打劫过程中出现意外：{str(e)}", roomid, sender)
await self.bot.send_text_message(roomid, f"❌打劫过程中出现意外：{str(e)}", sender)
return True, f"处理出错: {str(e)}"
def _handle_bailout(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
async def _handle_bailout(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理保释命令"""
sender = message.get("sender")
roomid = message.get("roomid", "")
@@ -631,14 +643,14 @@ class PointTradePlugin(MessagePluginInterface):
# 检查是否在群聊中
if not roomid:
self.message_util.send_text("❌保释功能仅在群聊中可用！", sender, "")
await self.bot.send_text_message(sender, "❌保释功能仅在群聊中可用！", "")
return True, "非群聊环境"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text(f"保释失败❌\n请指定一个保释目标！\n保释 @用户",
roomid, sender)
await self.bot.send_text_message(roomid, f"保释失败❌\n请指定一个保释目标！\n保释 @用户",
sender)
return True, "目标无效"
prisoner_wxid = next(iter(at_users))
@@ -646,7 +658,7 @@ class PointTradePlugin(MessagePluginInterface):
# 不能保释自己
if prisoner_wxid == bailout_wxid:
self.message_util.send_text("❌你不能保释自己！", roomid, sender)
await self.bot.send_text_message(roomid, "❌你不能保释自己！", sender)
return True, "不能保释自己"
try:
@@ -665,13 +677,13 @@ class PointTradePlugin(MessagePluginInterface):
f"✅ 保释成功！\n"
f"👤{bailout_name} 花费30积分保释了 👤{prisoner_name}"
)
self.message_util.send_text(output, roomid, sender)
await self.bot.send_text_message(roomid, output, sender)
return True, "保释成功"
else:
self.message_util.send_text(f"❌保释失败: {message}", roomid, sender)
await self.bot.send_text_message(roomid, f"❌保释失败: {message}", sender)
return True, "保释失败"
except Exception as e:
self.LOG.error(f"处理保释请求出错: {e}")
self.message_util.send_text(f"❌保释过程中出现意外：{str(e)}", roomid, sender)
await self.bot.send_text_message(roomid, f"❌保释过程中出现意外：{str(e)}", sender)
return True, f"处理出错: {str(e)}"