mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-09 22:00:01 +08:00
update code
This commit is contained in:
HibiKier
+87
-123
@@ -1,17 +1,16 @@
|
||||
from nonebot.typing import T_State
|
||||
from nonebot.adapters.cqhttp import Bot, MessageEvent, Event, GroupMessageEvent, Message
|
||||
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, UserExistLimiter, is_number
|
||||
from utils.utils import get_message_text, is_number
|
||||
from .data_source import get_pixiv_urls, download_pixiv_imgs, search_pixiv_urls
|
||||
from utils.message_builder import at
|
||||
from services.log import logger
|
||||
from nonebot.adapters.cqhttp.exception import NetworkError
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from aiohttp.client_exceptions import ClientConnectorError
|
||||
from nonebot.exception import IgnoredException
|
||||
from nonebot.message import run_preprocessor, run_postprocessor
|
||||
from typing import Optional
|
||||
from configs.config import NICKNAME
|
||||
from typing import Type
|
||||
from nonebot.rule import to_me
|
||||
import time
|
||||
|
||||
__plugin_name__ = "P站"
|
||||
@@ -38,17 +37,13 @@ p站排行榜 [参数] [数量](可选) [日期](可选)
|
||||
【注意空格！！】【在线搜索会较慢】
|
||||
---------------------------------
|
||||
'P站搜图帮助：
|
||||
可选参数：
|
||||
1.热度排序
|
||||
2.时间排序
|
||||
【使用时选择参数序号即可，R18仅可私聊】
|
||||
搜图 [关键词] [数量](可选) [排序方式](可选) [r18](可选)
|
||||
搜图 [关键词] [数量](可选) [页数](可选默认1) [r18](不屏蔽R-18，可选)
|
||||
示例：
|
||||
搜图 樱岛麻衣
|
||||
搜图 樱岛麻衣 5 1
|
||||
搜图 樱岛麻衣 5 2 r18
|
||||
搜图 樱岛麻衣 5
|
||||
搜图 樱岛麻衣 5 r18
|
||||
【默认为 热度排序】
|
||||
【注意空格！！】【在线搜索会较慢】【数量可能不符】
|
||||
【注意空格！！】【在线搜索会较慢】【数量可能不符？可能该页数量不够，也可能被R-18屏蔽】
|
||||
"""
|
||||
|
||||
rank_dict = {
|
||||
@@ -63,142 +58,84 @@ rank_dict = {
|
||||
"9": "week_r18g",
|
||||
}
|
||||
|
||||
_ulmt = UserExistLimiter()
|
||||
|
||||
|
||||
@run_preprocessor
|
||||
async def _(matcher: Matcher, bot: Bot, event: Event, state: T_State):
|
||||
if isinstance(event, MessageEvent):
|
||||
if matcher.module == "pixiv":
|
||||
if _ulmt.check(event.user_id):
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
await bot.send_group_msg(
|
||||
group_id=event.group_id,
|
||||
message=Message(f"{at(event.user_id)}P站排行榜或搜图正在搜索噢，不要重复触发命令呀"),
|
||||
)
|
||||
else:
|
||||
await bot.send_private_msg(
|
||||
user_id=event.user_id, message=f"P站排行榜或搜图正在搜索噢，不要重复触发命令呀"
|
||||
)
|
||||
raise IgnoredException("pixiv插件正在访问！")
|
||||
|
||||
|
||||
@run_postprocessor
|
||||
async def do_something(
|
||||
matcher: Matcher,
|
||||
exception: Optional[Exception],
|
||||
bot: Bot,
|
||||
event: Event,
|
||||
state: T_State,
|
||||
):
|
||||
if isinstance(event, MessageEvent):
|
||||
if matcher.module == "pixiv":
|
||||
_ulmt.set_False(event.user_id)
|
||||
|
||||
|
||||
pixiv_rank = on_command(
|
||||
"p站排行", aliases={"P站排行榜", "p站排行榜", "P站排行榜"}, priority=5, block=True
|
||||
"p站排行", aliases={"P站排行榜", "p站排行榜", "P站排行榜", "P站排行"}, priority=5, block=True, rule=to_me()
|
||||
)
|
||||
pixiv_keyword = on_command("搜图", priority=5, block=True)
|
||||
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()
|
||||
_ulmt.set_True(event.user_id)
|
||||
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 len(msg) == 0 or msg[0] == "":
|
||||
text_list, urls, code = await get_pixiv_urls(rank_dict.get("1"))
|
||||
elif len(msg) == 1:
|
||||
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)
|
||||
text_list, urls, code = await get_pixiv_urls(rank_dict.get(msg[0]))
|
||||
elif len(msg) == 2:
|
||||
text_list, urls, code = await get_pixiv_urls(rank_dict.get(msg[0]), int(msg[1]))
|
||||
elif len(msg) == 3:
|
||||
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)
|
||||
text_list, urls, code = await get_pixiv_urls(
|
||||
rank_dict.get(msg[0]), int(msg[1]), msg[2]
|
||||
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:
|
||||
await pixiv_keyword.finish(text_list[0])
|
||||
else:
|
||||
if not text_list or not urls:
|
||||
await pixiv_rank.finish("没有找到啊，等等再试试吧~V", at_sender=True)
|
||||
for i in range(len(text_list)):
|
||||
try:
|
||||
await pixiv_rank.send(
|
||||
text_list[i] + await download_pixiv_imgs(urls[i], event.user_id)
|
||||
)
|
||||
except (NetworkError, TimeoutError, ClientConnectorError):
|
||||
await pixiv_keyword.send("这张图网络炸了！", at_sender=True)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
|
||||
f" 查看了P站排行榜 code：{msg[0]}"
|
||||
)
|
||||
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()).strip()
|
||||
_ulmt.set_True(event.user_id)
|
||||
if event.message_type == "group":
|
||||
if msg.find("r18") != -1:
|
||||
msg = get_message_text(event.json())
|
||||
if isinstance(event, GroupMessageEvent):
|
||||
if 'r18' in msg.lower():
|
||||
await pixiv_keyword.finish("(脸红#) 你不会害羞的 八嘎！", at_sender=True)
|
||||
if msg.find("r18") == -1:
|
||||
r18 = 1
|
||||
else:
|
||||
r18 = 2
|
||||
msg = msg.replace("r18", "").strip()
|
||||
msg = msg.split(" ")
|
||||
msg = [m for m in msg if m]
|
||||
if len(msg) == 1:
|
||||
keyword = msg[0].strip()
|
||||
num = 5
|
||||
order = "popular"
|
||||
elif len(msg) == 2:
|
||||
keyword = msg[0].strip()
|
||||
if not is_number(msg[1].strip()):
|
||||
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].strip())
|
||||
order = "popular"
|
||||
elif len(msg) == 3:
|
||||
keyword = msg[0].strip()
|
||||
if not is_number(msg[1].strip()):
|
||||
await pixiv_keyword.finish("图片数量必须是数字！", at_sender=True)
|
||||
num = int(msg[1].strip())
|
||||
if not is_number(msg[2].strip()):
|
||||
await pixiv_keyword.finish("排序方式必须是数字！", at_sender=True)
|
||||
if msg[2].strip() == "1":
|
||||
order = "popular"
|
||||
else:
|
||||
order = "xxx"
|
||||
else:
|
||||
await pixiv_keyword.finish("参数不正确，一定要好好看看帮助啊！", at_sender=True)
|
||||
text_list, urls, code = await search_pixiv_urls(keyword, num, order, r18)
|
||||
if code != 200:
|
||||
await pixiv_keyword.finish(text_list[0])
|
||||
else:
|
||||
for i in range(len(text_list)):
|
||||
try:
|
||||
await pixiv_keyword.send(
|
||||
text_list[i] + await download_pixiv_imgs(urls[i], event.user_id)
|
||||
)
|
||||
except (NetworkError, TimeoutError, ClientConnectorError):
|
||||
await pixiv_keyword.send("这张图网络炸了！", at_sender=True)
|
||||
logger.info(
|
||||
f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'})"
|
||||
f" 查看了搜索 {keyword} R18：{r18}"
|
||||
)
|
||||
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])
|
||||
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):
|
||||
@@ -207,3 +144,30 @@ def check_date(date):
|
||||
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)
|
||||
data = {
|
||||
"type": "node",
|
||||
"data": {
|
||||
"name": f"这里是{NICKNAME}酱",
|
||||
"uin": f"{bot.self_id}",
|
||||
"content": _message,
|
||||
},
|
||||
}
|
||||
mes_list.append(data)
|
||||
idx += 1
|
||||
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)
|
||||
|
||||
+101
-69
@@ -1,14 +1,13 @@
|
||||
import aiohttp
|
||||
import aiofiles
|
||||
from configs.path_config import IMAGE_PATH
|
||||
from utils.utils import get_local_proxy
|
||||
from utils.user_agent import get_user_agent
|
||||
from bs4 import BeautifulSoup
|
||||
import feedparser
|
||||
from utils.message_builder import image
|
||||
from asyncio.exceptions import TimeoutError
|
||||
from configs.config import RSSHUBAPP
|
||||
from configs.config import HIBIAPI
|
||||
from aiohttp.client_exceptions import ClientConnectorError
|
||||
from typing import Optional
|
||||
from pathlib import Path
|
||||
import aiohttp
|
||||
import aiofiles
|
||||
import platform
|
||||
|
||||
if platform.system() == "Windows":
|
||||
@@ -17,82 +16,115 @@ if platform.system() == "Windows":
|
||||
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
||||
|
||||
|
||||
async def get_pixiv_urls(mode: str, num: int = 5, date: str = "") -> "list, list, int":
|
||||
url = f"{RSSHUBAPP}/pixiv/ranking/{mode}"
|
||||
rank_url = f"{HIBIAPI}/api/pixiv/rank"
|
||||
search_url = f"{HIBIAPI}/api/pixiv/search"
|
||||
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6;"
|
||||
" rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
|
||||
"Referer": "https://www.pixiv.net",
|
||||
}
|
||||
|
||||
|
||||
async def get_pixiv_urls(
|
||||
mode: str, num: int = 10, page: int = 1, date: Optional[str] = None
|
||||
) -> "list, int":
|
||||
params = {"mode": mode, "page": page}
|
||||
if date:
|
||||
url += f"/{date}"
|
||||
try:
|
||||
return await parser_data(url, num)
|
||||
except ClientConnectorError:
|
||||
return await get_pixiv_urls(mode, num, date)
|
||||
|
||||
|
||||
async def download_pixiv_imgs(urls: list, user_id: int) -> str:
|
||||
result = ""
|
||||
index = 0
|
||||
for img in urls:
|
||||
async with aiohttp.ClientSession(headers=get_user_agent()) as session:
|
||||
for _ in range(3):
|
||||
async with session.get(
|
||||
img, proxy=get_local_proxy(), timeout=2
|
||||
) as response:
|
||||
async with aiofiles.open(
|
||||
IMAGE_PATH + f"temp/{user_id}_{index}_pixiv.jpg", "wb"
|
||||
) as f:
|
||||
try:
|
||||
await f.write(await response.read())
|
||||
result += image(f"{user_id}_{index}_pixiv.jpg", "temp")
|
||||
index += 1
|
||||
break
|
||||
except TimeoutError:
|
||||
# result += '\n这张图下载失败了..\n'
|
||||
pass
|
||||
else:
|
||||
result += "\n这张图下载失败了..\n"
|
||||
return result
|
||||
params["date"] = date
|
||||
return await parser_data(rank_url, num, params, 'rank')
|
||||
|
||||
|
||||
async def search_pixiv_urls(
|
||||
keyword: str, num: int, order: str, r18: int
|
||||
keyword: str, num: int, page: int, r18: int
|
||||
) -> "list, list":
|
||||
url = f"{RSSHUBAPP}/pixiv/search/{keyword}/{order}/{r18}"
|
||||
return await parser_data(url, num)
|
||||
params = {"word": keyword, 'page': page}
|
||||
return await parser_data(search_url, num, params, 'search', r18)
|
||||
|
||||
|
||||
async def parser_data(url: str, num: int) -> "list, list, int":
|
||||
text_list = []
|
||||
urls = []
|
||||
async def parser_data(url: str, num: int, params: dict, type_: str, r18: int = 0) -> "list, int":
|
||||
info_list = []
|
||||
async with aiohttp.ClientSession() as session:
|
||||
for _ in range(3):
|
||||
try:
|
||||
async with session.get(
|
||||
url, proxy=get_local_proxy(), timeout=2
|
||||
url, params=params, proxy=get_local_proxy(), timeout=5
|
||||
) as response:
|
||||
data = feedparser.parse(await response.text())["entries"]
|
||||
break
|
||||
except TimeoutError:
|
||||
if response.status == 200:
|
||||
data = await response.json()
|
||||
if data.get("illusts"):
|
||||
data = data["illusts"]
|
||||
break
|
||||
except (TimeoutError, ClientConnectorError):
|
||||
pass
|
||||
else:
|
||||
return ["网络不太好，也许过一会就好了"], [], 998
|
||||
try:
|
||||
if len(data) == 0:
|
||||
return ["没有搜索到喔"], [], 997
|
||||
if num > len(data):
|
||||
num = len(data)
|
||||
data = data[:num]
|
||||
for data in data:
|
||||
soup = BeautifulSoup(data["summary"], "lxml")
|
||||
title = "标题：" + data["title"]
|
||||
pl = soup.find_all("p")
|
||||
author = pl[0].text.split("-")[0].strip()
|
||||
imgs = []
|
||||
text_list.append(f"{title}\n{author}\n")
|
||||
for p in pl[1:]:
|
||||
imgs.append(p.find("img").get("src"))
|
||||
urls.append(imgs)
|
||||
except ValueError:
|
||||
return ["是网站坏了啊，也许过一会就好了"], [], 999
|
||||
return text_list, urls, 200
|
||||
return ["网络不太好？没有该页数？也许过一会就好了..."], 998
|
||||
num = num if num < 30 else 30
|
||||
data = data[:num]
|
||||
for x in data:
|
||||
if type_ == 'search' and r18 == 1:
|
||||
if 'R-18' in str(x['tags']):
|
||||
continue
|
||||
title = x["title"]
|
||||
author = x["user"]["name"]
|
||||
urls = []
|
||||
if x["page_count"] == 1:
|
||||
urls.append(x["image_urls"]["large"])
|
||||
else:
|
||||
for j in x["meta_pages"]:
|
||||
urls.append(j["image_urls"]["large"])
|
||||
info_list.append((title, author, urls))
|
||||
return info_list, 200
|
||||
|
||||
|
||||
# asyncio.get_event_loop().run_until_complete(get_pixiv_urls('day'))
|
||||
async def download_pixiv_imgs(
|
||||
urls: list, user_id: int, forward_msg_index: int = None
|
||||
) -> str:
|
||||
result = ""
|
||||
index = 0
|
||||
for url in urls:
|
||||
async with aiohttp.ClientSession(headers=headers) as session:
|
||||
for _ in range(3):
|
||||
try:
|
||||
async with session.get(
|
||||
url, proxy=get_local_proxy(), timeout=3
|
||||
) as response:
|
||||
if response.status == 200:
|
||||
try:
|
||||
file = (
|
||||
f"{IMAGE_PATH}/temp/{user_id}_{forward_msg_index}_{index}_pixiv.jpg"
|
||||
if forward_msg_index is not None
|
||||
else f"{IMAGE_PATH}/temp/{user_id}_{index}_pixiv.jpg"
|
||||
)
|
||||
file = Path(file)
|
||||
if forward_msg_index is not None:
|
||||
async with aiofiles.open(
|
||||
file,
|
||||
"wb",
|
||||
) as f:
|
||||
await f.write(await response.read())
|
||||
result += image(
|
||||
f"{user_id}_{forward_msg_index}_{index}_pixiv.jpg",
|
||||
"temp",
|
||||
)
|
||||
else:
|
||||
async with aiofiles.open(
|
||||
file,
|
||||
"wb",
|
||||
) as f:
|
||||
await f.write(await response.read())
|
||||
result += image(
|
||||
f"{user_id}_{index}_pixiv.jpg", "temp"
|
||||
)
|
||||
index += 1
|
||||
break
|
||||
except OSError:
|
||||
file.unlink()
|
||||
except (TimeoutError, ClientConnectorError):
|
||||
# result += '\n这张图下载失败了..\n'
|
||||
pass
|
||||
else:
|
||||
result += "\n这张图下载失败了..\n"
|
||||
return result
|
||||
|
||||
|
||||
Reference in New Issue
Block a user