mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
|
from nonebot.adapters import Bot
|
|||
|
|
from nonebot.plugin import PluginMetadata
|
|||
|
|
from nonebot_plugin_alconna import Alconna, Args, Arparma, Match, on_alconna
|
|||
|
|
from nonebot_plugin_saa import Image, Text
|
|||
|
|
from nonebot_plugin_session import EventSession
|
|||
|
|
|
|||
|
|
from zhenxun.configs.utils import PluginExtraData
|
|||
|
|
from zhenxun.services.log import logger
|
|||
|
|
|
|||
|
|
from ._data_source import gen_keyword_pic, get_keyword_num
|
|||
|
|
from ._model.pixiv_keyword_user import PixivKeywordUser
|
|||
|
|
|
|||
|
|
__plugin_meta__ = PluginMetadata(
|
|||
|
|
name="查看pix图库",
|
|||
|
|
description="让我看看管理员私藏了多少货",
|
|||
|
|
usage="""
|
|||
|
|
指令：
|
|||
|
|
我的pix关键词
|
|||
|
|
显示pix关键词
|
|||
|
|
查看pix图库 ?[tag]: 查看指定tag图片数量，为空时查看整个图库
|
|||
|
|
""".strip(),
|
|||
|
|
extra=PluginExtraData(
|
|||
|
|
author="HibiKier",
|
|||
|
|
version="0.1",
|
|||
|
|
).dict(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
_my_matcher = on_alconna(Alconna("我的pix关键词"), priority=5, block=True)
|
|||
|
|
|
|||
|
|
_show_matcher = on_alconna(Alconna("显示pix关键词"), priority=5, block=True)
|
|||
|
|
|
|||
|
|
_pix_matcher = on_alconna(
|
|||
|
|
Alconna("查看pix图库", Args["keyword?", str]), priority=5, block=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@_my_matcher.handle()
|
|||
|
|
async def _(arparma: Arparma, session: EventSession):
|
|||
|
|
data = await PixivKeywordUser.filter(user_id=session.id1).values_list(
|
|||
|
|
"keyword", flat=True
|
|||
|
|
)
|
|||
|
|
if not data:
|
|||
|
|
await Text("您目前没有提供任何Pixiv搜图关键字...").finish(reply=True)
|
|||
|
|
await Text(f"您目前提供的如下关键字：\n\t" + "，".join(data)).send() # type: ignore
|
|||
|
|
logger.info("查看我的pix关键词", arparma.header_result, session=session)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@_show_matcher.handle()
|
|||
|
|
async def _(bot: Bot, arparma: Arparma, session: EventSession):
|
|||
|
|
_pass_keyword, not_pass_keyword = await PixivKeywordUser.get_current_keyword()
|
|||
|
|
if _pass_keyword or not_pass_keyword:
|
|||
|
|
image = await gen_keyword_pic(
|
|||
|
|
_pass_keyword, not_pass_keyword, session.id1 in bot.config.superusers
|
|||
|
|
)
|
|||
|
|
await Image(image.pic2bytes()).send() # type: ignore
|
|||
|
|
else:
|
|||
|
|
if session.id1 in bot.config.superusers:
|
|||
|
|
await Text(f"目前没有已收录或待收录的搜索关键词...").send()
|
|||
|
|
else:
|
|||
|
|
await Text(f"目前没有已收录的搜索关键词...").send()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@_pix_matcher.handle()
|
|||
|
|
async def _(bot: Bot, arparma: Arparma, session: EventSession, keyword: Match[str]):
|
|||
|
|
_keyword = ""
|
|||
|
|
if keyword.available:
|
|||
|
|
_keyword = keyword.result
|
|||
|
|
count, r18_count, count_, setu_count, r18_count_ = await get_keyword_num(_keyword)
|
|||
|
|
await Text(
|
|||
|
|
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_}"
|
|||
|
|
).send()
|
|||
|
|
logger.info("查看pix图库", arparma.header_result, session=session)
|