mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from nonebot import on_command
|
||
from nonebot.permission import SUPERUSER
|
||
from nonebot.typing import T_State
|
||
from nonebot.adapters.cqhttp import Bot, MessageEvent
|
||
from nonebot.rule import to_me
|
||
from utils.utils import get_bot
|
||
from services.log import logger
|
||
from models.group_info import GroupInfo
|
||
from models.friend_user import FriendUser
|
||
|
||
|
||
__zx_plugin_name__ = "更新群/好友信息 [Superuser]"
|
||
__plugin_usage__ = """
|
||
usage：
|
||
更新群/好友信息
|
||
指令：
|
||
更新群信息
|
||
更新好友信息
|
||
""".strip()
|
||
__plugin_des__ = "更新群/好友信息"
|
||
__plugin_cmd__ = [
|
||
"更新群信息",
|
||
"更新好友信息",
|
||
]
|
||
__plugin_version__ = 0.1
|
||
__plugin_author__ = "HibiKier"
|
||
|
||
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
|
||
)
|
||
|
||
|
||
@update_group_info.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
bot = get_bot()
|
||
gl = await bot.get_group_list()
|
||
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()
|
||
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]}")
|