diff options
Diffstat (limited to 'ATRI/__init__.py')
-rw-r--r-- | ATRI/__init__.py | 29 |
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 |