diff --git a/plugins/pix_gallery/__init__.py b/plugins/pix_gallery/__init__.py index 54bd7414..8e71f3b2 100755 --- a/plugins/pix_gallery/__init__.py +++ b/plugins/pix_gallery/__init__.py @@ -1,13 +1,5 @@ from configs.config import Config -from services.log import logger -from .model.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 __zx_plugin_name__ = "更新扩展图库Omega [Hidden]" @@ -67,69 +59,4 @@ Config.add_plugin_config( nonebot.load_plugins("plugins/pix_gallery") -driver: Driver = nonebot.get_driver() - -@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}")