mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
216 lines
6.8 KiB
Python
216 lines
6.8 KiB
Python
|
from asyncio.exceptions import TimeoutError
|
|||
|
|
|
|||
|
|
from httpx import NetworkError
|
|||
|
|
from nonebot.adapters import Bot
|
|||
|
|
from nonebot.plugin import PluginMetadata
|
|||
|
|
from nonebot.rule import to_me
|
|||
|
|
from nonebot_plugin_alconna import (
|
|||
|
|
Alconna,
|
|||
|
|
Args,
|
|||
|
|
Arparma,
|
|||
|
|
Match,
|
|||
|
|
Option,
|
|||
|
|
on_alconna,
|
|||
|
|
store_true,
|
|||
|
|
)
|
|||
|
|
from nonebot_plugin_saa import MessageFactory, Text
|
|||
|
|
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.utils import is_valid_date
|
|||
|
|
|
|||
|
|
from .data_source import download_pixiv_imgs, get_pixiv_urls, search_pixiv_urls
|
|||
|
|
|
|||
|
|
__plugin_meta__ = PluginMetadata(
|
|||
|
|
name="P站排行/搜图",
|
|||
|
|
description="P站排行榜直接冲，P站搜图跟着冲",
|
|||
|
|
usage="""
|
|||
|
|
P站排行：
|
|||
|
|
可选参数:
|
|||
|
|
类型：
|
|||
|
|
1. 日排行
|
|||
|
|
2. 周排行
|
|||
|
|
3. 月排行
|
|||
|
|
4. 原创排行
|
|||
|
|
5. 新人排行
|
|||
|
|
6. R18日排行
|
|||
|
|
7. R18周排行
|
|||
|
|
8. R18受男性欢迎排行
|
|||
|
|
9. R18重口排行【慎重！】
|
|||
|
|
【使用时选择参数序号即可，R18仅可私聊】
|
|||
|
|
p站排行 ?[参数] ?[数量] ?[日期]
|
|||
|
|
示例：
|
|||
|
|
p站排行 [无参数默认为日榜]
|
|||
|
|
p站排行 1
|
|||
|
|
p站排行 1 5
|
|||
|
|
p站排行 1 5 2018-4-25
|
|||
|
|
【注意空格！！】【在线搜索会较慢】
|
|||
|
|
---------------------------------
|
|||
|
|
P站搜图：
|
|||
|
|
搜图 [关键词] ?[数量] ?[页数=1] ?[r18](不屏蔽R-18)
|
|||
|
|
示例：
|
|||
|
|
搜图 樱岛麻衣
|
|||
|
|
搜图 樱岛麻衣 5
|
|||
|
|
搜图 樱岛麻衣 5 r18
|
|||
|
|
搜图 樱岛麻衣#1000users 5
|
|||
|
|
【多个关键词用#分割】
|
|||
|
|
【默认为 热度排序】
|
|||
|
|
【注意空格！！】【在线搜索会较慢】【数量可能不符？可能该页数量不够，也可能被R-18屏蔽】
|
|||
|
|
""".strip(),
|
|||
|
|
extra=PluginExtraData(
|
|||
|
|
author="HibiKier",
|
|||
|
|
version="0.1",
|
|||
|
|
aliases={"P站排行", "搜图"},
|
|||
|
|
menu_type="来点好康的",
|
|||
|
|
limits=[BaseBlock(result="P站排行榜或搜图正在搜索，请不要重复触发命令...")],
|
|||
|
|
configs=[
|
|||
|
|
RegisterConfig(
|
|||
|
|
key="TIMEOUT",
|
|||
|
|
value=10,
|
|||
|
|
help="图片下载超时限制",
|
|||
|
|
default_value=10,
|
|||
|
|
type=int,
|
|||
|
|
),
|
|||
|
|
RegisterConfig(
|
|||
|
|
key="MAX_PAGE_LIMIT",
|
|||
|
|
value=20,
|
|||
|
|
help="作品最大页数限制，超过的作品会被略过",
|
|||
|
|
default_value=20,
|
|||
|
|
type=int,
|
|||
|
|
),
|
|||
|
|
RegisterConfig(
|
|||
|
|
key="ALLOW_GROUP_R18",
|
|||
|
|
value=False,
|
|||
|
|
help="图允许群聊中使用 r18 参数",
|
|||
|
|
default_value=False,
|
|||
|
|
type=bool,
|
|||
|
|
),
|
|||
|
|
RegisterConfig(
|
|||
|
|
module="hibiapi",
|
|||
|
|
key="HIBIAPI",
|
|||
|
|
value="https://api.obfs.dev",
|
|||
|
|
help="如果没有自建或其他hibiapi请不要修改",
|
|||
|
|
default_value="https://api.obfs.dev",
|
|||
|
|
),
|
|||
|
|
RegisterConfig(
|
|||
|
|
module="pixiv",
|
|||
|
|
key="PIXIV_NGINX_URL",
|
|||
|
|
value="i.pixiv.re",
|
|||
|
|
help="Pixiv反向代理",
|
|||
|
|
),
|
|||
|
|
],
|
|||
|
|
).dict(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
rank_dict = {
|
|||
|
|
"1": "day",
|
|||
|
|
"2": "week",
|
|||
|
|
"3": "month",
|
|||
|
|
"4": "week_original",
|
|||
|
|
"5": "week_rookie",
|
|||
|
|
"6": "day_r18",
|
|||
|
|
"7": "week_r18",
|
|||
|
|
"8": "day_male_r18",
|
|||
|
|
"9": "week_r18g",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_rank_matcher = on_alconna(
|
|||
|
|
Alconna("p站排行", Args["rank_type", int, 1]["num", int, 10]["datetime?", str]),
|
|||
|
|
aliases={"p站排行榜"},
|
|||
|
|
priority=5,
|
|||
|
|
block=True,
|
|||
|
|
rule=to_me(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
_keyword_matcher = on_alconna(
|
|||
|
|
Alconna(
|
|||
|
|
"搜图",
|
|||
|
|
Args["keyword", str]["num", int, 10]["page", int, 1],
|
|||
|
|
Option("-r", action=store_true, help_text="是否屏蔽r18"),
|
|||
|
|
),
|
|||
|
|
priority=5,
|
|||
|
|
block=True,
|
|||
|
|
rule=to_me(),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@_rank_matcher.handle()
|
|||
|
|
async def _(
|
|||
|
|
bot: Bot,
|
|||
|
|
session: EventSession,
|
|||
|
|
arparma: Arparma,
|
|||
|
|
rank_type: int,
|
|||
|
|
num: int,
|
|||
|
|
datetime: Match[str],
|
|||
|
|
):
|
|||
|
|
gid = session.id3 or session.id2
|
|||
|
|
if not session.id1:
|
|||
|
|
await Text("用户id为空...").finish()
|
|||
|
|
code = 0
|
|||
|
|
info_list = []
|
|||
|
|
_datetime = None
|
|||
|
|
if datetime.available:
|
|||
|
|
_datetime = datetime.result
|
|||
|
|
if not is_valid_date(_datetime):
|
|||
|
|
await Text("日期不合法，示例: 2018-4-25").finish(reply=True)
|
|||
|
|
if rank_type in [6, 7, 8, 9]:
|
|||
|
|
if gid:
|
|||
|
|
await Text("羞羞脸！私聊里自己看！").finish(at_sender=True)
|
|||
|
|
info_list, code = await get_pixiv_urls(
|
|||
|
|
rank_dict[str(rank_type)], num, date=_datetime
|
|||
|
|
)
|
|||
|
|
if code != 200 and info_list:
|
|||
|
|
if isinstance(info_list[0], str):
|
|||
|
|
await Text(info_list[0]).finish()
|
|||
|
|
if not info_list:
|
|||
|
|
await Text("没有找到啊，等等再试试吧~V").send(at_sender=True)
|
|||
|
|
for title, author, urls in info_list:
|
|||
|
|
try:
|
|||
|
|
images = await download_pixiv_imgs(urls, session.id1) # type: ignore
|
|||
|
|
await MessageFactory(
|
|||
|
|
[Text(f"title: {title}\n"), Text(f"author: {author}\n")] + images
|
|||
|
|
).send()
|
|||
|
|
|
|||
|
|
except (NetworkError, TimeoutError):
|
|||
|
|
await Text("这张图网络直接炸掉了！").send()
|
|||
|
|
logger.info(
|
|||
|
|
f" 查看了P站排行榜 rank_type{rank_type}", arparma.header_result, session=session
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@_keyword_matcher.handle()
|
|||
|
|
async def _(
|
|||
|
|
bot: Bot, session: EventSession, arparma: Arparma, keyword: str, num: int, page: int
|
|||
|
|
):
|
|||
|
|
gid = session.id3 or session.id2
|
|||
|
|
if not session.id1:
|
|||
|
|
await Text("用户id为空...").finish()
|
|||
|
|
if gid:
|
|||
|
|
if arparma.find("r") and not Config.get_config(
|
|||
|
|
"pixiv_rank_search", "ALLOW_GROUP_R18"
|
|||
|
|
):
|
|||
|
|
await Text("(脸红#) 你不会害羞的 八嘎！").finish(at_sender=True)
|
|||
|
|
r18 = 0 if arparma.find("r") else 1
|
|||
|
|
info_list = None
|
|||
|
|
keyword = keyword.replace("#", " ")
|
|||
|
|
info_list, code = await search_pixiv_urls(keyword, num, page, r18)
|
|||
|
|
if code != 200 and isinstance(info_list[0], str):
|
|||
|
|
await Text(info_list[0]).finish()
|
|||
|
|
if not info_list:
|
|||
|
|
await Text("没有找到啊，等等再试试吧~V").finish(at_sender=True)
|
|||
|
|
for title, author, urls in info_list:
|
|||
|
|
try:
|
|||
|
|
images = await download_pixiv_imgs(urls, session.id1) # type: ignore
|
|||
|
|
await MessageFactory(
|
|||
|
|
[Text(f"title: {title}\n"), Text(f"author: {author}\n")] + images
|
|||
|
|
).send()
|
|||
|
|
|
|||
|
|
except (NetworkError, TimeoutError):
|
|||
|
|
await Text("这张图网络直接炸掉了！").send()
|
|||
|
|
logger.info(
|
|||
|
|
f" 查看了搜索 {keyword} R18：{r18}", arparma.header_result, session=session
|
|||
|
|
)
|