add 消息类型&API

app/api/endpoints/message.py

@@ -1,14 +1,20 @@
-from typing import Union, Any
+from typing import Union, Any, List
 
-from fastapi import APIRouter, BackgroundTasks
+from fastapi import APIRouter, BackgroundTasks, Depends
 from fastapi import Request
+from sqlalchemy.orm import Session
 from starlette.responses import PlainTextResponse
 
 from app import schemas
 from app.chain.message import MessageChain
 from app.core.config import settings
+from app.core.security import verify_token
+from app.db import get_db
+from app.db.systemconfig_oper import SystemConfigOper
 from app.log import logger
 from app.modules.wechat.WXBizMsgCrypt3 import WXBizMsgCrypt
+from app.schemas import Notification
+from app.schemas.types import SystemConfigKey, NotificationType
 
 router = APIRouter()
 
@@ -54,3 +60,35 @@ def wechat_verify(echostr: str, msg_signature: str,
         logger.error("微信请求验证失败 VerifyURL ret: %s" % str(ret))
     # 验证URL成功，将sEchoStr返回给企业号
     return PlainTextResponse(sEchoStr)
+
+
+@router.get("/switchs", summary="查询通知消息渠道开关", response_model=List[Notification])
+def read_switchs(db: Session = Depends(get_db),
+                 _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    查询通知消息渠道开关
+    """
+    return_list = []
+    # 读取数据库
+    switchs = SystemConfigOper(db).get(SystemConfigKey.NotificationChannels)
+    if not switchs:
+        for noti in NotificationType:
+            return_list.append(Notification(mtype=noti.value, switch=True))
+
+    return return_list
+
+
+@router.put("/switchs", summary="设置通知消息渠道开关", response_model=schemas.Response)
+def set_switchs(switchs: List[Notification],
+                db: Session = Depends(get_db),
+                _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    查询通知消息渠道开关
+    """
+    switch_list = []
+    for switch in switchs:
+        switch_list.append(switch.dict())
+    # 存入数据库
+    SystemConfigOper(db).set(SystemConfigKey.NotificationChannels, switch_list)
+
+    return schemas.Response(success=True)

app/api/endpoints/plugin.py

@@ -1,10 +1,12 @@
 from typing import Any, List
 
 from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
 
 from app import schemas
 from app.core.plugin import PluginManager
 from app.core.security import verify_token
+from app.db import get_db
 from app.db.systemconfig_oper import SystemConfigOper
 from app.schemas.types import SystemConfigKey
 
@@ -20,11 +22,12 @@ def all_plugins(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
 
 
 @router.get("/installed", summary="已安装插件", response_model=List[str])
-def installed_plugins(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
+def installed_plugins(db: Session = Depends(get_db),
+                      _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     查询用户已安装插件清单
     """
-    return SystemConfigOper().get(SystemConfigKey.UserInstalledPlugins) or []
+    return SystemConfigOper(db).get(SystemConfigKey.UserInstalledPlugins) or []
 
 
 @router.get("/{plugin_id}", summary="获取插件配置")
@@ -47,34 +50,37 @@ def set_plugin_config(plugin_id: str, conf: dict,
 
 @router.post("/{plugin_id}/install", summary="安装插件", response_model=schemas.Response)
 def install_plugin(plugin_id: str,
+                   db: Session = Depends(get_db),
                    _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     安装插件
     """
     # 已安装插件
-    install_plugins = SystemConfigOper().get(SystemConfigKey.UserInstalledPlugins) or []
+    install_plugins = SystemConfigOper(db).get(SystemConfigKey.UserInstalledPlugins) or []
     # 安装插件
     install_plugins.append(plugin_id)
     # 保存设置
-    SystemConfigOper().set(SystemConfigKey.UserInstalledPlugins, install_plugins)
+    SystemConfigOper(db).set(SystemConfigKey.UserInstalledPlugins, install_plugins)
     # 重载插件管理器
     PluginManager().init_config()
     return schemas.Response(success=True)
 
 
 @router.delete("/{plugin_id}", summary="卸载插件", response_model=schemas.Response)
-def uninstall_plugin(plugin_id: str, _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+def uninstall_plugin(plugin_id: str,
+                     db: Session = Depends(get_db),
+                     _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     卸载插件
     """
     # 删除已安装信息
-    install_plugins = SystemConfigOper().get(SystemConfigKey.UserInstalledPlugins) or []
+    install_plugins = SystemConfigOper(db).get(SystemConfigKey.UserInstalledPlugins) or []
     for plugin in install_plugins:
         if plugin == plugin_id:
             install_plugins.remove(plugin)
             break
     # 保存
-    SystemConfigOper().set(SystemConfigKey.UserInstalledPlugins, install_plugins)
+    SystemConfigOper(db).set(SystemConfigKey.UserInstalledPlugins, install_plugins)
     # 重载插件管理器
     PluginManager().init_config()
     return schemas.Response(success=True)

app/api/endpoints/system.py

@@ -5,9 +5,11 @@ from typing import Any, List, Union
 
 from fastapi import APIRouter, HTTPException, Depends
 from fastapi.responses import StreamingResponse
+from sqlalchemy.orm import Session
 
 from app import schemas
 from app.core.security import verify_token
+from app.db import get_db
 from app.db.systemconfig_oper import SystemConfigOper
 from app.helper.message import MessageHelper
 from app.helper.progress import ProgressHelper
@@ -38,22 +40,25 @@ def get_progress(process_type: str, token: str):
 
 
 @router.get("/setting/{key}", summary="查询系统设置", response_model=schemas.Response)
-def get_setting(key: str, _: schemas.TokenPayload = Depends(verify_token)):
+def get_setting(key: str,
+                db: Session = Depends(get_db),
+                _: schemas.TokenPayload = Depends(verify_token)):
     """
     查询系统设置
     """
     return schemas.Response(success=True, data={
-        "value": SystemConfigOper().get(key)
+        "value": SystemConfigOper(db).get(key)
     })
 
 
 @router.post("/setting/{key}", summary="更新系统设置", response_model=schemas.Response)
 def set_setting(key: str, value: Union[list, dict, str, int],
+                db: Session = Depends(get_db),
                 _: schemas.TokenPayload = Depends(verify_token)):
     """
     更新系统设置
     """
-    SystemConfigOper().set(key, value)
+    SystemConfigOper(db).set(key, value)
     return schemas.Response(success=True)