PagerMaid_Plugins_Pyro/httpcat/main.py
2023-07-01 20:18:58 +08:00

30 lines
1008 B
Python

from io import BytesIO
from pagermaid.enums import Client, Message, AsyncClient
from pagermaid.listener import listener
@listener(command="httpcat", description="获取 HTTP 状态码的图片。", parameters="[http 状态码]")
async def httpcat(client: Client, message: Message, request: AsyncClient):
try:
code = int(message.arguments)
except ValueError:
return await message.edit("http 状态码错误。")
if code < 100 or code > 599:
return await message.edit("http 状态码错误。")
pic = await request.get(f"https://http.cat/{code}.jpg")
if pic.status_code != 200:
return await message.edit("http 状态码错误。")
io = BytesIO(pic.content)
io.name = "1.jpg"
await client.send_photo(
message.chat.id,
io,
reply_to_message_id=(
message.reply_to_message_id or message.reply_to_top_message_id
)
if message.outgoing
else message.id,
)
await message.safe_delete()