summaryrefslogtreecommitdiff
path: root/ATRI/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'ATRI/__init__.py')
-rw-r--r--ATRI/__init__.py29
1 files changed, 23 insertions, 6 deletions
diff --git a/ATRI/__init__.py b/ATRI/__init__.py
index 23d4d2b..bfd6133 100644
--- a/ATRI/__init__.py
+++ b/ATRI/__init__.py
@@ -7,16 +7,17 @@
@Author : Kyomotoi
@Contact : [email protected]
@Github : https://github.com/Kyomotoi
-@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
+@License : Copyright © 2021 Kyomotoi, All Rights Reserved.
'''
__author__ = 'kyomotoi'
+import time
import shutil
from pathlib import Path
from .log import logger
-from .config import check_config
-from .service.send import Send
+from .config import check_config, RUNTIME_CONFIG
+from .service.httppost import HttpPost
from main import driver
@@ -29,15 +30,31 @@ async def startup_event() -> None:
@driver.on_shutdown
async def shutdown_event() -> None:
- logger.info('Thanks for using!')
+ logger.info('Thanks for using.')
+ logger.debug('bot已关闭,正在清理插件信息...')
try:
shutil.rmtree(PLUGIN_INFO_PATH)
except:
+ repo = ('清理插件信息失败',
+ '请前往 ATRI/data/service 下',
+ '将 plugins 整个文件夹删除')
+ logger.error(repo)
+ time.sleep(10)
pass
@driver.on_bot_connect
-async def az(bot):
- await Send.send_to_superuser('test')
+async def connect(bot):
+ for super in RUNTIME_CONFIG['superusers']:
+ await HttpPost.send_private_msg(
+ int(super), 'WebSocket 成功连接,数据开始传输~!')
+
[email protected]_bot_disconnect
+async def disconnect(bot):
+ for super in RUNTIME_CONFIG['superusers']:
+ try:
+ await HttpPost.send_private_msg(int(super), 'WebSocket 貌似断开了呢...')
+ except:
+ logger.error('WebSocket 已断开,等待重连')
from .exceptions import Error as Error