TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-11-16 11:42:15 +00:00
Code Issues Actions Projects Releases Wiki Activity

zpr 修复路径错误

Browse Source
This commit is contained in:
omg-xtao 2023-03-12 21:40:36 +08:00 committed by GitHub
parent 482ae0c0e9
commit 42a7eba065
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
zpr/main.py
View File

@ -6,21 +6,20 @@ from pagermaid.listener import listener
from pagermaid.enums import Client, Message, AsyncClient
from pyrogram.types import InputMediaPhoto
from pyrogram.errors import RPCError
from pathlib import Path
pixiv_img_host = "pixiv.yuki.sh"
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36 Edg/106.0.1370.42'
}
data_path = Path("data/zpr")
async def get_result(message, request, r18=2):
# r18: 0为非 R18，1为 R18，2为混合（在库中的分类，不等同于作品本身的 R18 标识）
# num: 图片的数量
# size: 返回图片的尺寸质量
zpr_path = f"{dirname(abspath(__file__))}/zpr/"
if not exists(zpr_path):
makedirs(zpr_path)
data_path.mkdir(parents=True, exist_ok=True)
size = "regular"
des = "出错了，没有纸片人看了。"
data = await request.get(
@ -29,25 +28,28 @@ async def get_result(message, request, r18=2):
timeout=10,
)
if data.status_code != 200:
return None, None, "连接二次元大门出错。。。"
return None, "连接二次元大门出错。。。"
await message.edit("已进入二次元 . . .")
try:
result = data.json()['data']
except Exception:
return None, None, "解析JSON出错。"
return None, "解析JSON出错。"
setu_list = [] # 发送
await message.edit("努力获取中 。。。")
for i in range(5):
urls = result[i]['urls'][size].replace('i.pixiv.re', pixiv_img_host)
img_name = f"{result[i]['pid']}_{i}.jpg"
file_path = zpr_path / img_name
try:
img = await request.get(urls, headers=headers, timeout=10)
with open(f"{zpr_path}{img_name}", mode="wb") as f:
if img.status_code != 200:
continue
with open(file_path, mode="wb") as f:
f.write(img.content)
except Exception:
return None, None, "连接二次元出错。。。"
setu_list.append(InputMediaPhoto(media=f"{zpr_path}{img_name}"))
return setu_list, zpr_path, des if setu_list else None
setu_list.append(InputMediaPhoto(media=str(file_path)))
return setu_list, des if setu_list else None
@listener(command="zpr",
@ -58,9 +60,9 @@ async def zpr(client: Client, message: Message, request: AsyncClient):
message_thread_id = message.reply_to_top_message_id or message.reply_to_message_id
message = await message.edit("正在前往二次元。。。")
try:
photoList, zpr_path, des = await get_result(message, request, r18=2 if arguments == "R18" else 0)
photoList, des = await get_result(message, request, r18=2 if arguments == "R18" else 0)
if not photoList:
shutil.rmtree(zpr_path)
shutil.rmtree("data/zpr")
return await message.edit(des)
with contextlib.suppress(Exception):
await message.edit("传送中。。。")