175 lines
5.0 KiB
Python
175 lines
5.0 KiB
Python
"""WPS消息构造和发送工具"""
|
||
import httpx
|
||
import logging
|
||
from typing import Dict, Any, Optional
|
||
from config import GetWebhookURL
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class MessageSender:
|
||
"""消息发送器"""
|
||
|
||
def __init__(self, webhook_url: Optional[str] = None):
|
||
"""初始化消息发送器
|
||
|
||
Args:
|
||
webhook_url: Webhook URL
|
||
"""
|
||
if webhook_url is None:
|
||
webhook_url = GetWebhookURL()
|
||
self.webhook_url = webhook_url
|
||
self.client: Optional[httpx.AsyncClient] = None
|
||
|
||
async def _get_client(self) -> httpx.AsyncClient:
|
||
"""获取HTTP客户端（懒加载）"""
|
||
if self.client is None:
|
||
self.client = httpx.AsyncClient(timeout=10.0)
|
||
return self.client
|
||
|
||
async def send_message(self, message: Dict[str, Any]) -> bool:
|
||
"""发送消息到WPS
|
||
|
||
Args:
|
||
message: 消息字典
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
try:
|
||
client = await self._get_client()
|
||
response = await client.post(self.webhook_url, json=message)
|
||
|
||
if response.status_code == 200:
|
||
logger.info(f"消息发送成功: {message.get('msgtype')}")
|
||
return True
|
||
else:
|
||
logger.error(f"消息发送失败: status={response.status_code}, body={response.text}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
logger.error(f"发送消息异常: {e}", exc_info=True)
|
||
return False
|
||
|
||
async def send_text(self, content: str, at_user_id: Optional[int] = None) -> bool:
|
||
"""发送文本消息
|
||
|
||
Args:
|
||
content: 文本内容
|
||
at_user_id: @用户ID（可选）
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
# 如果需要@人
|
||
if at_user_id:
|
||
content = f'<at user_id="{at_user_id}"></at> {content}'
|
||
|
||
message = {
|
||
"msgtype": "text",
|
||
"text": {
|
||
"content": content
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def send_markdown(self, text: str) -> bool:
|
||
"""发送Markdown消息
|
||
|
||
Args:
|
||
text: Markdown文本
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
message = {
|
||
"msgtype": "markdown",
|
||
"markdown": {
|
||
"text": text
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def send_link(self, title: str, text: str,
|
||
message_url: str = "", btn_title: str = "查看详情") -> bool:
|
||
"""发送链接消息
|
||
|
||
Args:
|
||
title: 标题
|
||
text: 文本内容
|
||
message_url: 跳转URL
|
||
btn_title: 按钮文字
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
message = {
|
||
"msgtype": "link",
|
||
"link": {
|
||
"title": title,
|
||
"text": text,
|
||
"messageUrl": message_url,
|
||
"btnTitle": btn_title
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def close(self):
|
||
"""关闭HTTP客户端"""
|
||
if self.client:
|
||
await self.client.aclose()
|
||
self.client = None
|
||
|
||
|
||
# 全局消息发送器实例
|
||
_sender_instance: Optional[MessageSender] = None
|
||
|
||
|
||
def get_message_sender() -> MessageSender:
|
||
"""获取全局消息发送器实例（单例模式）"""
|
||
global _sender_instance
|
||
if _sender_instance is None:
|
||
_sender_instance = MessageSender()
|
||
else:
|
||
# 更新Webhook URL以确保使用最新的值
|
||
_sender_instance.webhook_url = GetWebhookURL()
|
||
return _sender_instance
|
||
|
||
|
||
async def send_private_message(user_id: int, content: str, msg_type: str = 'text') -> bool:
|
||
"""发送私聊消息到用户个人webhook URL
|
||
|
||
Args:
|
||
user_id: 目标用户ID
|
||
content: 消息内容
|
||
msg_type: 消息类型 ('text' 或 'markdown')
|
||
|
||
Returns:
|
||
是否发送成功，如果用户没有个人URL则返回False
|
||
"""
|
||
from core.database import get_db
|
||
|
||
# 从数据库获取用户webhook URL
|
||
db = get_db()
|
||
webhook_url = db.get_user_webhook_url(user_id)
|
||
|
||
if not webhook_url:
|
||
logger.warning(f"用户 {user_id} 没有注册个人webhook URL，无法发送私聊消息")
|
||
return False
|
||
|
||
# 创建MessageSender实例（使用用户的个人URL）
|
||
sender = MessageSender(webhook_url=webhook_url)
|
||
|
||
try:
|
||
# 根据msg_type调用相应方法
|
||
if msg_type == 'markdown':
|
||
return await sender.send_markdown(content)
|
||
else:
|
||
return await sender.send_text(content)
|
||
except Exception as e:
|
||
logger.error(f"发送私聊消息失败: user_id={user_id}, error={e}", exc_info=True)
|
||
return False
|
||
finally:
|
||
# 关闭HTTP客户端
|
||
await sender.close()
|