From da888ff020805a38a17e5f83705aeb42ffa992ba Mon Sep 17 00:00:00 2001 From: Kyomotoi <1172294279@qq.com> Date: Sun, 7 Mar 2021 15:24:17 +0800 Subject: =?UTF-8?q?=E2=99=BB=EF=B8=8F=E2=9A=A1=EF=B8=8F=20=E9=87=8D?= =?UTF-8?q?=E6=9E=84=20Service=EF=BC=8C=E4=BC=98=E5=8C=96=E9=83=A8?= =?UTF-8?q?=E5=88=86=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/essential.py | 43 +++++++++++++++++++++++++++++++------------ 1 file changed, 31 insertions(+), 12 deletions(-) (limited to 'ATRI/plugins/essential.py') diff --git a/ATRI/plugins/essential.py b/ATRI/plugins/essential.py index a8ba021..f7d9840 100644 --- a/ATRI/plugins/essential.py +++ b/ATRI/plugins/essential.py @@ -1,9 +1,26 @@ +#!/usr/bin/env python3 +# -*- coding:utf-8 -*- +''' +File: essential.py +Created Date: 2021-02-04 12:53:00 +Author: Kyomotoi +Email: Kyomotoiowo@gmail.com +License: GPLv3 +Project: https://github.com/Kyomotoi/ATRI +-------- +Last Modified: Sunday, 7th March 2021 3:13:09 pm +Modified By: Kyomotoi (kyomotoiowo@gmail.com) +-------- +Copyright (c) 2021 Kyomotoi +''' + import time import json import shutil from pathlib import Path from random import choice +import nonebot from nonebot.adapters import Bot from nonebot.plugin import on_notice, on_request from nonebot.adapters.cqhttp.message import Message @@ -24,12 +41,14 @@ from ATRI.log import logger from ATRI.exceptions import WriteError from ATRI.config import nonebot_config from ATRI.rule import is_in_banlist -from ATRI.service.httppost import HttpPost -from main import driver +from ATRI.service import Service as sv PLUGIN_INFO_DIR = Path('.') / 'ATRI' / 'data' / 'service' / 'plugins' + +driver = nonebot.get_driver() + @driver.on_startup async def startup() -> None: logger.info("アトリは、高性能ですから！") @@ -56,7 +75,7 @@ async def shutdown() -> None: @driver.on_bot_connect async def connect(bot) -> None: for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( int(superuser), "WebSocket 成功连接，数据开始传输。" ) @@ -66,7 +85,7 @@ async def connect(bot) -> None: async def disconnect(bot) -> None: for superuser in nonebot_config["superusers"]: try: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( int(superuser), "WebSocket 貌似断开了呢..." ) @@ -110,7 +129,7 @@ async def _request_friend_event(bot, event: FriendRequestEvent) -> None: f"申请信息：{event.comment}\n" f"申请码：{event.flag}" ) - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -152,7 +171,7 @@ async def _request_group_event(bot, event: GroupRequestEvent) -> None: f"申请信息：{event.comment}\n" f"申请码：{event.flag}" ) - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -177,7 +196,7 @@ async def _group_member_event(bot: Bot, event) -> None: f"咱被群 {event.group_id} 里的 {event.operator_id} 扔出来了..." ) for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -192,7 +211,7 @@ group_admin_event = on_notice() async def _group_admin_event(bot: Bot, event: GroupAdminNoticeEvent) -> None: if event.is_tome(): for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=f"好欸！主人！我在群 {event.group_id} 成为了管理！！" ) @@ -211,7 +230,7 @@ async def _group_ban_event(bot: Bot, event: GroupBanNoticeEvent) -> None: f"时长...是 {event.duration} 秒" ) for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -221,7 +240,7 @@ async def _group_ban_event(bot: Bot, event: GroupBanNoticeEvent) -> None: f"咱在群 {event.group_id} 被 {event.operator_id} 上的口球解除了！" ) for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -271,7 +290,7 @@ async def _recall_event(bot: Bot, event) -> None: await bot.send(event, "咱看到惹~！") for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) @@ -293,7 +312,7 @@ async def _recall_event(bot: Bot, event) -> None: ) for superuser in nonebot_config["superusers"]: - await HttpPost.send_private_msg( + await sv.NetworkPost.send_private_msg( user_id=int(superuser), message=msg ) -- cgit v1.2.3