mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from pathlib import Path
|
||
|
||
import nonebot
|
||
from nonebot.adapters import Bot
|
||
|
||
from zhenxun.models.group_console import GroupConsole
|
||
from zhenxun.services.cache import DbCacheException
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.platform import PlatformUtils
|
||
|
||
nonebot.load_plugins(str(Path(__file__).parent.resolve()))
|
||
|
||
try:
|
||
from .__init_cache import CacheRoot
|
||
except DbCacheException as e:
|
||
raise SystemError(f"ERROR：{e}")
|
||
|
||
driver = nonebot.get_driver()
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
await CacheRoot.init_non_lazy_caches()
|
||
|
||
|
||
@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)} 条数据..."
|
||
)
|