zhenxun_bot/plugins/pix_gallery/pix_show_info.py

from nonebot import on_command
from utils.utils import get_message_text
from utils.message_builder import image
from nonebot.adapters.cqhttp import Bot, MessageEvent
from nonebot.typing import T_State
from .data_source import gen_keyword_pic, get_keyword_num
from models.pixiv_keyword_user import PixivKeywordUser
import asyncio

__plugin_name__ = "查看pix图库"

__plugin_usage__ = """
    查看pix图库 [keyword]：当keyword为空时检查所有图片
"""


my_keyword = on_command("我的pix关键词", aliases={"我的pix关键字"}, priority=1, block=True)

show_keyword = on_command("显示pix关键词", aliases={"显示pix关键字"}, priority=1, block=True)

show_pix = on_command("查看pix图库", priority=1, block=True)


@my_keyword.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State):
    data = await PixivKeywordUser.get_all_user_dict()
    if data.get(event.user_id) is None or not data[event.user_id]["keyword"]:
        await my_keyword.finish("您目前没有提供任何Pixiv搜图关键字...", at_sender=True)
    await my_keyword.send(
        f"您目前提供的如下关键字：\n\t" + "，".join(data[event.user_id]["keyword"])
    )


@show_keyword.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State):
    _pass_keyword, not_pass_keyword = await PixivKeywordUser.get_current_keyword()
    if _pass_keyword or not_pass_keyword:
        await show_keyword.send(
            image(
                b64=await asyncio.get_event_loop().run_in_executor(
                    None,
                    gen_keyword_pic,
                    _pass_keyword,
                    not_pass_keyword,
                    str(event.user_id) in bot.config.superusers,
                )
            )
        )
    else:
        if str(event.user_id) in bot.config.superusers:
            await show_keyword.finish(f"目前没有已收录或待收录的搜索关键词...")
        else:
            await show_keyword.finish(f"目前没有已收录的搜索关键词...")


@show_pix.handle()
async def _(bot: Bot, event: MessageEvent, state: T_State):
    keyword = get_message_text(event.json())
    count, r18_count, count_, setu_count, r18_count_ = await get_keyword_num(keyword)
    await show_pix.send(
        f"PIX图库：{keyword}\n"
        f"总数：{count + r18_count}\n"
        f"美图：{count}\n"
        f"R18：{r18_count}\n"
        f"---------------\n"
        f"Omega图库：{keyword}\n"
        f"总数：{count_ + setu_count + r18_count_}\n"
        f"美图：{count_}\n"
        f"色图：{setu_count}\n"
        f"R18：{r18_count_}"
    )