From 6826c10a095ae96682781c35bc8dd21814c89a7b Mon Sep 17 00:00:00 2001 From: HibiKier <775757368@qq.com> Date: Mon, 27 Feb 2023 19:44:47 +0800 Subject: [PATCH] fix bug --- utils/manager/group_manager.py | 15 +++++++++------ utils/manager/models.py | 11 +++++++---- utils/manager/plugin_data_manager.py | 8 ++------ 3 files changed, 18 insertions(+), 16 deletions(-) diff --git a/utils/manager/group_manager.py b/utils/manager/group_manager.py index e29ffe5ec..18e641eb7 100644 --- a/utils/manager/group_manager.py +++ b/utils/manager/group_manager.py @@ -1,16 +1,18 @@ import copy -from typing import List, Union, Dict, Callable, Any, Optional from pathlib import Path -from .models import BaseData, BaseGroup -from utils.manager.data_class import StaticData -from utils.utils import get_matchers, is_number -from configs.config import Config +from typing import Any, Callable, Dict, List, Optional, Union + import nonebot import ujson as json +from configs.config import Config +from utils.manager.data_class import StaticData +from utils.utils import get_matchers, is_number + +from .models import BaseData, BaseGroup Config.add_plugin_config( - "group_manager", "DEFAULT_GROUP_LEVEL", 5, help_="默认群权限", default_value=5 + "group_manager", "DEFAULT_GROUP_LEVEL", 5, help_="默认群权限", default_value=5, type=int ) Config.add_plugin_config( @@ -19,6 +21,7 @@ True, help_="默认进群总开关状态", default_value=True, + type=bool ) diff --git a/utils/manager/models.py b/utils/manager/models.py index d3cd88c9b..5ea67b9e9 100644 --- a/utils/manager/models.py +++ b/utils/manager/models.py @@ -1,8 +1,11 @@ +from enum import Enum from pathlib import Path -from typing import List, Optional, Dict, Literal, Tuple, Union, Any +from typing import Any, Dict, List, Literal, Optional, Tuple, Union + from pydantic import BaseModel + from configs.config import Config -from enum import Enum +from configs.utils import Config as zConfig class AdminSetting(BaseModel): @@ -33,7 +36,7 @@ class BaseData(BaseModel): """ white_group: List[int] = [] # 白名单 - close_task: List[str] = [] # 全局关闭的被动任务 + close_task: List[str] = [] # 全局关闭的被动任务 group_manager: Dict[str, BaseGroup] = {} # 群组管理 task: Dict[str, str] = {} # 被动任务 【英文:中文】 @@ -123,7 +126,7 @@ class PluginData(BaseModel): plugin_block: Optional[PluginBlock] plugin_count: Optional[PluginCount] plugin_resources: Optional[Dict[str, Union[str, Path]]] - plugin_configs: Optional[Dict[str, Dict[str, Any]]] + plugin_configs: Optional[Dict[str, zConfig]] plugin_status: Plugin class Config: diff --git a/utils/manager/plugin_data_manager.py b/utils/manager/plugin_data_manager.py index 0e5063d80..093dd0fec 100644 --- a/utils/manager/plugin_data_manager.py +++ b/utils/manager/plugin_data_manager.py @@ -1,4 +1,4 @@ -from typing import Dict, Any, Optional +from typing import Any, Dict, Optional from . import StaticData from .models import PluginData @@ -27,9 +27,5 @@ def add_plugin_info(self, info: PluginData): def get(self, item: str, default: Any = None) -> PluginData: return self._data.get(item, default) - def __getitem__(self, item) -> PluginData: + def __getitem__(self, item) -> Optional[PluginData]: return self._data.get(item) - - - -