zpr 随机获取一组涩涩纸片人 (#92)

Co-authored-by: Sourcery AI <>
Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com>

zpr/main.py

@@ -0,0 +1,57 @@
+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()