mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
|
from nonebot import Driver
|
|||
|
|
from services.db_context import db
|
|||
|
|
from asyncpg.exceptions import DuplicateColumnError
|
|||
|
|
from models.group_info import GroupInfo
|
|||
|
|
from nonebot.adapters.cqhttp import Bot
|
|||
|
|
from utils.user_agent import get_user_agent
|
|||
|
|
from services.log import logger
|
|||
|
|
from configs.path_config import TEXT_PATH
|
|||
|
|
from asyncio.exceptions import TimeoutError
|
|||
|
|
from pathlib import Path
|
|||
|
|
import aiohttp
|
|||
|
|
import nonebot
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import ujson as json
|
|||
|
|
except ModuleNotFoundError:
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
|
|||
|
|
driver: Driver = nonebot.get_driver()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@driver.on_startup
|
|||
|
|
async def update_city():
|
|||
|
|
"""
|
|||
|
|
部分插件需要中国省份城市
|
|||
|
|
这里直接更新，避免插件内代码重复
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
china_city = Path(TEXT_PATH) / "china_city.json"
|
|||
|
|
data = {}
|
|||
|
|
try:
|
|||
|
|
async with aiohttp.ClientSession(headers=get_user_agent()) as session:
|
|||
|
|
async with session.get(
|
|||
|
|
"http://www.weather.com.cn/data/city3jdata/china.html", timeout=5
|
|||
|
|
) as res:
|
|||
|
|
provinces_data = json.loads(await res.text(encoding="utf8"))
|
|||
|
|
for province in provinces_data.keys():
|
|||
|
|
data[provinces_data[province]] = []
|
|||
|
|
async with session.get(
|
|||
|
|
f"http://www.weather.com.cn/data/city3jdata/provshi/{province}.html",
|
|||
|
|
timeout=5,
|
|||
|
|
) as res:
|
|||
|
|
city_data = json.loads(await res.text(encoding="utf8"))
|
|||
|
|
for city in city_data.keys():
|
|||
|
|
data[provinces_data[province]].append(city_data[city])
|
|||
|
|
with open(china_city, "w", encoding="utf8") as f:
|
|||
|
|
json.dump(data, f, indent=4, ensure_ascii=False)
|
|||
|
|
logger.info("自动更新城市列表完成.....")
|
|||
|
|
except TimeoutError:
|
|||
|
|
logger.info("自动更新城市列表超时.....")
|
|||
|
|
|
|||
|
|
|
|||
|
|
@driver.on_startup
|
|||
|
|
async def _():
|
|||
|
|
"""
|
|||
|
|
数据库表结构变换
|
|||
|
|
"""
|
|||
|
|
sql_str = [
|
|||
|
|
"ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;" # group_info表添加一个group_flag
|
|||
|
|
]
|
|||
|
|
for sql in sql_str:
|
|||
|
|
try:
|
|||
|
|
query = db.text(sql)
|
|||
|
|
await db.first(query)
|
|||
|
|
except DuplicateColumnError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
@driver.on_bot_connect
|
|||
|
|
async def _(bot: Bot):
|
|||
|
|
"""
|
|||
|
|
版本某些需要的变换
|
|||
|
|
"""
|
|||
|
|
# 清空不存在的群聊信息，并将已所有已存在的群聊group_flag设置为1（认证所有已存在的群）
|
|||
|
|
if not await GroupInfo.get_group_info(114514):
|
|||
|
|
# 标识符，该功能只需执行一次
|
|||
|
|
await GroupInfo.add_group_info(
|
|||
|
|
114514,
|
|||
|
|
"114514",
|
|||
|
|
114514,
|
|||
|
|
114514,
|
|||
|
|
1
|
|||
|
|
)
|
|||
|
|
group_list = await bot.get_group_list()
|
|||
|
|
group_list = [g["group_id"] for g in group_list]
|
|||
|
|
_gl = [x.group_id for x in await GroupInfo.get_all_group()]
|
|||
|
|
if 114514 in _gl:
|
|||
|
|
_gl.remove(114514)
|
|||
|
|
for group_id in _gl:
|
|||
|
|
if group_id in group_list:
|
|||
|
|
if await GroupInfo.get_group_info(group_id):
|
|||
|
|
await GroupInfo.set_group_flag(group_id, 1)
|
|||
|
|
else:
|
|||
|
|
group_info = await bot.get_group_info(group_id=group_id)
|
|||
|
|
await GroupInfo.add_group_info(
|
|||
|
|
group_info["group_id"],
|
|||
|
|
group_info["group_name"],
|
|||
|
|
group_info["max_member_count"],
|
|||
|
|
group_info["member_count"],
|
|||
|
|
1
|
|||
|
|
)
|
|||
|
|
logger.info(f"已将群聊 {group_id} 添加认证...")
|
|||
|
|
else:
|
|||
|
|
await GroupInfo.delete_group_info(group_id)
|
|||
|
|
logger.info(f"移除不存在的群聊信息：{group_id}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|