from nonebot import on_command from nonebot.permission import SUPERUSER from models.level_user import LevelUser from nonebot.typing import T_State from nonebot.adapters.cqhttp import Bot, MessageEvent, Message from nonebot.rule import to_me from utils.utils import get_message_at, get_message_text, is_number, get_bot, scheduler from services.log import logger from .data_source import open_remind, close_remind from models.group_info import GroupInfo from models.friend_user import FriendUser from utils.message_builder import at from configs.path_config import IMAGE_PATH import asyncio import os __plugin_name__ = "超级用户指令 [Hidden]" __plugin_usage__ = "用法" super_cmd = on_command( "添加管理", aliases={"删除管理", "添加权限", "删除权限"}, rule=to_me(), priority=1, permission=SUPERUSER, block=True, ) oc_gb = on_command( "开启广播通知", aliases={"关闭广播通知"}, rule=to_me(), permission=SUPERUSER, priority=1, block=True, ) cls_group = on_command( "所有群组", rule=to_me(), permission=SUPERUSER, priority=1, block=True ) cls_friend = on_command( "所有好友", rule=to_me(), permission=SUPERUSER, priority=1, block=True ) del_group = on_command("退群", rule=to_me(), permission=SUPERUSER, priority=1, block=True) update_group_info = on_command( "更新群信息", rule=to_me(), permission=SUPERUSER, priority=1, block=True ) update_friend_info = on_command( "更新好友信息", rule=to_me(), permission=SUPERUSER, priority=1, block=True ) clear_data = on_command( "清理数据", rule=to_me(), permission=SUPERUSER, priority=1, block=True ) @super_cmd.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): try: args = get_message_text(event.json()).strip().split(" ") qq = get_message_at(event.json()) flag = -1 if not qq: if len(args) > 2: if is_number(args[0]) and is_number(args[1]) and is_number(args[2]): qq = int(args[0]) group_id = int(args[1]) level = int(args[2]) flag = 1 else: await super_cmd.finish("所有参数必须是数字！", at_sender=True) else: await super_cmd.finish( "权限参数不完全\n\t格式：添加/删除权限 [at] [level]" "\n\t格式：添加/删除权限 [qq] [group] [level]", at_sender=True, ) else: qq = int(qq[0]) group_id = event.group_id flag = 2 if is_number(args[0]): level = int(args[0]) else: await super_cmd.finish("权限等级必须是数字！", at_sender=True) if state["_prefix"]["raw_command"][:2] == "添加": if await LevelUser.set_level(qq, group_id, level, 1): result = "添加管理成功, 权限: " + str(level) else: result = "管理已存在, 更新权限: " + str(level) else: if await LevelUser.delete_level(qq, event.group_id): result = "删除管理成功!" else: result = "该账号无管理权限!" if flag == 2: await super_cmd.send(result) elif flag == 1: await bot.send_group_msg( group_id=group_id, message=Message(f"{at(qq)}管理员修改了你的权限" f"\n--------\n你当前的权限等级：{level}"), ) await super_cmd.send("修改成功") except Exception as e: await super_cmd.send("执行指令失败!") logger.error(f"执行指令失败 e{e}") @oc_gb.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): group = get_message_text(event.json()) if group: if is_number(group): group = int(group) for g in await bot.get_group_list(): if g["group_id"] == group: break else: await oc_gb.finish("没有加入这个群...", at_sender=True) # try: if state["_prefix"]["raw_command"] == "开启广播通知": logger.info(f"USER {event.user_id} 开启了 GROUP {group} 的广播") await oc_gb.finish(await open_remind(group, "gb"), at_sender=True) else: logger.info(f"USER {event.user_id} 关闭了 GROUP {group} 的广播") await oc_gb.finish(await close_remind(group, "gb"), at_sender=True) # except Exception as e: # await oc_gb.finish(f'关闭 {group} 的广播失败', at_sender=True) else: await oc_gb.finish("请输入正确的群号", at_sender=True) else: await oc_gb.finish("请输入要关闭广播的群号", at_sender=True) @del_group.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): group = get_message_text(event.json()) if group: if is_number(group): try: await bot.set_group_leave(group_id=int(group)) logger.info(f"退出群聊 {group} 成功") await del_group.finish(f"退出群聊 {group} 成功", at_sender=True) except Exception as e: logger.info(f"退出群聊 {group} 失败 e:{e}") else: await del_group.finish(f"请输入正确的群号", at_sender=True) else: await del_group.finish(f"请输入群号", at_sender=True) @cls_group.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): gl = await bot.get_group_list(self_id=int(bot.self_id)) msg = ["{group_id} {group_name}".format_map(g) for g in gl] msg = "\n".join(msg) msg = f"bot:{bot.self_id}\n| 群号 | 群名 | 共{len(gl)}个群\n" + msg await bot.send_private_msg( self_id=int(bot.self_id), user_id=int(list(bot.config.superusers)[0]), message=msg, ) @cls_friend.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): gl = await bot.get_friend_list(self_id=int(bot.self_id)) msg = ["{user_id} {nickname}".format_map(g) for g in gl] msg = "\n".join(msg) msg = f"| QQ号 | 昵称 | 共{len(gl)}个好友\n" + msg await bot.send_private_msg( self_id=int(bot.self_id), user_id=int(list(bot.config.superusers)[0]), message=msg, ) @update_group_info.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): bot = get_bot() gl = await bot.get_group_list(self_id=bot.self_id) gl = [g["group_id"] for g in gl] num = 0 rst = "" for g in gl: group_info = await bot.get_group_info(group_id=g) if await GroupInfo.add_group_info( group_info["group_id"], group_info["group_name"], group_info["max_member_count"], group_info["member_count"], ): num += 1 logger.info(f"自动更新群组 {g} 信息成功") else: logger.info(f"自动更新群组 {g} 信息失败") rst += f"{g} 更新失败\n" await update_group_info.send(f"成功更新了 {num} 个群的信息\n{rst[:-1]}") @update_friend_info.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): num = 0 rst = "" fl = await get_bot().get_friend_list(self_id=bot.self_id) for f in fl: if await FriendUser.add_friend_info(f["user_id"], f["nickname"]): logger.info(f'自动更新好友 {f["user_id"]} 信息成功') num += 1 else: logger.warning(f'自动更新好友 {f["user_id"]} 信息失败') rst += f'{f["user_id"]} 更新失败\n' await update_friend_info.send(f"成功更新了 {num} 个好友的信息\n{rst[:-1]}") @clear_data.handle() async def _(bot: Bot, event: MessageEvent, state: T_State): await clear_data.send("开始清理临时数据....") size = await asyncio.get_event_loop().run_in_executor( None, _clear_data ) await clear_data.send("共清理了 {:.2f}MB 的数据...".format(size / 1024 / 1024)) def _clear_data() -> float: size = 0 for dir_name in ['temp', 'rar', 'r18_rar']: dir_name = f'{IMAGE_PATH}/{dir_name}' if os.path.exists(dir_name): for file in os.listdir(dir_name): try: file_size = os.path.getsize(os.path.join(dir_name, file)) os.remove(os.path.join(dir_name, file)) except Exception as e: logger.error(f"清理临时数据错误...e：{e}") file_size = 0 size += file_size return float(size) # 早上好 @scheduler.scheduled_job( "cron", hour=1, minute=1, ) async def _(): size = await asyncio.get_event_loop().run_in_executor( None, _clear_data ) logger.info('自动清理临时数据完成，' + "共清理了 {:.2f}MB 的数据...".format(size / 1024 / 1024))