mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
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 = 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}"
|
||
)
|