feat:增加批量插入和查询日志告警功能
This commit is contained in:
32
service/message_service.py
Normal file
32
service/message_service.py
Normal file
@@ -0,0 +1,32 @@
|
||||
import json
|
||||
from enum import Enum
|
||||
|
||||
import requests
|
||||
|
||||
from config.setting import settings
|
||||
|
||||
|
||||
class MessageEnum(Enum):
|
||||
TEXT = 'text'
|
||||
MARKDOWN = 'markdown'
|
||||
MARKDOWN2 = 'markdown2'
|
||||
|
||||
|
||||
def send_wechat_message(message_type: MessageEnum, message: str):
|
||||
wechat_message = {}
|
||||
match message_type:
|
||||
case MessageEnum.TEXT:
|
||||
wechat_message = {"msgtype": "text", "text": {"content": message}}
|
||||
case MessageEnum.MARKDOWN:
|
||||
wechat_message = {"msgtype": "markdown", "markdown": {"content": message}}
|
||||
case MessageEnum.MARKDOWN2:
|
||||
wechat_message = {"msgtype": "markdown_v2", "markdown_v2": {"content": message}}
|
||||
|
||||
requests.post(
|
||||
settings.wechat_webhook_url,
|
||||
headers={"Content-Type": "application/json"},
|
||||
params={'key': settings.wechat_webhook_key},
|
||||
data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8')
|
||||
)
|
||||
|
||||
return True
|
||||
Reference in New Issue
Block a user