summaryrefslogtreecommitdiff
path: root/ATRI/plugins/essential.py
diff options
context:
space:
mode:
authorKyomotoi <1172294279@qq.com>2021-02-06 00:32:26 +0800
committerKyomotoi <1172294279@qq.com>2021-02-06 00:32:26 +0800
commitf5ceb8927f2e7f2a9e29d62c8e4cef876f917249 (patch)
tree40b9dcd6b7d3db486054e3aa9b5a04d25fa2284e /ATRI/plugins/essential.py
parenteb52fab79ada7efe6191e3a5f90179766feaded0 (diff)
downloadATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.gz
ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.tar.bz2
ATRI-f5ceb8927f2e7f2a9e29d62c8e4cef876f917249.zip
🏗 💩 更改项目结构，修复啥b BUG
Diffstat (limited to 'ATRI/plugins/essential.py')
-rw-r--r--ATRI/plugins/essential.py297
1 files changed, 297 insertions, 0 deletions
diff --git a/ATRI/plugins/essential.py b/ATRI/plugins/essential.py
new file mode 100644
index 0000000..22f641f
--- /dev/null
+++ b/ATRI/plugins/essential.py
@@ -0,0 +1,297 @@
+import time
+import json
+import shutil
+from pathlib import Path
+from random import choice
+
+from nonebot.adapters import Bot
+from nonebot.plugin import on_notice, on_request
+from nonebot.adapters.cqhttp.message import Message
+from nonebot.adapters.cqhttp import (
+ FriendRequestEvent,
+ GroupRequestEvent,
+ GroupIncreaseNoticeEvent,
+ GroupDecreaseNoticeEvent,
+ GroupAdminNoticeEvent,
+ GroupBanNoticeEvent,
+ LuckyKingNotifyEvent,
+ GroupUploadNoticeEvent,
+ GroupRecallNoticeEvent,
+ FriendRecallNoticeEvent
+)
+
+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
+
+
+PLUGIN_INFO_DIR = Path('.') / 'ATRI' / 'data' / 'service' / 'plugins'
+
+@driver.on_startup
+async def startup() -> None:
+ logger.info("アトリは、高性能ですから！")
+
+
+@driver.on_shutdown
+async def shutdown() -> None:
+ logger.info("Thanks for using.")
+ logger.debug("bot已停止运行，正在清理插件信息...")
+ try:
+ shutil.rmtree(PLUGIN_INFO_DIR)
+ logger.debug("成功！")
+ except:
+ repo = (
+ '清理插件信息失败',
+ '请前往 ATRI/data/service 下',
+ '将 plugins 整个文件夹删除'
+ )
+ logger.error(repo)
+ time.sleep(10)
+ pass
+
+
+@driver.on_bot_connect
+async def connect(bot) -> None:
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ int(superuser),
+ "WebSocket 成功连接，数据开始传输。"
+ )
+
+
+@driver.on_bot_disconnect
+async def disconnect(bot) -> None:
+ for superuser in nonebot_config["superusers"]:
+ try:
+ await HttpPost.send_private_msg(
+ int(superuser),
+ "WebSocket 貌似断开了呢..."
+ )
+ except:
+ logger.error("WebSocket 已断开，等待重连")
+
+
+ESSENTIAL_DIR = Path('.') / 'ATRI' / 'data' / 'database' / 'essential'
+
+# 处理：好友请求
+request_friend_event = on_request(rule=is_in_banlist())
+
+@request_friend_event.handle()
+async def _request_friend_event(bot, event: FriendRequestEvent) -> None:
+ file_name = "request_friend.json"
+ path = ESSENTIAL_DIR / file_name
+ path.parent.mkdir(exist_ok=True, parents=True)
+
+ try:
+ data = json.loads(path.read_bytes())
+ except FileExistsError:
+ data = {}
+ data[event.flag] = {
+ "user_id": event.user_id,
+ "comment": event.comment
+ }
+ try:
+ with open(path, 'w', encoding='utf-8') as r:
+ r.write(
+ json.dumps(
+ data, indent=4
+ )
+ )
+ except WriteError:
+ raise WriteError("Writing file failed!")
+
+ for superuser in nonebot_config["superusers"]:
+ msg = (
+ "主人，收到一条好友请求：\n"
+ f"请求人：{event.get_user_id()}\n"
+ f"申请信息：{event.comment}\n"
+ f"申请码：{event.flag}"
+ )
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+
+
+# 处理：邀请入群，如身为管理，还附有入群请求
+request_group_event = on_request(rule=is_in_banlist())
+
+@request_group_event.handle()
+async def _request_group_event(bot, event: GroupRequestEvent) -> None:
+ file_name = "request_group.json"
+ path = ESSENTIAL_DIR / file_name
+ path.parent.mkdir(exist_ok=True, parents=True)
+
+ try:
+ data = json.loads(path.read_bytes())
+ except FileExistsError:
+ data = {}
+ data[event.flag] = {
+ "user_id": event.user_id,
+ "group_id": event.group_id,
+ "sub_type": event.sub_type,
+ "comment": event.comment
+ }
+ try:
+ with open(path, 'w', encoding='utf-8') as r:
+ r.write(
+ json.dumps(
+ data, indent=4
+ )
+ )
+ except WriteError:
+ raise WriteError("Writing file failed!")
+
+ for superuser in nonebot_config["superusers"]:
+ msg = (
+ "主人，收到一条入群请求：\n"
+ f"请求人：{event.get_user_id()}\n"
+ f"申请信息：{event.comment}\n"
+ f"申请码：{event.flag}"
+ )
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+
+
+# 处理群成员变动
+group_member_event = on_notice()
+
+@group_member_event.handle()
+async def _group_member_event(bot: Bot, event) -> None:
+ if isinstance(event, GroupIncreaseNoticeEvent):
+ msg = (
+ "好欸！事新人！\n"
+ f"在下 {choice(list(nonebot_config['nickname']))} 哒!w!"
+ )
+ await group_member_event.finish(msg)
+
+ elif isinstance(event, GroupDecreaseNoticeEvent):
+ if event.is_tome():
+ msg = (
+ "呜呜呜，主人"
+ f"咱被群 {event.group_id} 里的 {event.operator_id} 扔出来了..."
+ )
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+ else:
+ await group_member_event.finish(f"{event.user_id} 离开了我们...")
+
+
+# 处理群管理事件
+group_admin_event = on_notice()
+
+@group_admin_event.handle()
+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(
+ user_id=int(superuser),
+ message=f"好欸！主人！我在群 {event.group_id} 成为了管理！！"
+ )
+
+
+# 处理群禁言事件
+group_ban_event = on_notice()
+
+@group_ban_event.handle()
+async def _group_ban_event(bot: Bot, event: GroupBanNoticeEvent) -> None:
+ if event.is_tome():
+ if event.duration:
+ msg = (
+ "那个..。，主人\n"
+ f"咱在群 {event.group_id} 被 {event.operator_id} 塞上了口球...\n"
+ f"时长...是 {event.duration} 秒"
+ )
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+ else:
+ msg = (
+ "好欸！主人\n"
+ f"咱在群 {event.group_id} 被 {event.operator_id} 上的口球解除了！"
+ )
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+
+
+# 处理群红包运气王事件
+lucky_read_bag_event = on_notice()
+
+@lucky_read_bag_event.handle()
+async def _lucky_read_bag_event(bot, event: LuckyKingNotifyEvent) -> None:
+ msg = (
+ "8行，这可忍？"
+ f"gkd [CQ:at,qq={event.user_id}] 发一个！"
+ )
+ await lucky_read_bag_event.finish(Message(msg))
+
+
+# 处理群文件上传事件
+group_file_upload_event = on_notice()
+
+@group_file_upload_event.handle()
+async def _group_file_upload_event(bot, event: GroupUploadNoticeEvent) -> None:
+ await group_file_upload_event.finish("让我康康传了啥好东西")
+
+
+# 处理撤回事件
+recall_event = on_notice()
+
+@recall_event.handle()
+async def _recall_event(bot: Bot, event) -> None:
+ if isinstance(event, GroupRecallNoticeEvent):
+ repo = await bot.call_api(
+ "get_msg",
+ message_id=event.message_id
+ )
+ repo = str(repo["message"])
+ if "CQ" in repo:
+ repo = repo.replace("CQ", "QC")
+
+ msg = (
+ "主人，咱拿到了一条撤回信息！\n"
+ f"{event.user_id}@[群:{event.group_id}]\n"
+ "撤回了\n"
+ f"{repo}"
+ )
+
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
+
+ elif isinstance(event, FriendRecallNoticeEvent):
+ repo = await bot.call_api(
+ "get_msg",
+ message_id=event.message_id
+ )
+ repo = str(repo["message"])
+ if "CQ" in repo:
+ repo = repo.replace("CQ", "QC")
+
+ msg = (
+ "主人，咱拿到了一条撤回信息！\n"
+ f"{event.user_id}@[私聊]"
+ "撤回了\n"
+ f"{repo}"
+ )
+
+ for superuser in nonebot_config["superusers"]:
+ await HttpPost.send_private_msg(
+ user_id=int(superuser),
+ message=msg
+ )
Powered by cgit, Theme cgit-dark created by Ted Yin, updated by Mole Shang.