From 62a3aa46441d9f2f2e5e4857c6769d7ac99e14d9 Mon Sep 17 00:00:00 2001 From: Kyomotoi <0w0@imki.moe> Date: Fri, 4 Nov 2022 10:13:25 +0800 Subject: =?UTF-8?q?=E2=9C=A8=20=E4=B8=BA=E6=9C=8D=E5=8A=A1=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E6=96=B0=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/service.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/ATRI/service.py b/ATRI/service.py index 6562482..ff7b1e5 100644 --- a/ATRI/service.py +++ b/ATRI/service.py @@ -5,6 +5,7 @@ from types import ModuleType from pydantic import BaseModel from typing import List, Set, Tuple, Type, Union, Optional +from nonebot import get_bot from nonebot.matcher import Matcher from nonebot.dependencies import Dependent from nonebot.typing import ( @@ -15,9 +16,9 @@ from nonebot.typing import ( ) from nonebot.rule import Rule, command, keyword, regex from nonebot.adapters import Bot, Event -from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent +from nonebot.adapters.onebot.v11 import Message, PrivateMessageEvent, GroupMessageEvent -from ATRI.permission import MASTER, Permission +from ATRI.permission import MASTER, Permission, MASTER_LIST from ATRI.exceptions import ReadFileError, WriteFileError @@ -82,7 +83,7 @@ class Service: self._main_cmd = (str(),) self._path = Path(".") / "data" / "plugins" / self.service - self._path.mkdir(parents=True, exist_ok=True) + # self._path.mkdir(parents=True, exist_ok=True) def document(self, context: str) -> "Service": """为服务添加说明""" @@ -358,6 +359,12 @@ class Service: del kwargs["aliases"] return self.on_command(_cmd, docs, **kwargs) + + @staticmethod + async def send_to_master(message: Union[str, Message]): + bot = get_bot() + for m in MASTER_LIST: + await bot.send_private_msg(user_id=m, message=message) class ServiceTools: -- cgit v1.2.3