update 0.0.6.0

2026-09-29 08:39:59 +08:00 · 2021-11-04 16:11:50 +08:00
parent 21c1950898
commit 19d650c35d
217 changed files with 8213 additions and 5697 deletions
@@ -0,0 +1,111 @@
+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}")
+
+
+
+
+