mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-25 15:42:18 +00:00
54dc353ded
Co-authored-by: Sourcery AI <> Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com>
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
from pagermaid.listener import listener
|
||
from pagermaid.single_utils import safe_remove
|
||
from pagermaid.enums import Client, Message, AsyncClient
|
||
from pyrogram.types import InputMediaPhoto
|
||
|
||
|
||
async def get_result(message, request, r18=0):
|
||
# r18: 0为非 R18，1为 R18，2为混合（在库中的分类，不等同于作品本身的 R18 标识）
|
||
# num: 图片的数量
|
||
# size: 返回图片的尺寸质量
|
||
size = "regular"
|
||
des = "出错了，没有纸片人看了。"
|
||
data = await request.get((f"https://api.lolicon.app/setu/v2?num=5&r18={r18}&size={size}"))
|
||
if data.status_code != 200:
|
||
return None, None, "连接二次元大门出错。。。"
|
||
await message.edit("已进入二次元 . . .")
|
||
try:
|
||
result = data.json()['data']
|
||
except Exception:
|
||
return None, None, "解析JSON出错。"
|
||
setuList = [] # 发送
|
||
delList = [] # 删除
|
||
await message.edit("努力获取中 。。。")
|
||
for i in range(5):
|
||
urls = result[i]['urls'][size]
|
||
imgname = (f"{result[i]['pid']}_{i}.png")
|
||
try:
|
||
img = await request.get(urls)
|
||
with open(imgname, mode="wb") as f:
|
||
f.write(img.content)
|
||
except Exception:
|
||
return None, None, "连接二次元出错。。。"
|
||
setuList.append(InputMediaPhoto(imgname))
|
||
delList.append(imgname)
|
||
return setuList, delList, des if setuList else None
|
||
|
||
|
||
@listener(command="zpr",
|
||
description="随机获取一组涩涩纸片人。",
|
||
parameters="{r18}")
|
||
async def zpr(client: Client, message: Message, request: AsyncClient):
|
||
arguments = message.arguments.upper().strip()
|
||
message = await message.edit("正在前往二次元。。。")
|
||
try:
|
||
photoList, delList, des = await get_result(message, request, r18=1 if arguments == "R18" else 0)
|
||
except Exception as e:
|
||
return await message.edit(f"发生错误：{e}")
|
||
if not photoList:
|
||
return await message.edit(des)
|
||
try:
|
||
await message.edit("传送中。。。")
|
||
await client.send_media_group(message.chat.id, photoList)
|
||
for i in range(5):
|
||
safe_remove(delList[i])
|
||
except Exception:
|
||
await client.send_message(message.chat.id, des)
|
||
await message.safe_delete()
|