zhenxun_bot/utils/manager/plugin_data_manager.py

from typing import Any, Dict, Optional

from . import StaticData
from .models import PluginData


class PluginDataManager(StaticData[PluginData]):
    """
    插件所有信息管理
    """

    def __init__(self):
        super().__init__(None)
        self._data: Dict[str, PluginData] = {}

    def add_plugin_info(self, info: PluginData):
        """
        说明:
            添加插件信息
        参数:
            :param info: PluginInfo
        """
        if info.model in self._data.keys() and self._data[info.model] == info:
            raise ValueError(f"PluginInfoManager {info.model}:{info.name} 插件名称及类型已存在")
        self._data[info.model] = info

    def get(self, item: str, default: Any = None) -> Optional[PluginData]:
        return self._data.get(item, default)

    def __getitem__(self, item) -> Optional[PluginData]:
        return self._data.get(item)