update code

plugins/pix_gallery/__init__.py

@@ -1,14 +1,76 @@
+from configs.config import HIBIAPI
+from services.log import logger
+from models.omega_pixiv_illusts import OmegaPixivIllusts
+from pathlib import Path
+from nonebot import Driver
+from typing import List
+from datetime import datetime
+import nonebot
+import asyncio
+import os
+
+driver: Driver = nonebot.get_driver()
+
+illust_url = f"{HIBIAPI}/api/pixiv/illust"
 
 
+@driver.on_startup
+async def _init_omega_pixiv_illusts():
+    omega_pixiv_illusts = None
+    for file in os.listdir("."):
+        if "omega_pixiv_illusts" in file and ".sql" in file:
+            omega_pixiv_illusts = Path() / file
+    if omega_pixiv_illusts:
+        with open(omega_pixiv_illusts, "r", encoding="utf8") as f:
+            lines = f.readlines()
+        tasks = []
+        length = len([x for x in lines if "INSERT INTO" in x.upper()])
+        all_pid = await OmegaPixivIllusts.get_all_pid()
+        index = 0
+        logger.info('检测到OmegaPixivIllusts数据库，准备开始更新....')
+        for line in lines:
+            if "INSERT INTO" in line.upper():
+                index += 1
+                tasks.append(asyncio.ensure_future(_tasks(line, all_pid, length, index)))
+        await asyncio.gather(*tasks)
+        omega_pixiv_illusts.unlink()
 
 
-
-
-
-
-
-
-
-
-
+async def _tasks(line: str, all_pid: List[int], length: int, index: int):
+    data = line.split("VALUES", maxsplit=1)[-1].strip()
+    if data.startswith("("):
+        data = data[1:]
+    if data.endswith(");"):
+        data = data[:-2]
+    x = data.split(maxsplit=3)
+    pid = int(x[1][:-1].strip())
+    if pid in all_pid:
+        logger.info(f'添加OmegaPixivIllusts图库数据已存在 ---> pid：{pid}')
+        return
+    uid = int(x[2][:-1].strip())
+    x = x[3].split(", '")
+    title = x[0].strip()[1:-1]
+    tmp = x[1].split(', ')
+    author = tmp[0].strip()[:-1]
+    nsfw_tag = int(tmp[1])
+    width = int(tmp[2])
+    height = int(tmp[3])
+    tags = x[2][:-1]
+    url = x[3][:-1]
+    if await OmegaPixivIllusts.add_image_data(
+        pid,
+        title,
+        width,
+        height,
+        url,
+        uid,
+        author,
+        nsfw_tag,
+        tags,
+        datetime.min,
+        datetime.min
+    ):
+        logger.info(f"成功添加OmegaPixivIllusts图库数据 pid：{pid} 本次预计存储 {length} 张，已更新第 {index} 张")
+    else:
+        logger.info(f"添加OmegaPixivIllusts图库数据已存在 ---> pid：{pid}")