feat:重构代码模块

message/wechat_message.py

@@ -0,0 +1,40 @@
+from enum import Enum
+
+import requests
+import json
+
+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}}
+
+    requests.post(
+        webhook_url,
+        headers={"Content-Type": "application/json"},
+        params={'key': webhook_key_estun},
+        data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8')
+    )
+
+    if has_sweet_hut:
+        requests.post(
+            webhook_url,
+            headers={"Content-Type": "application/json"},
+            params={'key': webhook_key_sweet_hut},
+            data=json.dumps(wechat_message, ensure_ascii=False).encode('utf-8')
+        )