33 lines
927 B
Python
33 lines
927 B
Python
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
|