46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from enum import Enum
|
|
|
|
import json
|
|
|
|
from message.request_message import send_request
|
|
|
|
webhook_url = 'https://qyapi.weixin.qq.com/cgi-bin/webhook/send'
|
|
webhook_key_estun = 'f09eb098-f709-4fc6-83b5-4a08a8431b8f'
|
|
webhook_key_sweet_hut = '5ccae111-43a1-4672-a94f-0d34713259c0'
|
|
|
|
|
|
class MessageEnum(Enum):
|
|
TEXT = 'text'
|
|
MARKDOWN = 'markdown'
|
|
MARKDOWN2 = 'markdown2'
|
|
|
|
|
|
def send_wechat_message(message_type: MessageEnum, message: str, has_sweet_hut=False):
|
|
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}}
|
|
|
|
send_request(
|
|
url=webhook_url,
|
|
method="POST",
|
|
headers={"Content-Type": "application/json"},
|
|
params={'key': webhook_key_estun},
|
|
data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8'),
|
|
timeout=5
|
|
)
|
|
|
|
if has_sweet_hut:
|
|
send_request(
|
|
url=webhook_url,
|
|
method="POST",
|
|
headers={"Content-Type": "application/json"},
|
|
params={'key': webhook_key_sweet_hut},
|
|
data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8'),
|
|
timeout=5
|
|
)
|