380 lines
9.5 KiB
Python
380 lines
9.5 KiB
Python
|
"""
|
|||
|
|
通知工具
|
|||
|
|
提供发送通知的功能（邮件、Webhook等）
|
|||
|
|
"""
|
|||
|
|
import httpx
|
|||
|
|
from typing import Dict, Any, Optional, List
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
from enum import Enum
|
|||
|
|
|
|||
|
|
|
|||
|
|
class NotificationType(Enum):
|
|||
|
|
"""通知类型"""
|
|||
|
|
EMAIL = "email"
|
|||
|
|
WEBHOOK = "webhook"
|
|||
|
|
SMS = "sms"
|
|||
|
|
DINGTALK = "dingtalk"
|
|||
|
|
WECHAT = "wechat"
|
|||
|
|
SLACK = "slack"
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class NotificationConfig:
|
|||
|
|
"""通知配置"""
|
|||
|
|
# Email配置
|
|||
|
|
smtp_host: str = ""
|
|||
|
|
smtp_port: int = 587
|
|||
|
|
smtp_user: str = ""
|
|||
|
|
smtp_password: str = ""
|
|||
|
|
from_email: str = ""
|
|||
|
|
|
|||
|
|
# Webhook配置
|
|||
|
|
webhook_url: str = ""
|
|||
|
|
webhook_secret: str = ""
|
|||
|
|
|
|||
|
|
# 钉钉配置
|
|||
|
|
dingtalk_webhook: str = ""
|
|||
|
|
|
|||
|
|
# Slack配置
|
|||
|
|
slack_webhook: str = ""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class NotificationTool:
|
|||
|
|
"""
|
|||
|
|
通知工具
|
|||
|
|
|
|||
|
|
支持多种通知渠道：
|
|||
|
|
- Email (SMTP)
|
|||
|
|
- Webhook
|
|||
|
|
- 钉钉
|
|||
|
|
- Slack
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self, config: Optional[NotificationConfig] = None):
|
|||
|
|
self.config = config or NotificationConfig()
|
|||
|
|
|
|||
|
|
async def send_email(
|
|||
|
|
self,
|
|||
|
|
to: str,
|
|||
|
|
subject: str,
|
|||
|
|
body: str,
|
|||
|
|
cc: Optional[List[str]] = None,
|
|||
|
|
is_html: bool = False
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
发送邮件
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
to: 收件人
|
|||
|
|
subject: 主题
|
|||
|
|
body: 内容
|
|||
|
|
cc: 抄送列表
|
|||
|
|
is_html: 是否HTML格式
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
发送结果
|
|||
|
|
"""
|
|||
|
|
if not self.config.smtp_host:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": "Email not configured"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import smtplib
|
|||
|
|
from email.mime.text import MIMEText
|
|||
|
|
from email.mime.multipart import MIMEMultipart
|
|||
|
|
|
|||
|
|
# 构建邮件
|
|||
|
|
msg = MIMEMultipart('alternative')
|
|||
|
|
msg['Subject'] = subject
|
|||
|
|
msg['From'] = self.config.from_email or self.config.smtp_user
|
|||
|
|
msg['To'] = to
|
|||
|
|
|
|||
|
|
if cc:
|
|||
|
|
msg['Cc'] = ",".join(cc)
|
|||
|
|
|
|||
|
|
# 添加内容
|
|||
|
|
content_type = "html" if is_html else "plain"
|
|||
|
|
msg.attach(MIMEText(body, content_type))
|
|||
|
|
|
|||
|
|
# 发送
|
|||
|
|
with smtplib.SMTP(self.config.smtp_host, self.config.smtp_port) as server:
|
|||
|
|
server.starttls()
|
|||
|
|
server.login(self.config.smtp_user, self.config.smtp_password)
|
|||
|
|
server.send_message(msg)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"success": True,
|
|||
|
|
"type": "email",
|
|||
|
|
"to": to,
|
|||
|
|
"subject": subject
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": str(e),
|
|||
|
|
"type": "email"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async def send_webhook(
|
|||
|
|
self,
|
|||
|
|
url: str,
|
|||
|
|
data: Dict[str, Any],
|
|||
|
|
method: str = "POST",
|
|||
|
|
headers: Optional[Dict[str, str]] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
发送Webhook
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
url: Webhook URL
|
|||
|
|
data: 请求数据
|
|||
|
|
method: HTTP方法
|
|||
|
|
headers: 请求头
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
发送结果
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
async with httpx.AsyncClient(timeout=10) as client:
|
|||
|
|
response = await client.request(
|
|||
|
|
method=method,
|
|||
|
|
url=url,
|
|||
|
|
json=data,
|
|||
|
|
headers=headers
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"success": response.status_code < 400,
|
|||
|
|
"status_code": response.status_code,
|
|||
|
|
"type": "webhook",
|
|||
|
|
"url": url
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": str(e),
|
|||
|
|
"type": "webhook"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async def send_dingtalk(
|
|||
|
|
self,
|
|||
|
|
message: str,
|
|||
|
|
webhook: Optional[str] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
发送钉钉消息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
message: 消息内容
|
|||
|
|
webhook: 自定义webhook URL
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
发送结果
|
|||
|
|
"""
|
|||
|
|
url = webhook or self.config.dingtalk_webhook
|
|||
|
|
if not url:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": "Dingtalk webhook not configured"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
async with httpx.AsyncClient(timeout=10) as client:
|
|||
|
|
response = await client.post(
|
|||
|
|
url,
|
|||
|
|
json={
|
|||
|
|
"msgtype": "text",
|
|||
|
|
"text": {
|
|||
|
|
"content": message
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
result = response.json()
|
|||
|
|
return {
|
|||
|
|
"success": result.get("errcode") == 0,
|
|||
|
|
"type": "dingtalk",
|
|||
|
|
"response": result
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": str(e),
|
|||
|
|
"type": "dingtalk"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async def send_slack(
|
|||
|
|
self,
|
|||
|
|
message: str,
|
|||
|
|
channel: Optional[str] = None,
|
|||
|
|
webhook: Optional[str] = None
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
发送Slack消息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
message: 消息内容
|
|||
|
|
channel: 频道
|
|||
|
|
webhook: 自定义webhook URL
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
发送结果
|
|||
|
|
"""
|
|||
|
|
url = webhook or self.config.slack_webhook
|
|||
|
|
if not url:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": "Slack webhook not configured"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
payload = {"text": message}
|
|||
|
|
if channel:
|
|||
|
|
payload["channel"] = channel
|
|||
|
|
|
|||
|
|
async with httpx.AsyncClient(timeout=10) as client:
|
|||
|
|
response = await client.post(url, json=payload)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"success": response.status_code == 200,
|
|||
|
|
"type": "slack",
|
|||
|
|
"status_code": response.status_code
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": str(e),
|
|||
|
|
"type": "slack"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async def send(
|
|||
|
|
self,
|
|||
|
|
type: str,
|
|||
|
|
message: str,
|
|||
|
|
**kwargs
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""
|
|||
|
|
统一发送接口
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
type: 通知类型 (email, webhook, dingtalk, slack)
|
|||
|
|
message: 消息内容
|
|||
|
|
**kwargs: 其他参数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
发送结果
|
|||
|
|
"""
|
|||
|
|
type = type.lower()
|
|||
|
|
|
|||
|
|
if type == "email":
|
|||
|
|
return await self.send_email(
|
|||
|
|
to=kwargs.get("to", ""),
|
|||
|
|
subject=kwargs.get("subject", "Notification"),
|
|||
|
|
body=message,
|
|||
|
|
cc=kwargs.get("cc")
|
|||
|
|
)
|
|||
|
|
elif type == "webhook":
|
|||
|
|
return await self.send_webhook(
|
|||
|
|
url=kwargs.get("url", ""),
|
|||
|
|
data=kwargs.get("data", {"message": message})
|
|||
|
|
)
|
|||
|
|
elif type == "dingtalk":
|
|||
|
|
return await self.send_dingtalk(
|
|||
|
|
message=message,
|
|||
|
|
webhook=kwargs.get("webhook")
|
|||
|
|
)
|
|||
|
|
elif type == "slack":
|
|||
|
|
return await self.send_slack(
|
|||
|
|
message=message,
|
|||
|
|
channel=kwargs.get("channel"),
|
|||
|
|
webhook=kwargs.get("webhook")
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
return {
|
|||
|
|
"success": False,
|
|||
|
|
"error": f"Unknown notification type: {type}"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 全局通知工具
|
|||
|
|
notification_tool = NotificationTool()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 便捷函数
|
|||
|
|
async def send_notification(
|
|||
|
|
type: str,
|
|||
|
|
message: str,
|
|||
|
|
**kwargs
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""发送通知"""
|
|||
|
|
return await notification_tool.send(type, message, **kwargs)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def send_email(
|
|||
|
|
to: str,
|
|||
|
|
subject: str,
|
|||
|
|
body: str
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""发送邮件"""
|
|||
|
|
return await notification_tool.send_email(to, subject, body)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def send_webhook(
|
|||
|
|
url: str,
|
|||
|
|
data: Dict[str, Any]
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""发送Webhook"""
|
|||
|
|
return await notification_tool.send_webhook(url, data)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 工具定义
|
|||
|
|
SEND_NOTIFICATION_TOOL = {
|
|||
|
|
"name": "send_notification",
|
|||
|
|
"description": "Send notifications via email, webhook, dingtalk, or slack.",
|
|||
|
|
"parameters": {
|
|||
|
|
"type": "object",
|
|||
|
|
"properties": {
|
|||
|
|
"type": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "Notification type: email, webhook, dingtalk, slack",
|
|||
|
|
"enum": ["email", "webhook", "dingtalk", "slack"]
|
|||
|
|
},
|
|||
|
|
"message": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "The notification message"
|
|||
|
|
},
|
|||
|
|
"to": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "For email: recipient email address"
|
|||
|
|
},
|
|||
|
|
"subject": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "For email: email subject"
|
|||
|
|
},
|
|||
|
|
"url": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "For webhook: webhook URL"
|
|||
|
|
},
|
|||
|
|
"data": {
|
|||
|
|
"type": "object",
|
|||
|
|
"description": "For webhook: JSON data to send"
|
|||
|
|
},
|
|||
|
|
"webhook": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "Custom webhook URL for dingtalk/slack"
|
|||
|
|
},
|
|||
|
|
"channel": {
|
|||
|
|
"type": "string",
|
|||
|
|
"description": "For slack: channel name"
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
"required": ["type", "message"]
|
|||
|
|
}
|
|||
|
|
}
|