mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
248 lines
8.7 KiB
Python
248 lines
8.7 KiB
Python
import random
|
||
|
||
from nonebot.adapters import Bot
|
||
from nonebot.plugin import PluginMetadata
|
||
from nonebot_plugin_alconna import (
|
||
Alconna,
|
||
Args,
|
||
Arparma,
|
||
Match,
|
||
Option,
|
||
on_alconna,
|
||
store_true,
|
||
)
|
||
from nonebot_plugin_session import EventSession
|
||
|
||
from zhenxun.configs.config import Config
|
||
from zhenxun.configs.utils import BaseBlock, PluginExtraData, RegisterConfig
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.message import MessageUtils
|
||
from zhenxun.utils.platform import PlatformUtils
|
||
from zhenxun.utils.withdraw_manage import WithdrawManager
|
||
|
||
from ._data_source import get_image
|
||
from ._model.omega_pixiv_illusts import OmegaPixivIllusts
|
||
from ._model.pixiv import Pixiv
|
||
|
||
__plugin_meta__ = PluginMetadata(
|
||
name="PIX",
|
||
description="这里是PIX图库！",
|
||
usage="""
|
||
指令：
|
||
pix ?*[tags]: 通过 tag 获取相似图片，不含tag时随机抽取
|
||
pid [uid]: 通过uid获取图片
|
||
pix pid[pid]: 查看图库中指定pid图片
|
||
示例：pix 萝莉 白丝
|
||
示例：pix 萝莉 白丝 10 （10为数量）
|
||
示例：pix #02 （当tag只有1个tag且为数字时，使用#标记，否则将被判定为数量）
|
||
示例：pix 34582394 （查询指定uid图片）
|
||
示例：pix pid:12323423 （查询指定pid图片）
|
||
""".strip(),
|
||
extra=PluginExtraData(
|
||
author="HibiKier",
|
||
version="0.1",
|
||
superuser_help="""
|
||
指令：
|
||
pix -s ?*[tags]: 通过tag获取色图，不含tag时随机
|
||
pix -r ?*[tags]: 通过tag获取r18图，不含tag时随机
|
||
""",
|
||
menu_type="来点好康的",
|
||
limits=[BaseBlock(result="您有PIX图片正在处理，请稍等...")],
|
||
configs=[
|
||
RegisterConfig(
|
||
key="MAX_ONCE_NUM2FORWARD",
|
||
value=None,
|
||
help="单次发送的图片数量达到指定值时转发为合并消息",
|
||
default_value=None,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="ALLOW_GROUP_SETU",
|
||
value=False,
|
||
help="允许非超级用户使用-s参数",
|
||
default_value=False,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="ALLOW_GROUP_R18",
|
||
value=False,
|
||
help="允许非超级用户使用-r参数",
|
||
default_value=False,
|
||
type=bool,
|
||
),
|
||
],
|
||
).dict(),
|
||
)
|
||
|
||
# pix = on_command("pix", aliases={"PIX", "Pix"}, priority=5, block=True)
|
||
|
||
_matcher = on_alconna(
|
||
Alconna(
|
||
"pix",
|
||
Args["tags?", str] / "\n",
|
||
Option("-s", action=store_true, help_text="色图"),
|
||
Option("-r", action=store_true, help_text="r18"),
|
||
),
|
||
priority=5,
|
||
block=True,
|
||
)
|
||
|
||
PIX_RATIO = None
|
||
OMEGA_RATIO = None
|
||
|
||
|
||
@_matcher.handle()
|
||
async def _(bot: Bot, session: EventSession, arparma: Arparma, tags: Match[str]):
|
||
global PIX_RATIO, OMEGA_RATIO
|
||
gid = session.id3 or session.id2
|
||
if not session.id1:
|
||
await MessageUtils.build_message("用户id为空...").finish()
|
||
if PIX_RATIO is None:
|
||
pix_omega_pixiv_ratio = Config.get_config("pix", "PIX_OMEGA_PIXIV_RATIO")
|
||
PIX_RATIO = pix_omega_pixiv_ratio[0] / (
|
||
pix_omega_pixiv_ratio[0] + pix_omega_pixiv_ratio[1]
|
||
)
|
||
OMEGA_RATIO = 1 - PIX_RATIO
|
||
num = 1
|
||
# keyword = arg.extract_plain_text().strip()
|
||
keyword = ""
|
||
spt = tags.result.split() if tags.available else []
|
||
if arparma.find("s"):
|
||
nsfw_tag = 1
|
||
elif arparma.find("r"):
|
||
nsfw_tag = 2
|
||
else:
|
||
nsfw_tag = 0
|
||
if session.id1 not in bot.config.superusers:
|
||
if (nsfw_tag == 1 and not Config.get_config("pix", "ALLOW_GROUP_SETU")) or (
|
||
nsfw_tag == 2 and not Config.get_config("pix", "ALLOW_GROUP_R18")
|
||
):
|
||
await MessageUtils.build_message(
|
||
"你不能看这些噢，这些都是是留给管理员看的..."
|
||
).finish()
|
||
if (n := len(spt)) == 1:
|
||
if str(spt[0]).isdigit() and int(spt[0]) < 100:
|
||
num = int(spt[0])
|
||
keyword = ""
|
||
elif spt[0].startswith("#"):
|
||
keyword = spt[0][1:]
|
||
elif n > 1:
|
||
if str(spt[-1]).isdigit():
|
||
num = int(spt[-1])
|
||
if num > 10:
|
||
if session.id1 not in bot.config.superusers or (
|
||
session.id1 in bot.config.superusers and num > 30
|
||
):
|
||
num = random.randint(1, 10)
|
||
await MessageUtils.build_message(
|
||
f"太贪心了，就给你发 {num}张 好了"
|
||
).send()
|
||
spt = spt[:-1]
|
||
keyword = " ".join(spt)
|
||
pix_num = int(num * PIX_RATIO) + 15 if PIX_RATIO != 0 else 0
|
||
omega_num = num - pix_num + 15
|
||
if str(keyword).isdigit():
|
||
if num == 1:
|
||
pix_num = 15
|
||
omega_num = 15
|
||
all_image = await Pixiv.query_images(
|
||
uid=int(keyword), num=pix_num, r18=1 if nsfw_tag == 2 else 0
|
||
) + await OmegaPixivIllusts.query_images(
|
||
uid=int(keyword), num=omega_num, nsfw_tag=nsfw_tag
|
||
)
|
||
elif keyword.lower().startswith("pid"):
|
||
pid = keyword.replace("pid", "").replace(":", "").replace("：", "")
|
||
if not str(pid).isdigit():
|
||
await MessageUtils.build_message("PID必须是数字...").finish(reply_to=True)
|
||
all_image = await Pixiv.query_images(
|
||
pid=int(pid), r18=1 if nsfw_tag == 2 else 0
|
||
)
|
||
if not all_image:
|
||
all_image = await OmegaPixivIllusts.query_images(
|
||
pid=int(pid), nsfw_tag=nsfw_tag
|
||
)
|
||
num = len(all_image)
|
||
else:
|
||
tmp = await Pixiv.query_images(
|
||
spt, r18=1 if nsfw_tag == 2 else 0, num=pix_num
|
||
) + await OmegaPixivIllusts.query_images(spt, nsfw_tag=nsfw_tag, num=omega_num)
|
||
tmp_ = []
|
||
all_image = []
|
||
for x in tmp:
|
||
if x.pid not in tmp_:
|
||
all_image.append(x)
|
||
tmp_.append(x.pid)
|
||
if not all_image:
|
||
await MessageUtils.build_message(
|
||
f"未在图库中找到与 {keyword} 相关Tag/UID/PID的图片..."
|
||
).finish(reply_to=True)
|
||
msg_list = []
|
||
for _ in range(num):
|
||
img_url = None
|
||
author = None
|
||
if not all_image:
|
||
await MessageUtils.build_message("坏了...发完了，没图了...").finish()
|
||
img = random.choice(all_image)
|
||
all_image.remove(img) # type: ignore
|
||
if isinstance(img, OmegaPixivIllusts):
|
||
img_url = img.url
|
||
author = img.uname
|
||
elif isinstance(img, Pixiv):
|
||
img_url = img.img_url
|
||
author = img.author
|
||
pid = img.pid
|
||
title = img.title
|
||
uid = img.uid
|
||
if img_url:
|
||
_img = await get_image(img_url, session.id1)
|
||
if _img:
|
||
if Config.get_config("pix", "SHOW_INFO"):
|
||
msg_list.append(
|
||
MessageUtils.build_message(
|
||
[
|
||
f"title：{title}\n"
|
||
f"author：{author}\n"
|
||
f"PID：{pid}\nUID：{uid}\n",
|
||
_img,
|
||
]
|
||
)
|
||
)
|
||
else:
|
||
msg_list.append(_img)
|
||
logger.info(
|
||
f" 查看PIX图库PID: {pid}", arparma.header_result, session=session
|
||
)
|
||
else:
|
||
msg_list.append(MessageUtils.build_message("这张图似乎下载失败了"))
|
||
logger.info(
|
||
f" 查看PIX图库PID: {pid}，下载图片出错",
|
||
arparma.header_result,
|
||
session=session,
|
||
)
|
||
if (
|
||
Config.get_config("pix", "MAX_ONCE_NUM2FORWARD")
|
||
and num >= Config.get_config("pix", "MAX_ONCE_NUM2FORWARD")
|
||
and gid
|
||
):
|
||
for msg in msg_list:
|
||
receipt = await msg.send()
|
||
if receipt:
|
||
message_id = receipt.msg_ids[0]["message_id"]
|
||
await WithdrawManager.withdraw_message(
|
||
bot,
|
||
str(message_id),
|
||
Config.get_config("pix", "WITHDRAW_PIX_MESSAGE"),
|
||
session,
|
||
)
|
||
else:
|
||
for msg in msg_list:
|
||
receipt = await msg.send()
|
||
if receipt:
|
||
message_id = receipt.msg_ids[0]["message_id"]
|
||
await WithdrawManager.withdraw_message(
|
||
bot,
|
||
message_id,
|
||
Config.get_config("pix", "WITHDRAW_PIX_MESSAGE"),
|
||
session,
|
||
)
|