mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from pathlib import Path
|
|
|
|
import nonebot
|
|
from nonebot.adapters import Bot
|
|
|
|
from zhenxun.models.group_console import GroupConsole
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.platform import PlatformUtils
|
|
|
|
nonebot.load_plugins(str(Path(__file__).parent.resolve()))
|
|
|
|
|
|
driver = nonebot.get_driver()
|
|
|
|
|
|
@driver.on_bot_connect
|
|
async def _(bot: Bot):
|
|
"""将bot已存在的群组添加群认证
|
|
|
|
参数:
|
|
bot: Bot
|
|
"""
|
|
if PlatformUtils.get_platform(bot) != "qq":
|
|
return
|
|
logger.debug(f"更新Bot: {bot.self_id} 的群认证...")
|
|
group_list, _ = await PlatformUtils.get_group_list(bot)
|
|
db_group_list = await GroupConsole.all().values_list("group_id", flat=True)
|
|
create_list = []
|
|
update_id = []
|
|
for group in group_list:
|
|
if group.group_id not in db_group_list:
|
|
group.group_flag = 1
|
|
create_list.append(group)
|
|
else:
|
|
update_id.append(group.group_id)
|
|
if create_list:
|
|
await GroupConsole.bulk_create(create_list, 10)
|
|
else:
|
|
await GroupConsole.filter(group_id__in=update_id).update(group_flag=1)
|
|
logger.debug(
|
|
f"更新Bot: {bot.self_id} 的群认证完成，共创建 {len(create_list)} 条数据，"
|
|
f"共修改 {len(update_id)} 条数据..."
|
|
)
|