zhenxun_bot/basic_plugins/init_plugin_config/init_plugins_limit.py

from utils.manager import (
    plugins2cd_manager,
    plugins2block_manager,
    plugins2count_manager,
)
from utils.utils import get_matchers
from configs.path_config import DATA_PATH
import nonebot


def init_plugins_cd_limit():
    """
    加载 cd 限制
    """
    plugins2cd_file = DATA_PATH / "configs" / "plugins2cd.yaml"
    plugins2cd_file.parent.mkdir(exist_ok=True, parents=True)
    _data = {}
    for matcher in get_matchers(True):
        if not plugins2cd_manager.get_plugin_cd_data(matcher.plugin_name):
            _plugin = nonebot.plugin.get_plugin(matcher.plugin_name)
            try:
                _module = _plugin.module
                plugin_cd_limit = _module.__getattribute__("__plugin_cd_limit__")
                plugins2cd_manager.add_cd_limit(
                    matcher.plugin_name, **plugin_cd_limit
                )
            except AttributeError:
                pass
    if not plugins2cd_manager.keys():
        plugins2cd_manager.add_cd_limit(
            "这是一个示例"
        )
    plugins2cd_manager.save()
    plugins2cd_manager.reload_cd_limit()


def init_plugins_block_limit():
    """
    加载阻塞限制
    """
    plugins2block_file = DATA_PATH / "configs" / "plugins2block.yaml"
    plugins2block_file.parent.mkdir(exist_ok=True, parents=True)
    for matcher in get_matchers(True):
        if not plugins2block_manager.get_plugin_block_data(matcher.plugin_name):
            _plugin = matcher.plugin
            try:
                _module = _plugin.module
                plugin_block_limit = _module.__getattribute__("__plugin_block_limit__")
                plugins2block_manager.add_block_limit(
                    matcher.plugin_name, **plugin_block_limit
                )
            except AttributeError:
                pass
    if not plugins2block_manager.keys():
        plugins2block_manager.add_block_limit(
            "这是一个示例"
        )
    plugins2block_manager.save()
    plugins2block_manager.reload_block_limit()


def init_plugins_count_limit():
    """
    加载次数限制
    """
    plugins2count_file = DATA_PATH / "configs" / "plugins2count.yaml"
    plugins2count_file.parent.mkdir(exist_ok=True, parents=True)
    _data = {}
    _matchers = get_matchers()
    for matcher in _matchers:
        if not plugins2count_manager.get_plugin_count_data(matcher.plugin_name):
            _plugin = nonebot.plugin.get_plugin(matcher.plugin_name)
            try:
                _module = _plugin.module
                plugin_count_limit = _module.__getattribute__("__plugin_count_limit__")
                plugins2count_manager.add_count_limit(
                    matcher.plugin_name, **plugin_count_limit
                )
            except AttributeError:
                pass
    if not plugins2count_manager.keys():
        plugins2count_manager.add_count_limit(
            "这是一个示例"
        )
    plugins2count_manager.save()
    plugins2count_manager.reload_count_limit()