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

调整 message_util 的发送text方法名，方便替换wcf.send_text。加入了虚拟群组管理功能；

Browse Source
This commit is contained in:
liuwei
2025-04-18 12:05:16 +08:00
parent e2c1375e54
commit 5b0a703b46
18 changed files with 1173 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/beautyleg/main.py
View File

@@ -110,8 +110,8 @@ class BeautyLegPlugin(MessagePluginInterface):
# 获取随机图片
random_file_path = self._get_random_file_from_dir(self.image_folder)
if not random_file_path:
self.message_util.send_text_msg(f"\n❌未找到美腿图片资源",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"\n❌未找到美腿图片资源",
(roomid if roomid else sender), sender)
return True, "未找到图片资源"
# 发送图片

22
plugins/dify/main.py
View File

@@ -134,7 +134,7 @@ class DifyPlugin(MessagePluginInterface):
query = re.sub(r"@.*?[\u2005|\s]", "", content).strip()
if not query:
self.message_util.send_text_msg("请在@我的同时提供问题内容", roomid, sender)
self.message_util.send_text("请在@我的同时提供问题内容", roomid, sender)
return False, "没有提供问题内容"
# self.message_util.send_text_msg("⏳AI 正在加油，请稍候… 😊", roomid, sender)
@@ -153,15 +153,15 @@ class DifyPlugin(MessagePluginInterface):
self.message_util.send_file(response, roomid)
else:
# 如果是普通文本，使用发送文本方法
self.message_util.send_text_msg(response, roomid, sender)
self.message_util.send_text(response, roomid, sender)
return True, "发送成功"
else:
self.message_util.send_text_msg("❌未能获取到回复，请稍后再试", roomid, sender)
self.message_util.send_text("❌未能获取到回复，请稍后再试", roomid, sender)
return False, "未获取到回复"
except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}")
self.message_util.send_text_msg(f"❌请求出错：{str(e)}", roomid, sender)
self.message_util.send_text(f"❌请求出错：{str(e)}", roomid, sender)
return False, f"处理出错: {e}"
# 原有的命令处理逻辑
@@ -170,8 +170,8 @@ class DifyPlugin(MessagePluginInterface):
# 检查命令格式
if len(parts) < 2 or not parts[1].strip():
self.message_util.send_text_msg(f"{self.command_format}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"{self.command_format}",
(roomid if roomid else sender), sender)
return False, "命令格式错误"
# 检查权限
@@ -201,17 +201,17 @@ class DifyPlugin(MessagePluginInterface):
self.message_util.send_file(response, (roomid if roomid else sender))
else:
# 如果是普通文本，使用发送文本方法
self.message_util.send_text_msg(response, (roomid if roomid else sender), sender if roomid else "")
self.message_util.send_text(response, (roomid if roomid else sender), sender if roomid else "")
return True, "发送成功"
else:
self.message_util.send_text_msg("❌未能获取到回复，请稍后再试", (roomid if roomid else sender),
sender if roomid else "")
self.message_util.send_text("❌未能获取到回复，请稍后再试", (roomid if roomid else sender),
sender if roomid else "")
return False, "未获取到回复"
except Exception as e:
self.LOG.error(f"处理Dify聊天请求出错: {e}")
self.message_util.send_text_msg(f"❌请求出错：{str(e)}", (roomid if roomid else sender),
sender if roomid else "")
self.message_util.send_text(f"❌请求出错：{str(e)}", (roomid if roomid else sender),
sender if roomid else "")
return False, f"处理出错: {e}"
def _chat_with_dify(self, session_id: str, user_id: str, query: str) -> Optional[str]:

58
plugins/game_task/main.py
View File

@@ -148,8 +148,8 @@ class GameTaskPlugin(MessagePluginInterface):
self._handle_list_uncompleted_tasks(sender, roomid)
return True, "列举未完成任务成功"
else:
self.message_util.send_text_msg(f"❌未知命令！\n{self.command_format}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌未知命令！\n{self.command_format}",
(roomid if roomid else sender), sender)
return False, "未知命令"
except Exception as e:
@@ -162,7 +162,7 @@ class GameTaskPlugin(MessagePluginInterface):
# 检查并添加群聊
if not self.encyclopedia_db.check_group_exists(roomid):
self.encyclopedia_db.add_group(roomid)
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 群 {roomid} 已就位，准备开燥！",
(roomid if roomid else sender),
sender
@@ -172,7 +172,7 @@ class GameTaskPlugin(MessagePluginInterface):
player = self.encyclopedia_db.get_player(sender, roomid)
if not player:
self.encyclopedia_db.add_player(sender, roomid, wx_nick_name)
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 哇塞，{wx_nick_name} 你来啦！\n"
f"🌟 群 {roomid} 瞬间燃爆！\n"
f"🎈 快来接任务，秀翻全场！指令： /t",
@@ -181,7 +181,7 @@ class GameTaskPlugin(MessagePluginInterface):
)
except Exception as e:
self.LOG.error(f"加入游戏出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 加入游戏出错，请稍后再试！",
(roomid if roomid else sender),
sender
@@ -198,7 +198,7 @@ class GameTaskPlugin(MessagePluginInterface):
# 检查并添加群聊
if not self.encyclopedia_db.check_group_exists(roomid):
self.encyclopedia_db.add_group(roomid)
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 群 {roomid} 已就位，准备开燥！",
(roomid if roomid else sender),
sender
@@ -210,7 +210,7 @@ class GameTaskPlugin(MessagePluginInterface):
# 添加当前用户为玩家
self.encyclopedia_db.add_player(sender, roomid, wx_nick_name)
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 哇塞，{wx_nick_name} 你是第一个玩家！\n"
f"🌟 已自动为你加入游戏！\n"
f"🎈 现在就为你准备题目...",
@@ -223,7 +223,7 @@ class GameTaskPlugin(MessagePluginInterface):
player_dict = {p['player_id']: p['player_name'] for p in players}
if sender not in player_dict:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😅 嘿，你谁啊？\n"
f"🌟 先用 /s 报名，不然没法玩哦！",
(roomid if roomid else sender),
@@ -245,14 +245,14 @@ class GameTaskPlugin(MessagePluginInterface):
)
if not active_task_id:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 任务创建失败，请稍后再试！",
(roomid if roomid else sender),
sender
)
return
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎁 {player_dict[sender]}，你的专属任务闪亮登场！\n"
f"🎀 任务ID: {active_task_id}\n"
f"🎈 问题：[{category}]{question}\n"
@@ -263,7 +263,7 @@ class GameTaskPlugin(MessagePluginInterface):
)
except Exception as e:
self.LOG.error(f"获取任务出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 获取任务出错，请稍后再试！",
(roomid if roomid else sender),
sender
@@ -278,7 +278,7 @@ class GameTaskPlugin(MessagePluginInterface):
roomid = message.get("roomid", "")
parts = content.split(" ", 2)
if len(parts) < 3:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😅 喂，格式不对啊！\n"
f"🌟 正确姿势：/a [任务ID] [答案]\n"
f"🎈 比如：/a 1 钒",
@@ -293,7 +293,7 @@ class GameTaskPlugin(MessagePluginInterface):
# 获取玩家信息
player = self.encyclopedia_db.get_player(sender, roomid)
if not player:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😅 嘿，你是路人甲吗？\n"
f"🌟 用 /s 先加入群 {roomid} 吧！",
(roomid if roomid else sender),
@@ -304,7 +304,7 @@ class GameTaskPlugin(MessagePluginInterface):
player_name = player['player_name']
if not task_id.isdigit():
self.message_util.send_text_msg(
self.message_util.send_text(
f"😅 喂，任务ID得是数字好吗？\n"
f"🌟 比如：1\n"
f"🎈 别瞎搞，重新来！",
@@ -319,7 +319,7 @@ class GameTaskPlugin(MessagePluginInterface):
task_data = self.encyclopedia_db.get_task_by_id(roomid, active_task_id)
if not task_data:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 哎哟，任务 task_{active_task_id} 不翼而飞啦！\n"
f"🌼 可能被别人抢先一步咯！",
(roomid if roomid else sender),
@@ -328,7 +328,7 @@ class GameTaskPlugin(MessagePluginInterface):
return False, "0"
if task_data['status'] == 'completed':
self.message_util.send_text_msg(
self.message_util.send_text(
f"😄 哈哈，你慢了一步！\n"
f"🌟 任务 task_{active_task_id} 已经完结\n"
f"🎈 快去抢新任务吧！",
@@ -362,7 +362,7 @@ class GameTaskPlugin(MessagePluginInterface):
self.encyclopedia_db.complete_task(active_task_id)
if sender == holder_id:
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 {player_name} 你是天才吗？\n"
f"🌟 任务：{question}\n"
f"🎈 答对啦，简直无敌！\n"
@@ -372,7 +372,7 @@ class GameTaskPlugin(MessagePluginInterface):
sender
)
else:
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎉 {player_name} 抢答王上线！\n"
f"🌟 任务：{question}\n"
f"🎈 原主：{holder_name} 被你截胡啦！\n"
@@ -385,7 +385,7 @@ class GameTaskPlugin(MessagePluginInterface):
# 扣除积分
self.encyclopedia_db.update_player_points(sender, roomid, -1)
points = -1
self.message_util.send_text_msg(
self.message_util.send_text(
f"😅 {player_name} 你这是要笑死我吗？\n"
f"🌼 任务：{question}\n"
f"🎈 你答：{answer}\n"
@@ -409,7 +409,7 @@ class GameTaskPlugin(MessagePluginInterface):
ranks = self.encyclopedia_db.get_player_ranking(roomid, 10)
if not ranks:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 群 {roomid} 冷冷清清\n"
f"🌟 快来一起燥起来吧！",
(roomid if roomid else sender),
@@ -421,14 +421,14 @@ class GameTaskPlugin(MessagePluginInterface):
for i, row in enumerate(ranks, 1):
rank_text += f"🐓 {i}. {row['player_name']}: {row['points']} 分\n"
self.message_util.send_text_msg(
self.message_util.send_text(
rank_text,
(roomid if roomid else sender),
sender
)
except Exception as e:
self.LOG.error(f"显示排行榜出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 获取排行榜出错，请稍后再试！",
(roomid if roomid else sender),
sender
@@ -441,7 +441,7 @@ class GameTaskPlugin(MessagePluginInterface):
tasks = self.encyclopedia_db.get_active_tasks_in_group(roomid)
if not tasks:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😄 群 {roomid} 现在一片祥和\n"
f"🌟 没任务？快用 /t 搞一个！",
(roomid if roomid else sender),
@@ -457,14 +457,14 @@ class GameTaskPlugin(MessagePluginInterface):
f"🌼 大佬：{task['player_name']}\n"
)
self.message_util.send_text_msg(
self.message_util.send_text(
task_text,
(roomid if roomid else sender),
sender
)
except Exception as e:
self.LOG.error(f"显示活跃任务出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 获取活跃任务出错，请稍后再试！",
(roomid if roomid else sender),
sender
@@ -477,7 +477,7 @@ class GameTaskPlugin(MessagePluginInterface):
tasks = self.encyclopedia_db.get_active_tasks_in_group(roomid)
if not tasks:
self.message_util.send_text_msg(
self.message_util.send_text(
f"😄 群 {roomid} 全员开挂？\n"
f"🌟 没未完成任务，快用 /t 再战！",
(roomid if roomid else sender),
@@ -493,14 +493,14 @@ class GameTaskPlugin(MessagePluginInterface):
f"🌼 主人：{task['player_name']}\n"
)
self.message_util.send_text_msg(
self.message_util.send_text(
task_text,
(roomid if roomid else sender),
sender
)
except Exception as e:
self.LOG.error(f"列举未完成任务出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"😔 获取未完成任务出错，请稍后再试！",
(roomid if roomid else sender),
sender
@@ -542,7 +542,7 @@ class GameTaskPlugin(MessagePluginInterface):
)
if active_task_id:
self.message_util.send_text_msg(
self.message_util.send_text(
f"🎁 新任务来袭，够不够刺激？\n"
f"🎀 任务ID: {active_task_id}\n"
f"🌟 幸运鹅：{holder_name}\n"

2
plugins/group_member_change/main.py
View File

@@ -201,7 +201,7 @@ class GroupMemberChangePlugin(MessagePluginInterface):
退群时间: {now_time}
"""
# 使用message_util发送消息
self.message_util.send_text_msg(message, group_id)
self.message_util.send_text(message, group_id)
self.LOG.info(f"已发送退群通知: {nickname} 退出群 {group_id}")
def _send_join_notification(self, group_id: str, wxid: str, nickname: str):

141
plugins/group_virtual/README.md Normal file
View File

@@ -0,0 +1,141 @@
# 跨群聊天插件需求分析与文档
## 1. 需求概述
开发一个名为"跨群聊天"的微信机器人插件，该插件允许用户将多个微信群组合成一个虚拟聊天组，实现群组间消息的自动转发，使不同群的成员能够进行跨群交流。
## 2. 功能需求
### 2.1 核心功能
1. **群组管理**
- 创建虚拟聊天组
- 向虚拟聊天组中添加/删除微信群
- 查看当前所有虚拟聊天组及其包含的群
- 解散虚拟聊天组
2. **消息转发**
- 自动捕获虚拟聊天组内任一群的消息
- 将消息转发至同一虚拟聊天组内的其他群
- 支持文本、图片等消息类型的转发（不支持语音、表情）
3. **消息标识**
- 转发的消息需标明原始发送者信息（昵称）
- 标明消息来源群组
- 避免消息循环转发
### 2.2 扩展功能
1. **权限控制**
- 设置管理员权限，只有管理员可以管理虚拟聊天组
- 支持黑名单功能，可以屏蔽特定用户的消息不被转发
2. **消息过滤**
- 支持关键词过滤，含有特定关键词的消息不转发
- 支持消息类型过滤，可选择只转发特定类型的消息（仅支持文本、图片）
3. **统计分析**
- 记录转发消息数量
- 统计各群组活跃度
## 3. 技术架构
### 3.1 数据结构
1. **虚拟聊天组(ChatGroup)**
- ID: 唯一标识符
- 名称: 虚拟聊天组名称
- 群组列表: 包含的微信群ID列表
- 创建时间
- 创建者
2. **消息记录(Message)**
- 消息ID
- 原始群ID
- 发送者ID
- 消息内容
- 消息类型
- 发送时间
- 转发状态
### 3.2 接口设计
1. **管理接口**
- 创建虚拟聊天组: `createChatGroup(name, creatorId)`
- 添加群到虚拟聊天组: `addGroupToChatGroup(chatGroupId, wxGroupId)`
- 从虚拟聊天组移除群: `removeGroupFromChatGroup(chatGroupId, wxGroupId)`
- 解散虚拟聊天组: `deleteChatGroup(chatGroupId)`
- 查询虚拟聊天组: `listChatGroups()`, `getChatGroupDetail(chatGroupId)`
2. **消息处理接口**
- 消息接收: `receiveMessage(wxGroupId, senderId, content, type)`
- 消息转发: `forwardMessage(message, targetGroupIds)`
- 消息过滤: `filterMessage(message, rules)`
## 4. 数据存储
使用JSON文件存储虚拟聊天组配置和相关数据：
```json
{
"chatGroups": [
{
"id": "cg001",
"name": "技术交流联盟",
"groups": ["wxid_group1", "wxid_group2", "wxid_group3"],
"createdAt": "2023-05-01T10:00:00Z",
"createdBy": "admin_user_id"
}
]
}
```
## 5. 用户交互
### 5.1 命令格式
用户通过在群内发送特定格式的命令来管理虚拟聊天组：
- 创建虚拟聊天组: `#创建跨群 [名称]`
- 添加群到虚拟聊天组: `#添加群 [虚拟聊天组ID] [当前群名称]`
- 查看所有虚拟聊天组: `#查看跨群列表`
- 查看虚拟聊天组详情: `#查看跨群 [虚拟聊天组ID]`
- 退出虚拟聊天组: `#退出跨群 [虚拟聊天组ID]`
- 解散虚拟聊天组: `#解散跨群 [虚拟聊天组ID]`
### 5.2 消息展示格式
转发的消息格式示例：
```
[来自群"技术交流群"]
张三：这是一条测试消息
```
## 6. 实现步骤
1. 创建插件基础结构
2. 实现数据存储模块
3. 实现群组管理功能
4. 实现消息捕获与转发功能
5. 实现用户命令解析
6. 添加消息过滤功能
7. 实现权限控制
8. 测试与优化
## 7. 注意事项
1. 避免消息循环转发
2. 确保消息转发的实时性
3. 注意微信API的限制，避免触发封号风险
4. 保护用户隐私，不存储敏感信息
5. 考虑高并发情况下的性能问题
## 8. 后续优化方向
1. 支持更多消息类型的转发
2. 提供Web管理界面
3. 增加更细粒度的权限控制
4. 支持定时消息和定向转发
5. 添加AI智能过滤功能
这份需求文档提供了跨群聊天插件的基本框架和实现思路，可以根据实际开发过程进行调整和完善。

7
plugins/group_virtual/__init__.py Normal file
View File

@@ -0,0 +1,7 @@
# 从当前包的main模块导入GroupVirtualPlugin类
from .main import GroupVirtualPlugin
# 提供get_plugin函数，返回插件实例
def get_plugin():
"""获取插件实例"""
return GroupVirtualPlugin()

3
plugins/group_virtual/config.toml Normal file
View File

@@ -0,0 +1,3 @@
enable = true
command = ["#创建跨群", "#添加群", "#查看跨群列表", "#查看跨群", "#退出跨群", "#解散跨群"]

184
plugins/group_virtual/main.py Normal file
View File

@@ -0,0 +1,184 @@
import logging
from typing import Dict, Any, List, Optional, Tuple
from wcferry import Wcf
from message_util import MessageUtil
from plugin_common.message_plugin_interface import MessagePluginInterface
from plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.wechat.contact_manager import ContactManager
from db.connection import DBConnectionManager
from db.group_virtual_redis import GroupVirtualRedisDB
class GroupVirtualPlugin(MessagePluginInterface):
"""跨群聊天插件"""
@property
def name(self) -> str:
return "跨群聊天"
@property
def version(self) -> str:
return "1.0.0"
@property
def description(self) -> str:
return "将多个微信群组合成一个虚拟聊天组，实现群组间消息的自动转发"
@property
def author(self) -> str:
return "Trae AI"
@property
def command_prefix(self) -> Optional[str]:
return None # 不使用命令前缀，因为不处理命令
@property
def commands(self) -> List[str]:
return [] # 不处理任何命令
def __init__(self):
super().__init__()
self.data = None
self.message_cache = set() # 用于防止消息循环转发
self.group_virtual_redis = None
def initialize(self, context: Dict[str, Any]) -> bool:
"""初始化插件"""
self.LOG = logging.getLogger(f"Plugin.{self.name}")
self.LOG.info(f"正在初始化 {self.name} 插件...")
# 保存上下文对象
self.wcf: Wcf = context.get("wcf")
self.event_system = context.get("event_system")
self.message_util: MessageUtil = context.get("message_util")
# 初始化配置
self.enable = self._config.get("GroupVirtual", {}).get("enable", True)
# 初始化Redis数据库操作类
self.db_manager = DBConnectionManager.get_instance()
self.group_virtual_redis = GroupVirtualRedisDB(self.db_manager)
# 从Redis加载数据
self.data = self.group_virtual_redis.load_chat_groups()
self.LOG.info(f"[{self.name}] 插件初始化完成")
return True
def start(self) -> bool:
"""启动插件"""
self.LOG.info(f"[{self.name}] 插件已启动")
self.status = PluginStatus.RUNNING
return True
def stop(self) -> bool:
"""停止插件"""
self.LOG.info(f"[{self.name}] 插件已停止")
self.status = PluginStatus.STOPPED
return True
def can_process(self, message: Dict[str, Any]) -> bool:
"""检查是否可以处理该消息"""
if not self.enable:
return False
# 只处理群消息
roomid = message.get("roomid", "")
if not roomid:
return False
# 检查该群是否在任何虚拟聊天组中
chat_groups = self._get_chat_groups_by_group_id(roomid)
return len(chat_groups) > 0
@plugin_stats_decorator(plugin_name="跨群聊天")
def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
roomid = message.get("roomid", "")
sender = message.get("sender", "")
# 检查是否是机器人自己发送的消息
if sender == self.wcf.self_wxid:
return False, "不转发自己的消息"
# 获取该群所在的所有虚拟聊天组
chat_groups = self._get_chat_groups_by_group_id(roomid)
if not chat_groups:
return False, "该群不在任何虚拟聊天组中"
# 生成消息唯一标识（用于防止循环转发）
message_id = f"{message.get('id')}_{message.get('timestamp')}"
if message_id in self.message_cache:
return False, "消息已经转发过"
# 添加到缓存
self.message_cache.add(message_id)
# 缓存大小控制
if len(self.message_cache) > 1000:
self.message_cache = set(list(self.message_cache)[-500:])
# 处理每个虚拟聊天组的转发
for chat_group in chat_groups:
self._forward_message_in_chat_group(message, chat_group)
return True, "消息已转发"
def _forward_message_in_chat_group(self, message: Dict[str, Any], chat_group: Dict[str, Any]):
"""在虚拟聊天组内转发消息"""
roomid = message.get("roomid", "")
sender = message.get("sender", "")
con = ContactManager.get_instance()
# 获取发送者昵称
sender_name = con.get_contact_name(sender) or "未知用户"
# 获取源群名称
source_group_name = None
for group in chat_group["groups"]:
if group["id"] == roomid:
source_group_name = group["name"]
break
if not source_group_name:
source_group_name = "未知群组"
# 构建转发消息前缀
prefix = f"[{sender_name}@{source_group_name}]："
# 根据消息类型处理
msg_type = message.get("type", 0)
# 文本消息
if msg_type == 1:
content = message.get("content", "")
forward_content = f"{prefix}{content}"
# 转发到其他群
for group in chat_group["groups"]:
if group["id"] != roomid: # 不转发回源群
self.message_util.send_text(forward_content, group["id"])
# 图片消息 暂时不支持。
# elif msg_type == 3:
# # 先发送前缀
# for group in chat_group["groups"]:
# if group["id"] != roomid:
# self.message_util.send_text(f"{prefix}[图片]", group["id"])
# # 转发图片
# image_path = message.get("file_path", "")
# if image_path and os.path.exists(image_path):
# self.message_util.send_image(image_path, group["id"])
def _get_chat_group(self, chat_group_id: str) -> Optional[Dict[str, Any]]:
"""获取虚拟聊天组"""
return self.group_virtual_redis.get_chat_group(chat_group_id)
def _get_chat_groups_by_group_id(self, group_id: str) -> List[Dict[str, Any]]:
"""获取包含指定群的所有虚拟聊天组"""
return self.group_virtual_redis.get_chat_groups_by_group_id(group_id)
def _save_data(self):
"""保存数据到Redis"""
self.group_virtual_redis.save_chat_groups(self.data)

26
plugins/message_sign/main.py
View File

@@ -240,8 +240,8 @@ class MessageSignPlugin(MessagePluginInterface):
# 如果 sign_stat 已经大于或等于今天的零点，则认为用户已经签到过了
if sign_stat >= today_start:
self.message_util.send_text_msg(f"您今天已经签到过了！",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"您今天已经签到过了！",
(roomid if roomid else sender), sender)
return False, "已签到"
# 在_handle_sign_in方法中，修改断签处理逻辑
@@ -316,13 +316,13 @@ class MessageSignPlugin(MessagePluginInterface):
daily_vocab = self.get_random_vocabulary()
output += f"\n今日词汇：{daily_vocab}"
self.message_util.send_text_msg(output, (roomid if roomid else sender), sender)
self.message_util.send_text(output, (roomid if roomid else sender), sender)
return True, "签到成功"
except Exception as e:
self.LOG.error(f"处理签到请求出错: {e}")
self.message_util.send_text_msg(f"签到出错：{e}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"签到出错：{e}",
(roomid if roomid else sender), sender)
return False, f"处理出错: {e}"
def reset_today_count_if_needed(self):
@@ -373,7 +373,7 @@ class MessageSignPlugin(MessagePluginInterface):
# 检查用户是否有签到记录
if not user_record:
self.message_util.send_text_msg(
self.message_util.send_text(
"❌ 您还没有签到记录，请先进行签到！",
(roomid if roomid else sender), sender
)
@@ -406,7 +406,7 @@ class MessageSignPlugin(MessagePluginInterface):
else:
# 昨天已经签到了或者断签超过一天，不需要补签
self.LOG.info(f"不符合补签条件，last_sign_date: {last_sign_date}, sign_stat: {sign_stat}")
self.message_util.send_text_msg(
self.message_util.send_text(
"❌ 您昨天已经签到过了或断签超过一天，不符合补签条件！",
(roomid if roomid else sender), sender
)
@@ -415,7 +415,7 @@ class MessageSignPlugin(MessagePluginInterface):
# 今天未签到，检查是否符合补签条件（只能补签昨天）
if not last_sign_date or last_sign_date < day_before_yesterday:
self.LOG.info(f"断签超过一天，last_sign_date: {last_sign_date}")
self.message_util.send_text_msg(
self.message_util.send_text(
"❌ 只能补签断签一天的情况！您已断签超过一天或没有签到记录。",
(roomid if roomid else sender), sender
)
@@ -423,7 +423,7 @@ class MessageSignPlugin(MessagePluginInterface):
if last_sign_date >= yesterday:
self.LOG.info(f"昨天已签到，last_sign_date: {last_sign_date}")
self.message_util.send_text_msg(
self.message_util.send_text(
"❌ 您昨天已经签到过了，不需要补签！",
(roomid if roomid else sender), sender
)
@@ -435,7 +435,7 @@ class MessageSignPlugin(MessagePluginInterface):
user_points = points_db.get_user_points(sender, roomid)
if not user_points or user_points["total_points"] < self.makeup_cost:
self.message_util.send_text_msg(
self.message_util.send_text(
f"❌ 积分不足！补签需要 {self.makeup_cost} 积分，您当前只有 {user_points.get('total_points', 0)} 积分。",
(roomid if roomid else sender), sender
)
@@ -448,7 +448,7 @@ class MessageSignPlugin(MessagePluginInterface):
)
if not deduct_success:
self.message_util.send_text_msg(
self.message_util.send_text(
f"❌ 扣除积分失败：{deduct_result.get('error', '未知错误')}",
(roomid if roomid else sender), sender
)
@@ -507,7 +507,7 @@ class MessageSignPlugin(MessagePluginInterface):
success_message += f"💰 当前积分：{user_points['total_points'] - self.makeup_cost}"
self.message_util.send_text_msg(
self.message_util.send_text(
success_message,
(roomid if roomid else sender), sender
)
@@ -516,7 +516,7 @@ class MessageSignPlugin(MessagePluginInterface):
except Exception as e:
self.LOG.error(f"处理补签请求出错: {e}")
self.message_util.send_text_msg(
self.message_util.send_text(
f"❌ 补签出错：{e}",
(roomid if roomid else sender), sender
)

94
plugins/point_trade/main.py
View File

@@ -149,8 +149,8 @@ class PointTradePlugin(MessagePluginInterface):
elif command[0] in self._commands:
return self._handle_transfer_points(message)
else:
self.message_util.send_text_msg(f"❌未知命令！{self.command_format}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌未知命令！{self.command_format}",
(roomid if roomid else sender), sender)
return True, "未知命令"
def _handle_transfer_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
@@ -164,21 +164,21 @@ class PointTradePlugin(MessagePluginInterface):
# 检查命令格式
if len(command) < 3:
self.message_util.send_text_msg(f"❌命令格式错误！积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌命令格式错误！积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
return True, "命令格式错误"
# 检查积分数是否为正整数
if not command[1].isdigit():
self.message_util.send_text_msg(f"🈚️转账积分无效(必须为正整数!) \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"🈚️转账积分无效(必须为正整数!) \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
return True, "积分无效"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text_msg(f"转账失败❌\n🈚️转账人无效！ \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"转账失败❌\n🈚️转账人无效！ \n积分转账 积分数 @用户",
(roomid if roomid else sender), sender)
return True, "转账人无效"
reward_points = int(command[1])
@@ -197,11 +197,11 @@ class PointTradePlugin(MessagePluginInterface):
error_msg = result.get("error", "未知错误")
if "积分不足" in error_msg:
current_points = result.get("current_points", 0)
self.message_util.send_text_msg(
self.message_util.send_text(
f"❌转账失败!\n你的积分不足以进行转账！当前积分：{current_points}，你需要 {reward_points} 积分。",
(roomid if roomid else sender), sender)
else:
self.message_util.send_text_msg(
self.message_util.send_text(
f"❌转账失败!\n{error_msg}",
(roomid if roomid else sender), sender)
return True, f"转账失败: {error_msg}"
@@ -224,13 +224,13 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{to_user_name} 当前积分: {to_user.get('total_points', 0)}"
)
self.message_util.send_text_msg(output, (roomid if roomid else sender), sender)
self.message_util.send_text(output, (roomid if roomid else sender), sender)
return True, "转账成功"
except Exception as e:
self.LOG.error(f"积分交易出错: {e}")
self.message_util.send_text_msg(f"❌积分交易失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌积分交易失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
return True, f"处理出错: {str(e)}"
def _handle_my_points(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
@@ -243,8 +243,8 @@ class PointTradePlugin(MessagePluginInterface):
user_points = self.points_db.get_user_points(sender, roomid)
if not user_points:
self.message_util.send_text_msg(f"❌未找到你的积分记录！请先参与积分活动[签到，答题/t]。",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌未找到你的积分记录！请先参与积分活动[签到，答题/t]。",
(roomid if roomid else sender), sender)
return False, "未找到积分记录"
# 获取用户昵称
@@ -299,13 +299,13 @@ class PointTradePlugin(MessagePluginInterface):
f"{recent_txs}"
)
self.message_util.send_text_msg(output, (roomid if roomid else sender), sender)
self.message_util.send_text(output, (roomid if roomid else sender), sender)
return True, "查询积分成功"
except Exception as e:
self.LOG.error(f"查询积分出错: {e}")
self.message_util.send_text_msg(f"❌查询积分失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"❌查询积分失败！请稍后重试。错误: {str(e)}",
(roomid if roomid else sender), sender)
return True, f"处理出错: {str(e)}"
def _handle_points_ranking(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
@@ -315,7 +315,7 @@ class PointTradePlugin(MessagePluginInterface):
wcf: Wcf = message.get("wcf")
if not roomid:
self.message_util.send_text_msg("❌积分排行榜仅在群聊中可用！", sender, "")
self.message_util.send_text("❌积分排行榜仅在群聊中可用！", sender, "")
return True, "非群聊环境"
try:
@@ -323,7 +323,7 @@ class PointTradePlugin(MessagePluginInterface):
ranking = self.points_db.get_points_ranking(roomid, 10)
if not ranking:
self.message_util.send_text_msg("❌暂无积分排行数据！请先参与积分活动。", roomid, sender)
self.message_util.send_text("❌暂无积分排行数据！请先参与积分活动。", roomid, sender)
return True, "无排行数据"
# 构建排行榜消息
@@ -348,12 +348,12 @@ class PointTradePlugin(MessagePluginInterface):
f"更新时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
)
self.message_util.send_text_msg(output, roomid, sender)
self.message_util.send_text(output, roomid, sender)
return True, "查询排行榜成功"
except Exception as e:
self.LOG.error(f"查询积分排行榜出错: {e}")
self.message_util.send_text_msg(f"❌查询积分排行榜失败！请稍后重试。错误: {str(e)}", roomid, sender)
self.message_util.send_text(f"❌查询积分排行榜失败！请稍后重试。错误: {str(e)}", roomid, sender)
return True, f"处理出错: {str(e)}"
def at_list(self, xml):
@@ -457,13 +457,13 @@ class PointTradePlugin(MessagePluginInterface):
# 检查是否在群聊中
if not roomid:
self.message_util.send_text_msg("❌打劫功能仅在群聊中可用！", sender, "")
self.message_util.send_text("❌打劫功能仅在群聊中可用！", 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_msg("❌打劫功能仅在晚上18:00-21:00之间开放！请在开放时间再来。", roomid, sender)
self.message_util.send_text("❌打劫功能仅在晚上18:00-21:00之间开放！请在开放时间再来。", roomid, sender)
return True, "时间限制"
# 检查是否在押
@@ -472,8 +472,8 @@ 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_msg(f"❌你正在服刑！\n剩余时间: {hours}小时{minutes}分钟\n可请求他人花费30积分保释。", roomid,
sender)
self.message_util.send_text(f"❌你正在服刑！\n剩余时间: {hours}小时{minutes}分钟\n可请求他人花费30积分保释。", roomid,
sender)
return True, "在押状态"
# 检查冷却时间
@@ -484,15 +484,15 @@ 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_msg(f"❌你最近已经打劫过了，需要冷却 {minutes}分{seconds}秒 后才能再次打劫！",
roomid, sender)
self.message_util.send_text(f"❌你最近已经打劫过了，需要冷却 {minutes}分{seconds}秒 后才能再次打劫！",
roomid, sender)
return True, "冷却中"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text_msg(f"打劫失败❌\n请指定一个打劫目标！\n打劫 @用户",
roomid, sender)
self.message_util.send_text(f"打劫失败❌\n请指定一个打劫目标！\n打劫 @用户",
roomid, sender)
return True, "目标无效"
target_wxid = next(iter(at_users))
@@ -500,7 +500,7 @@ class PointTradePlugin(MessagePluginInterface):
# 不能打劫自己
if target_wxid == robber_wxid:
self.message_util.send_text_msg("❌你不能打劫自己！", roomid, sender)
self.message_util.send_text("❌你不能打劫自己！", roomid, sender)
return True, "不能打劫自己"
try:
@@ -509,11 +509,11 @@ class PointTradePlugin(MessagePluginInterface):
target_points = self.points_db.get_user_points(target_wxid, roomid)
if not robber_points:
self.message_util.send_text_msg("❌你没有积分记录，无法进行打劫！请先参与积分活动。", roomid, sender)
self.message_util.send_text("❌你没有积分记录，无法进行打劫！请先参与积分活动。", roomid, sender)
return True, "打劫者无积分"
if not target_points:
self.message_util.send_text_msg("❌目标没有积分记录，无法进行打劫！", roomid, sender)
self.message_util.send_text("❌目标没有积分记录，无法进行打劫！", roomid, sender)
return True, "目标无积分"
robber_total = robber_points.get('total_points', 0)
@@ -521,11 +521,11 @@ class PointTradePlugin(MessagePluginInterface):
# 检查最低积分要求
if robber_total < self.rob_min_points:
self.message_util.send_text_msg(f"❌你的积分不足 {self.rob_min_points} 点，无法进行打劫！", roomid, sender)
self.message_util.send_text(f"❌你的积分不足 {self.rob_min_points} 点，无法进行打劫！", roomid, sender)
return True, "打劫者积分不足"
if target_total < self.rob_min_points:
self.message_util.send_text_msg(f"❌目标积分不足 {self.rob_min_points} 点，不值得打劫！", roomid, sender)
self.message_util.send_text(f"❌目标积分不足 {self.rob_min_points} 点，不值得打劫！", roomid, sender)
return True, "目标积分不足"
# 获取用户昵称
@@ -581,10 +581,10 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{target_name} 当前积分: {from_user.get('total_points', 0)}"
)
self.message_util.send_text_msg(output, roomid, sender)
self.message_util.send_text(output, roomid, sender)
return True, "打劫成功"
else:
self.message_util.send_text_msg(f"❌打劫过程中出现问题：{result.get('error', '未知错误')}", roomid, sender)
self.message_util.send_text(f"❌打劫过程中出现问题：{result.get('error', '未知错误')}", roomid, sender)
return True, "打劫失败"
else:
@@ -618,15 +618,15 @@ class PointTradePlugin(MessagePluginInterface):
f"👤{robber_name} 当前积分: {from_user.get('total_points', 0)}"
)
self.message_util.send_text_msg(output, roomid, sender)
self.message_util.send_text(output, roomid, sender)
return True, "打劫失败"
else:
self.message_util.send_text_msg(f"❌处理打劫惩罚时出现问题：{result.get('error', '未知错误')}", roomid, sender)
self.message_util.send_text(f"❌处理打劫惩罚时出现问题：{result.get('error', '未知错误')}", roomid, sender)
return True, "处理惩罚失败"
except Exception as e:
self.LOG.error(f"处理打劫请求出错: {e}")
self.message_util.send_text_msg(f"❌打劫过程中出现意外：{str(e)}", roomid, sender)
self.message_util.send_text(f"❌打劫过程中出现意外：{str(e)}", roomid, sender)
return True, f"处理出错: {str(e)}"
def _handle_bailout(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
@@ -638,14 +638,14 @@ class PointTradePlugin(MessagePluginInterface):
# 检查是否在群聊中
if not roomid:
self.message_util.send_text_msg("❌保释功能仅在群聊中可用！", sender, "")
self.message_util.send_text("❌保释功能仅在群聊中可用！", sender, "")
return True, "非群聊环境"
# 检查@用户是否有效
at_users = self.at_list(xml)
if len(at_users) != 1:
self.message_util.send_text_msg(f"保释失败❌\n请指定一个保释目标！\n保释 @用户",
roomid, sender)
self.message_util.send_text(f"保释失败❌\n请指定一个保释目标！\n保释 @用户",
roomid, sender)
return True, "目标无效"
prisoner_wxid = next(iter(at_users))
@@ -653,7 +653,7 @@ class PointTradePlugin(MessagePluginInterface):
# 不能保释自己
if prisoner_wxid == bailout_wxid:
self.message_util.send_text_msg("❌你不能保释自己！", roomid, sender)
self.message_util.send_text("❌你不能保释自己！", roomid, sender)
return True, "不能保释自己"
try:
@@ -672,13 +672,13 @@ class PointTradePlugin(MessagePluginInterface):
f"✅ 保释成功！\n"
f"👤{bailout_name} 花费30积分保释了 👤{prisoner_name}"
)
self.message_util.send_text_msg(output, roomid, sender)
self.message_util.send_text(output, roomid, sender)
return True, "保释成功"
else:
self.message_util.send_text_msg(f"❌保释失败: {message}", roomid, sender)
self.message_util.send_text(f"❌保释失败: {message}", roomid, sender)
return True, "保释失败"
except Exception as e:
self.LOG.error(f"处理保释请求出错: {e}")
self.message_util.send_text_msg(f"❌保释过程中出现意外：{str(e)}", roomid, sender)
self.message_util.send_text(f"❌保释过程中出现意外：{str(e)}", roomid, sender)
return True, f"处理出错: {str(e)}"

4
plugins/video_man/main.py
View File

@@ -109,8 +109,8 @@ class VideoManPlugin(MessagePluginInterface):
file_abspath = self._download_video("https://api.guiguiya.com/api/video/fuji?type=json")
if not file_abspath:
self.message_util.send_text_msg(f"\n❌视频下载失败，请稍后再试",
(roomid if roomid else sender), sender)
self.message_util.send_text(f"\n❌视频下载失败，请稍后再试",
(roomid if roomid else sender), sender)
return False, "视频下载失败"
# 发送视频