fix webpush api

app/modules/webpush/__init__.py

@@ -0,0 +1,66 @@
+import json
+from typing import Union, Tuple
+
+from pywebpush import webpush, WebPushException
+
+from app.core.config import global_vars, settings
+from app.log import logger
+from app.modules import _ModuleBase, checkMessage
+from app.schemas import MessageChannel, Notification
+
+
+class WebPushModule(_ModuleBase):
+    def init_module(self) -> None:
+        pass
+
+    @staticmethod
+    def get_name() -> str:
+        return "VoceChat"
+
+    def stop(self):
+        pass
+
+    def test(self) -> Tuple[bool, str]:
+        """
+        测试模块连接性
+        """
+        return True, ""
+
+    def init_setting(self) -> Tuple[str, Union[str, bool]]:
+        pass
+
+    @checkMessage(MessageChannel.WebPush)
+    def post_message(self, message: Notification) -> None:
+        """
+        发送消息
+        :param message: 消息内容
+        :return: 成功或失败
+        """
+        if not message.title and not message.text:
+            logger.warn("标题和内容不能同时为空")
+            return
+        try:
+            if message.title:
+                caption = message.title
+                content = message.text
+            else:
+                caption = message.text
+                content = ""
+            try:
+                for sub in global_vars.get_subscriptions():
+                    webpush(
+                        subscription_info=sub,
+                        data=json.dumps({
+                            "title": caption,
+                            "body": content
+                        }),
+                        vapid_private_key=settings.VAPID.get("privateKey"),
+                        vapid_claims={
+                            "sub": settings.VAPID.get("subject")
+                        },
+                    )
+            except WebPushException as err:
+                print("WebPush Error:", str(err))
+
+        except Exception as msg_e:
+            logger.error(f"发送消息失败：{msg_e}")