nezumi/astrbot_plugin_qq_group_daily_analysis
1
0
Fork 0
mirror of https://github.com/Nezumi-2711/astrbot_plugin_qq_group_daily_analysis.git synced 2026-09-22 13:38:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复 Discord 适配器客户端访问问题

Browse Source 
问题：self.bot 是 DiscordPlatformAdapter 对象，没有 get_channel 方法。

修复：

添加 _get_discord_client() 辅助方法来正确获取 Discord 客户端（通过 DiscordPlatformAdapter.client）
将所有方法中的 self.bot.xxx 替换为 self._discord_client.xxx
This commit is contained in:
SXP-Simon
2026-02-08 19:44:16 +08:00
parent e780e3cbe9
commit df28aa1076
1 changed files with 202 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/infrastructure/platform/adapters/discord_adapter.py
+202 -132
View File
@@ -32,16 +32,17 @@ from ..base import PlatformAdapter
logger = logging.getLogger(__name__)
class DiscordAdapter(PlatformAdapter):
"""
Discord 平台适配器
实现 PlatformAdapter 接口，提供 Discord 平台的消息操作。
使用方式：
1. 通过 PlatformAdapterFactory.create("discord", bot_instance, config) 创建
2. 或直接实例化：DiscordAdapter(bot_instance, config)
配置参数：
- bot_user_id: 机器人的 Discord 用户 ID（用于过滤自己的消息）
"""
@@ -50,10 +51,35 @@ class DiscordAdapter(PlatformAdapter):
super().__init__(bot_instance, config)
# 机器人自己的用户 ID，用于过滤消息
self.bot_user_id = str(config.get("bot_user_id", "")) if config else ""
# 尝试从 bot 实例获取 ID
if not self.bot_user_id and hasattr(self.bot, "user") and self.bot.user:
self.bot_user_id = str(self.bot.user.id)
# 获取实际的 Discord 客户端
self._discord_client = self._get_discord_client()
# 尝试从 Discord 客户端获取 ID
if not self.bot_user_id and self._discord_client:
if hasattr(self._discord_client, "user") and self._discord_client.user:
self.bot_user_id = str(self._discord_client.user.id)
def _get_discord_client(self) -> Any:
"""
获取实际的 Discord 客户端实例
AstrBot 的 DiscordPlatformAdapter 将 Discord 客户端存储在 self.client 中
"""
# 如果 bot 本身就是 Discord client (有 get_channel 方法)
if hasattr(self.bot, "get_channel"):
return self.bot
# 如果 bot 是 DiscordPlatformAdapter，client 在 self.bot.client 中
if hasattr(self.bot, "client"):
return self.bot.client
# 尝试其他可能的属性名
for attr in ["_client", "discord_client", "_discord_client"]:
if hasattr(self.bot, attr):
client = getattr(self.bot, attr)
if hasattr(client, "get_channel"):
return client
logger.warning(f"无法从 {type(self.bot).__name__} 获取 Discord 客户端")
return None
def _init_capabilities(self) -> PlatformCapabilities:
"""初始化 Discord 平台能力"""
@@ -70,13 +96,13 @@ class DiscordAdapter(PlatformAdapter):
) -> List[UnifiedMessage]:
"""
获取 Discord 频道消息历史
参数：
group_id: Discord 频道 ID
days: 获取多少天内的消息
max_count: 最大消息数量
before_id: 从此消息 ID 之前开始获取（用于分页）
返回：
UnifiedMessage 列表
"""
@@ -86,30 +112,27 @@ class DiscordAdapter(PlatformAdapter):
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
channel = self._discord_client.get_channel(channel_id)
if not channel:
# 尝试 fetch (API调用)
try:
channel = await self.bot.fetch_channel(channel_id)
channel = await self._discord_client.fetch_channel(channel_id)
except Exception:
logger.warning(f"无法找到频道 ID: {group_id}")
return []
# 检查频道是否支持历史记录
if not hasattr(channel, "history"):
logger.warning(f"频道 {group_id} 不支持历史消息获取")
return []
end_time = datetime.now()
start_time = end_time - timedelta(days=days)
messages = []
# 构建 history 参数
history_kwargs = {
"limit": max_count,
"after": start_time
}
history_kwargs = {"limit": max_count, "after": start_time}
if before_id:
try:
# before 可以接受 Message 对象或 ID (int)
@@ -122,15 +145,15 @@ class DiscordAdapter(PlatformAdapter):
# 过滤机器人自己的消息（如果配置了 ID）
if self.bot_user_id and str(msg.author.id) == self.bot_user_id:
continue
unified = self._convert_message(msg, group_id)
if unified:
messages.append(unified)
# 按时间升序排序
messages.sort(key=lambda m: m.timestamp)
return messages
except Exception as e:
logger.error(f"获取 Discord 消息失败: {e}", exc_info=True)
return []
@@ -138,72 +161,87 @@ class DiscordAdapter(PlatformAdapter):
def _convert_message(self, raw_msg: Any, group_id: str) -> Optional[UnifiedMessage]:
"""
将 Discord 消息转换为统一格式
参数：
raw_msg: Discord 原始消息对象 (discord.Message)
group_id: 频道 ID
返回：
UnifiedMessage 或 None
"""
try:
contents = []
# 1. 文本内容
if raw_msg.content:
contents.append(MessageContent(
type=MessageContentType.TEXT,
text=raw_msg.content
))
contents.append(
MessageContent(type=MessageContentType.TEXT, text=raw_msg.content)
)
# 2. 附件处理
for attachment in raw_msg.attachments:
content_type = attachment.content_type or ""
if content_type.startswith("image/"):
contents.append(MessageContent(
type=MessageContentType.IMAGE,
url=attachment.url
))
contents.append(
MessageContent(
type=MessageContentType.IMAGE, url=attachment.url
)
)
elif content_type.startswith("video/"):
contents.append(MessageContent(
type=MessageContentType.VIDEO,
url=attachment.url
))
contents.append(
MessageContent(
type=MessageContentType.VIDEO, url=attachment.url
)
)
elif content_type.startswith("audio/"):
contents.append(MessageContent(
type=MessageContentType.VOICE,
url=attachment.url
))
contents.append(
MessageContent(
type=MessageContentType.VOICE, url=attachment.url
)
)
else:
contents.append(MessageContent(
type=MessageContentType.FILE,
url=attachment.url,
raw_data={"filename": attachment.filename, "size": attachment.size}
))
contents.append(
MessageContent(
type=MessageContentType.FILE,
url=attachment.url,
raw_data={
"filename": attachment.filename,
"size": attachment.size,
},
)
)
# 3. 嵌入内容 (Embeds) - 通常是富文本或图片
for embed in raw_msg.embeds:
if embed.image:
contents.append(MessageContent(
type=MessageContentType.IMAGE,
url=embed.image.url
))
contents.append(
MessageContent(
type=MessageContentType.IMAGE, url=embed.image.url
)
)
# 其他 embed 内容暂作为未知类型或文本处理
if embed.description:
contents.append(MessageContent(
type=MessageContentType.TEXT,
text=f"\n[Embed] {embed.description}"
))
contents.append(
MessageContent(
type=MessageContentType.TEXT,
text=f"\n[Embed] {embed.description}",
)
)
# 4. 贴纸 (Stickers)
if raw_msg.stickers:
for sticker in raw_msg.stickers:
contents.append(MessageContent(
type=MessageContentType.IMAGE, # 贴纸视为图片
url=sticker.url,
raw_data={"sticker_id": str(sticker.id), "sticker_name": sticker.name}
))
contents.append(
MessageContent(
type=MessageContentType.IMAGE, # 贴纸视为图片
url=sticker.url,
raw_data={
"sticker_id": str(sticker.id),
"sticker_name": sticker.name,
},
)
)
# 发送者名片 (昵称)
sender_card = None
if hasattr(raw_msg.author, "nick") and raw_msg.author.nick:
@@ -214,14 +252,16 @@ class DiscordAdapter(PlatformAdapter):
return UnifiedMessage(
message_id=str(raw_msg.id),
sender_id=str(raw_msg.author.id),
sender_name=raw_msg.author.name, # 用户名
sender_card=sender_card, # 服务器昵称
sender_name=raw_msg.author.name, # 用户名
sender_card=sender_card, # 服务器昵称
group_id=group_id,
text_content=raw_msg.content,
contents=tuple(contents),
timestamp=int(raw_msg.created_at.timestamp()),
platform="discord",
reply_to_id=str(raw_msg.reference.message_id) if raw_msg.reference else None,
reply_to_id=str(raw_msg.reference.message_id)
if raw_msg.reference
else None,
)
except Exception as e:
logger.error(f"转换 Discord 消息失败: {e}")
@@ -245,22 +285,36 @@ class DiscordAdapter(PlatformAdapter):
},
"message": [],
}
# 构造消息链
for content in msg.contents:
if content.type == MessageContentType.TEXT:
raw_msg["message"].append({"type": "text", "data": {"text": content.text}})
raw_msg["message"].append(
{"type": "text", "data": {"text": content.text}}
)
elif content.type == MessageContentType.IMAGE:
raw_msg["message"].append({"type": "image", "data": {"url": content.url, "file": content.url}})
raw_msg["message"].append(
{
"type": "image",
"data": {"url": content.url, "file": content.url},
}
)
elif content.type == MessageContentType.AT:
raw_msg["message"].append({"type": "at", "data": {"qq": content.at_user_id}})
raw_msg["message"].append(
{"type": "at", "data": {"qq": content.at_user_id}}
)
elif content.type == MessageContentType.REPLY:
if content.raw_data and "reply_id" in content.raw_data:
raw_msg["message"].append({"type": "reply", "data": {"id": content.raw_data["reply_id"]}})
raw_msg["message"].append(
{
"type": "reply",
"data": {"id": content.raw_data["reply_id"]},
}
)
# 其他类型暂忽略或作为未知
raw_messages.append(raw_msg)
return raw_messages
# ==================== IMessageSender ====================
@@ -272,28 +326,28 @@ class DiscordAdapter(PlatformAdapter):
reply_to: Optional[str] = None,
) -> bool:
"""发送文本消息到 Discord 频道"""
if not discord: return False
if not discord:
return False
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
if not hasattr(channel, "send"):
return False
reference = None
if reply_to:
try:
# 创建 MessageReference
reference = discord.MessageReference(
message_id=int(reply_to),
channel_id=channel_id
message_id=int(reply_to), channel_id=channel_id
)
except ValueError:
pass
await channel.send(content=text, reference=reference)
return True
except Exception as e:
@@ -307,14 +361,15 @@ class DiscordAdapter(PlatformAdapter):
caption: str = "",
) -> bool:
"""发送图片到 Discord 频道"""
if not discord: return False
if not discord:
return False
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
if not hasattr(channel, "send"):
return False
@@ -331,7 +386,7 @@ class DiscordAdapter(PlatformAdapter):
file_to_send = discord.File(image_path)
await channel.send(content=caption, file=file_to_send)
return True
except Exception as e:
logger.error(f"Discord 发送图片失败: {e}")
return False
@@ -343,8 +398,9 @@ class DiscordAdapter(PlatformAdapter):
filename: Optional[str] = None,
) -> bool:
"""发送文件到 Discord 频道"""
if not discord: return False
if not discord:
return False
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
@@ -353,7 +409,7 @@ class DiscordAdapter(PlatformAdapter):
if not hasattr(channel, "send"):
return False
file_to_send = discord.File(file_path, filename=filename)
await channel.send(file=file_to_send)
return True
@@ -365,17 +421,18 @@ class DiscordAdapter(PlatformAdapter):
async def get_group_info(self, group_id: str) -> Optional[UnifiedGroup]:
"""获取 Discord 频道信息"""
if not discord: return None
if not discord:
return None
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
# 尝试获取 Guild 信息
guild = getattr(channel, "guild", None)
group_name = getattr(channel, "name", str(channel.id))
if guild:
# 如果是公会频道，可以用 Guild 信息补充
@@ -383,9 +440,9 @@ class DiscordAdapter(PlatformAdapter):
owner_id = str(guild.owner_id)
else:
# 私信或群组私信
member_count = len(getattr(channel, "recipients", [])) + 1 # +1 for bot
member_count = len(getattr(channel, "recipients", [])) + 1 # +1 for bot
owner_id = str(getattr(channel, "owner_id", ""))
return UnifiedGroup(
group_id=str(channel.id),
group_name=group_name,
@@ -400,14 +457,15 @@ class DiscordAdapter(PlatformAdapter):
async def get_group_list(self) -> List[str]:
"""获取机器人所在的所有频道 ID (仅列出 TextChannel)"""
if not discord: return []
if not discord:
return []
try:
# 遍历所有 Guilds 和 Channels
channel_ids = []
for guild in self.bot.guilds:
for guild in self._discord_client.guilds:
for channel in guild.text_channels:
channel_ids.append(str(channel.id))
channel_ids.append(str(channel.id))
return channel_ids
except Exception as e:
logger.error(f"Discord 获取群组列表失败: {e}")
@@ -415,26 +473,29 @@ class DiscordAdapter(PlatformAdapter):
async def get_member_list(self, group_id: str) -> List[UnifiedMember]:
"""获取 Discord 服务器成员列表"""
if not discord: return []
if not discord:
return []
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
guild = getattr(channel, "guild", None)
if not guild:
# 非公会频道（如 DM），返回收件人
members = []
for user in getattr(channel, "recipients", []):
members.append(UnifiedMember(
user_id=str(user.id),
nickname=user.display_name,
card=None,
role="member",
join_time=None
))
members.append(
UnifiedMember(
user_id=str(user.id),
nickname=user.display_name,
card=None,
role="member",
join_time=None,
)
)
return members
# 公会频道
@@ -447,14 +508,18 @@ class DiscordAdapter(PlatformAdapter):
role = "owner"
elif member.guild_permissions.administrator:
role = "admin"
members.append(UnifiedMember(
user_id=str(member.id),
nickname=member.name,
card=member.nick or member.global_name, # 优先显示服务器昵称
role=role,
join_time=int(member.joined_at.timestamp()) if member.joined_at else None,
))
members.append(
UnifiedMember(
user_id=str(member.id),
nickname=member.name,
card=member.nick or member.global_name, # 优先显示服务器昵称
role=role,
join_time=int(member.joined_at.timestamp())
if member.joined_at
else None,
)
)
return members
except Exception as e:
logger.error(f"Discord 获取成员列表失败: {e}")
@@ -466,14 +531,15 @@ class DiscordAdapter(PlatformAdapter):
user_id: str,
) -> Optional[UnifiedMember]:
"""获取特定成员信息"""
if not discord: return None
if not discord:
return None
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
guild = getattr(channel, "guild", None)
if not guild:
# 私信，尝试 fetch user
@@ -483,28 +549,30 @@ class DiscordAdapter(PlatformAdapter):
nickname=user.name,
card=user.display_name,
role="member",
join_time=None
join_time=None,
)
member = guild.get_member(int(user_id))
if not member:
member = await guild.fetch_member(int(user_id))
if not member:
return None
role = "member"
if member.id == guild.owner_id:
role = "owner"
elif member.guild_permissions.administrator:
role = "admin"
return UnifiedMember(
user_id=str(member.id),
nickname=member.name,
card=member.nick or member.global_name,
role=role,
join_time=int(member.joined_at.timestamp()) if member.joined_at else None,
join_time=int(member.joined_at.timestamp())
if member.joined_at
else None,
)
except Exception as e:
logger.error(f"Discord 获取成员信息失败: {e}")
@@ -518,18 +586,19 @@ class DiscordAdapter(PlatformAdapter):
size: int = 100,
) -> Optional[str]:
"""获取 Discord 用户头像 URL"""
if not discord: return None
if not discord:
return None
try:
user = self.bot.get_user(int(user_id))
user = self._discord_client.get_user(int(user_id))
if not user:
user = await self.bot.fetch_user(int(user_id))
user = await self._discord_client.fetch_user(int(user_id))
if user:
# 调整 size 到最接近的 2 的幂次方
allowed_sizes = [16, 32, 64, 128, 256, 512, 1024, 2048, 4096]
target_size = min(allowed_sizes, key=lambda x: abs(x - size))
# display_avatar 自动处理默认头像
return user.display_avatar.with_size(target_size).url
return None
@@ -551,19 +620,20 @@ class DiscordAdapter(PlatformAdapter):
size: int = 100,
) -> Optional[str]:
"""获取 Discord 服务器图标 URL"""
if not discord: return None
if not discord:
return None
try:
channel_id = int(group_id)
channel = self.bot.get_channel(channel_id)
if not channel:
channel = await self.bot.fetch_channel(channel_id)
guild = getattr(channel, "guild", None)
if guild and guild.icon:
allowed_sizes = [16, 32, 64, 128, 256, 512, 1024, 2048, 4096]
target_size = min(allowed_sizes, key=lambda x: abs(x - size))
return guild.icon.with_size(target_size).url
allowed_sizes = [16, 32, 64, 128, 256, 512, 1024, 2048, 4096]
target_size = min(allowed_sizes, key=lambda x: abs(x - size))
return guild.icon.with_size(target_size).url
return None
except Exception:
return None