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}"
|
|||
|
|
)
|