mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-10 22:30:02 +08:00
update 0.0.6.2
This commit is contained in:
HibiKi
Regular → Executable
+212
-213
@@ -1,213 +1,212 @@
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
|
||||
from nonebot.matcher import Matcher
|
||||
from nonebot import on_command
|
||||
from utils.utils import get_message_text, is_number
|
||||
from .data_source import get_pixiv_urls, download_pixiv_imgs, search_pixiv_urls
|
||||
from services.log import logger
|
||||
from nonebot.adapters.cqhttp.exception import NetworkError
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from aiohttp.client_exceptions import ClientConnectorError
|
||||
from utils.message_builder import custom_forward_msg
|
||||
from configs.config import Config
|
||||
from typing import Type
|
||||
from nonebot.rule import to_me
|
||||
import time
|
||||
|
||||
__zx_plugin_name__ = "P站排行/搜图"
|
||||
|
||||
__plugin_usage__ = """
|
||||
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
|
||||
【默认为 热度排序】
|
||||
【注意空格！！】【在线搜索会较慢】【数量可能不符？可能该页数量不够，也可能被R-18屏蔽】
|
||||
""".strip()
|
||||
__plugin_des__ = "P站排行榜直接冲，P站搜图跟着冲"
|
||||
__plugin_cmd__ = ["p站排行 ?[参数] ?[数量] ?[日期]", "搜图 [关键词] ?[数量] ?[页数=1] ?[r18](不屏蔽R-18)"]
|
||||
__plugin_type__ = ("来点好康的",)
|
||||
__plugin_version__ = 0.1
|
||||
__plugin_author__ = "HibiKier"
|
||||
__plugin_settings__ = {
|
||||
"level": 9,
|
||||
"default_status": True,
|
||||
"limit_superuser": False,
|
||||
"cmd": ["p站排行", "搜图", "p站搜图", "P站搜图"],
|
||||
}
|
||||
__plugin_block_limit__ = {"rst": "P站排行榜或搜图正在搜索，请不要重复触发命令..."}
|
||||
__plugin_configs__ = {
|
||||
"TIMEOUT": {
|
||||
"value": 10,
|
||||
"help": "图片下载超时限制",
|
||||
"default_value": 10
|
||||
}
|
||||
}
|
||||
Config.add_plugin_config(
|
||||
"hibiapi",
|
||||
"HIBIAPI",
|
||||
"https://api.obfs.dev",
|
||||
help_="如果没有自建或其他hibiapi请不要修改",
|
||||
default_value="https://api.obfs.dev",
|
||||
)
|
||||
Config.add_plugin_config(
|
||||
"pixiv",
|
||||
"PIXIV_NGINX_URL",
|
||||
"i.pixiv.re",
|
||||
help_="Pixiv反向代理"
|
||||
)
|
||||
|
||||
|
||||
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",
|
||||
}
|
||||
|
||||
|
||||
pixiv_rank = on_command(
|
||||
"p站排行",
|
||||
aliases={"P站排行榜", "p站排行榜", "P站排行榜", "P站排行"},
|
||||
priority=5,
|
||||
block=True,
|
||||
rule=to_me(),
|
||||
)
|
||||
pixiv_keyword = on_command("搜图", priority=5, block=True, rule=to_me())
|
||||
|
||||
|
||||
@pixiv_rank.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
msg = get_message_text(event.json()).strip()
|
||||
msg = msg.split(" ")
|
||||
msg = [m for m in msg if m]
|
||||
code = 0
|
||||
info_list = []
|
||||
if not msg:
|
||||
msg = ["1"]
|
||||
if msg[0] in ["6", "7", "8", "9"]:
|
||||
if event.message_type == "group":
|
||||
await pixiv_rank.finish("羞羞脸！私聊里自己看！", at_sender=True)
|
||||
if (n := len(msg)) == 0 or msg[0] == "":
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get("1"))
|
||||
elif n == 1:
|
||||
if msg[0] not in ["1", "2", "3", "4", "5", "6", "7", "8", "9"]:
|
||||
await pixiv_rank.finish("要好好输入要看什么类型的排行榜呀！", at_sender=True)
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get(msg[0]))
|
||||
elif n == 2:
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get(msg[0]), int(msg[1]))
|
||||
elif n == 3:
|
||||
if not check_date(msg[2]):
|
||||
await pixiv_rank.finish("日期格式错误了", at_sender=True)
|
||||
info_list, code = await get_pixiv_urls(
|
||||
rank_dict.get(msg[0]), int(msg[1]), date=msg[2]
|
||||
)
|
||||
else:
|
||||
await pixiv_rank.finish("格式错了噢，参数不够？看看帮助？", at_sender=True)
|
||||
if code != 200 and info_list:
|
||||
await pixiv_rank.finish(info_list[0])
|
||||
if not info_list:
|
||||
await pixiv_rank.finish("没有找到啊，等等再试试吧~V", at_sender=True)
|
||||
await send_image(info_list, pixiv_rank, bot, event)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||||
f" 查看了P站排行榜 code：{msg[0]}"
|
||||
)
|
||||
|
||||
|
||||
@pixiv_keyword.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
msg = get_message_text(event.json())
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
if "r18" in msg.lower():
|
||||
await pixiv_keyword.finish("(脸红#) 你不会害羞的 八嘎！", at_sender=True)
|
||||
r18 = 0 if "r18" in msg else 1
|
||||
msg = msg.replace("r18", "").strip().split()
|
||||
msg = [m.strip() for m in msg if m]
|
||||
keyword = None
|
||||
info_list = None
|
||||
num = 10
|
||||
page = 1
|
||||
if (n := len(msg)) == 1:
|
||||
keyword = msg[0]
|
||||
if n > 1:
|
||||
if not is_number(msg[1]):
|
||||
await pixiv_keyword.finish("图片数量必须是数字！", at_sender=True)
|
||||
num = int(msg[1])
|
||||
if n > 2:
|
||||
if not is_number(msg[2]):
|
||||
await pixiv_keyword.finish("页数数量必须是数字！", at_sender=True)
|
||||
page = int(msg[2])
|
||||
if keyword:
|
||||
info_list, code = await search_pixiv_urls(keyword, num, page, r18)
|
||||
if code != 200:
|
||||
await pixiv_keyword.finish(info_list[0])
|
||||
if not info_list:
|
||||
await pixiv_keyword.finish("没有找到啊，等等再试试吧~V", at_sender=True)
|
||||
await send_image(info_list, pixiv_keyword, bot, event)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||||
f" 查看了搜索 {keyword} R18：{r18}"
|
||||
)
|
||||
|
||||
|
||||
def check_date(date):
|
||||
try:
|
||||
time.strptime(date, "%Y-%m-%d")
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
|
||||
async def send_image(
|
||||
info_list: list, matcher: Type[Matcher], bot: Bot, event: MessageEvent
|
||||
):
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
await pixiv_rank.send("开始下载整理数据...")
|
||||
idx = 0
|
||||
mes_list = []
|
||||
for title, author, urls in info_list:
|
||||
_message = f"title: {title}\nauthor: {author}\n" + await download_pixiv_imgs(urls, event.user_id, idx)
|
||||
mes_list.append(_message)
|
||||
idx += 1
|
||||
mes_list = custom_forward_msg(mes_list, bot.self_id)
|
||||
await bot.send_group_forward_msg(group_id=event.group_id, messages=mes_list)
|
||||
else:
|
||||
for title, author, urls in info_list:
|
||||
try:
|
||||
await matcher.send(
|
||||
f"title: {title}\n"
|
||||
f"author: {author}\n"
|
||||
+ await download_pixiv_imgs(urls, event.user_id)
|
||||
)
|
||||
except (NetworkError, TimeoutError, ClientConnectorError):
|
||||
await matcher.send("这张图网络直接炸掉了！", at_sender=True)
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
|
||||
from nonebot.matcher import Matcher
|
||||
from nonebot import on_command
|
||||
from utils.utils import get_message_text, is_number
|
||||
from .data_source import get_pixiv_urls, download_pixiv_imgs, search_pixiv_urls
|
||||
from services.log import logger
|
||||
from nonebot.adapters.cqhttp.exception import NetworkError
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from utils.message_builder import custom_forward_msg
|
||||
from configs.config import Config
|
||||
from typing import Type
|
||||
from nonebot.rule import to_me
|
||||
import time
|
||||
|
||||
__zx_plugin_name__ = "P站排行/搜图"
|
||||
|
||||
__plugin_usage__ = """
|
||||
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
|
||||
【默认为 热度排序】
|
||||
【注意空格！！】【在线搜索会较慢】【数量可能不符？可能该页数量不够，也可能被R-18屏蔽】
|
||||
""".strip()
|
||||
__plugin_des__ = "P站排行榜直接冲，P站搜图跟着冲"
|
||||
__plugin_cmd__ = ["p站排行 ?[参数] ?[数量] ?[日期]", "搜图 [关键词] ?[数量] ?[页数=1] ?[r18](不屏蔽R-18)"]
|
||||
__plugin_type__ = ("来点好康的",)
|
||||
__plugin_version__ = 0.1
|
||||
__plugin_author__ = "HibiKier"
|
||||
__plugin_settings__ = {
|
||||
"level": 9,
|
||||
"default_status": True,
|
||||
"limit_superuser": False,
|
||||
"cmd": ["p站排行", "搜图", "p站搜图", "P站搜图"],
|
||||
}
|
||||
__plugin_block_limit__ = {"rst": "P站排行榜或搜图正在搜索，请不要重复触发命令..."}
|
||||
__plugin_configs__ = {
|
||||
"TIMEOUT": {
|
||||
"value": 10,
|
||||
"help": "图片下载超时限制",
|
||||
"default_value": 10
|
||||
}
|
||||
}
|
||||
Config.add_plugin_config(
|
||||
"hibiapi",
|
||||
"HIBIAPI",
|
||||
"https://api.obfs.dev",
|
||||
help_="如果没有自建或其他hibiapi请不要修改",
|
||||
default_value="https://api.obfs.dev",
|
||||
)
|
||||
Config.add_plugin_config(
|
||||
"pixiv",
|
||||
"PIXIV_NGINX_URL",
|
||||
"i.pixiv.re",
|
||||
help_="Pixiv反向代理"
|
||||
)
|
||||
|
||||
|
||||
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",
|
||||
}
|
||||
|
||||
|
||||
pixiv_rank = on_command(
|
||||
"p站排行",
|
||||
aliases={"P站排行榜", "p站排行榜", "P站排行榜", "P站排行"},
|
||||
priority=5,
|
||||
block=True,
|
||||
rule=to_me(),
|
||||
)
|
||||
pixiv_keyword = on_command("搜图", priority=5, block=True, rule=to_me())
|
||||
|
||||
|
||||
@pixiv_rank.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
msg = get_message_text(event.json()).strip()
|
||||
msg = msg.split(" ")
|
||||
msg = [m for m in msg if m]
|
||||
code = 0
|
||||
info_list = []
|
||||
if not msg:
|
||||
msg = ["1"]
|
||||
if msg[0] in ["6", "7", "8", "9"]:
|
||||
if event.message_type == "group":
|
||||
await pixiv_rank.finish("羞羞脸！私聊里自己看！", at_sender=True)
|
||||
if (n := len(msg)) == 0 or msg[0] == "":
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get("1"))
|
||||
elif n == 1:
|
||||
if msg[0] not in ["1", "2", "3", "4", "5", "6", "7", "8", "9"]:
|
||||
await pixiv_rank.finish("要好好输入要看什么类型的排行榜呀！", at_sender=True)
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get(msg[0]))
|
||||
elif n == 2:
|
||||
info_list, code = await get_pixiv_urls(rank_dict.get(msg[0]), int(msg[1]))
|
||||
elif n == 3:
|
||||
if not check_date(msg[2]):
|
||||
await pixiv_rank.finish("日期格式错误了", at_sender=True)
|
||||
info_list, code = await get_pixiv_urls(
|
||||
rank_dict.get(msg[0]), int(msg[1]), date=msg[2]
|
||||
)
|
||||
else:
|
||||
await pixiv_rank.finish("格式错了噢，参数不够？看看帮助？", at_sender=True)
|
||||
if code != 200 and info_list:
|
||||
await pixiv_rank.finish(info_list[0])
|
||||
if not info_list:
|
||||
await pixiv_rank.finish("没有找到啊，等等再试试吧~V", at_sender=True)
|
||||
await send_image(info_list, pixiv_rank, bot, event)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||||
f" 查看了P站排行榜 code：{msg[0]}"
|
||||
)
|
||||
|
||||
|
||||
@pixiv_keyword.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
msg = get_message_text(event.json())
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
if "r18" in msg.lower():
|
||||
await pixiv_keyword.finish("(脸红#) 你不会害羞的 八嘎！", at_sender=True)
|
||||
r18 = 0 if "r18" in msg else 1
|
||||
msg = msg.replace("r18", "").strip().split()
|
||||
msg = [m.strip() for m in msg if m]
|
||||
keyword = None
|
||||
info_list = None
|
||||
num = 10
|
||||
page = 1
|
||||
if (n := len(msg)) == 1:
|
||||
keyword = msg[0]
|
||||
if n > 1:
|
||||
if not is_number(msg[1]):
|
||||
await pixiv_keyword.finish("图片数量必须是数字！", at_sender=True)
|
||||
num = int(msg[1])
|
||||
if n > 2:
|
||||
if not is_number(msg[2]):
|
||||
await pixiv_keyword.finish("页数数量必须是数字！", at_sender=True)
|
||||
page = int(msg[2])
|
||||
if keyword:
|
||||
info_list, code = await search_pixiv_urls(keyword, num, page, r18)
|
||||
if code != 200:
|
||||
await pixiv_keyword.finish(info_list[0])
|
||||
if not info_list:
|
||||
await pixiv_keyword.finish("没有找到啊，等等再试试吧~V", at_sender=True)
|
||||
await send_image(info_list, pixiv_keyword, bot, event)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||||
f" 查看了搜索 {keyword} R18：{r18}"
|
||||
)
|
||||
|
||||
|
||||
def check_date(date):
|
||||
try:
|
||||
time.strptime(date, "%Y-%m-%d")
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
|
||||
async def send_image(
|
||||
info_list: list, matcher: Type[Matcher], bot: Bot, event: MessageEvent
|
||||
):
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
await pixiv_rank.send("开始下载整理数据...")
|
||||
idx = 0
|
||||
mes_list = []
|
||||
for title, author, urls in info_list:
|
||||
_message = f"title: {title}\nauthor: {author}\n" + await download_pixiv_imgs(urls, event.user_id, idx)
|
||||
mes_list.append(_message)
|
||||
idx += 1
|
||||
mes_list = custom_forward_msg(mes_list, bot.self_id)
|
||||
await bot.send_group_forward_msg(group_id=event.group_id, messages=mes_list)
|
||||
else:
|
||||
for title, author, urls in info_list:
|
||||
try:
|
||||
await matcher.send(
|
||||
f"title: {title}\n"
|
||||
f"author: {author}\n"
|
||||
+ await download_pixiv_imgs(urls, event.user_id)
|
||||
)
|
||||
except (NetworkError, TimeoutError):
|
||||
await matcher.send("这张图网络直接炸掉了！", at_sender=True)
|
||||
|
||||
Reference in New Issue
Block a user