mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 09:44:50 +00:00
289 lines
9.7 KiB
Python
289 lines
9.7 KiB
Python
# * @author Yasir Aris M <yasiramunandar@gmail.com>
|
|
# * @date 2023-06-21 22:12:27
|
|
# * @projectName MissKatyPyro
|
|
# * Copyright ©YasirPedia All rights reserved
|
|
from io import BytesIO
|
|
|
|
from pyrogram import Client, filters
|
|
from pyrogram.types import Message
|
|
|
|
from misskaty import app
|
|
from misskaty.helper.http import fetch
|
|
|
|
|
|
class QuotlyException(Exception):
|
|
pass
|
|
|
|
|
|
async def get_message_sender_id(ctx: Message):
|
|
if ctx.forward_date:
|
|
if ctx.forward_sender_name:
|
|
return 1
|
|
elif ctx.forward_from:
|
|
return ctx.forward_from.id
|
|
elif ctx.forward_from_chat:
|
|
return ctx.forward_from_chat.id
|
|
else:
|
|
return 1
|
|
elif ctx.from_user:
|
|
return ctx.from_user.id
|
|
elif ctx.sender_chat:
|
|
return ctx.sender_chat.id
|
|
else:
|
|
return 1
|
|
|
|
|
|
async def get_message_sender_name(ctx: Message):
|
|
if ctx.forward_date:
|
|
if ctx.forward_sender_name:
|
|
return ctx.forward_sender_name
|
|
elif ctx.forward_from:
|
|
return (
|
|
f"{ctx.forward_from.first_name} {ctx.forward_from.last_name}"
|
|
if ctx.forward_from.last_name
|
|
else ctx.forward_from.first_name
|
|
)
|
|
|
|
elif ctx.forward_from_chat:
|
|
return ctx.forward_from_chat.title
|
|
else:
|
|
return ""
|
|
elif ctx.from_user:
|
|
if ctx.from_user.last_name:
|
|
return f"{ctx.from_user.first_name} {ctx.from_user.last_name}"
|
|
else:
|
|
return ctx.from_user.first_name
|
|
elif ctx.sender_chat:
|
|
return ctx.sender_chat.title
|
|
else:
|
|
return ""
|
|
|
|
|
|
async def get_custom_emoji(ctx: Message):
|
|
if ctx.forward_date:
|
|
return (
|
|
""
|
|
if ctx.forward_sender_name
|
|
or not ctx.forward_from
|
|
and ctx.forward_from_chat
|
|
or not ctx.forward_from
|
|
else ctx.forward_from.emoji_status.custom_emoji_id
|
|
)
|
|
|
|
return ctx.from_user.emoji_status.custom_emoji_id if ctx.from_user else ""
|
|
|
|
|
|
async def get_message_sender_username(ctx: Message):
|
|
if ctx.forward_date:
|
|
if (
|
|
not ctx.forward_sender_name
|
|
and not ctx.forward_from
|
|
and ctx.forward_from_chat
|
|
and ctx.forward_from_chat.username
|
|
):
|
|
return ctx.forward_from_chat.username
|
|
elif (
|
|
not ctx.forward_sender_name
|
|
and not ctx.forward_from
|
|
and ctx.forward_from_chat
|
|
or ctx.forward_sender_name
|
|
or not ctx.forward_from
|
|
):
|
|
return ""
|
|
else:
|
|
return ctx.forward_from.username or ""
|
|
elif ctx.from_user and ctx.from_user.username:
|
|
return ctx.from_user.username
|
|
elif (
|
|
ctx.from_user
|
|
or ctx.sender_chat
|
|
and not ctx.sender_chat.username
|
|
or not ctx.sender_chat
|
|
):
|
|
return ""
|
|
else:
|
|
return ctx.sender_chat.username
|
|
|
|
|
|
async def get_message_sender_photo(ctx: Message):
|
|
if ctx.forward_date:
|
|
if (
|
|
not ctx.forward_sender_name
|
|
and not ctx.forward_from
|
|
and ctx.forward_from_chat
|
|
and ctx.forward_from_chat.photo
|
|
):
|
|
return {
|
|
"small_file_id": ctx.forward_from_chat.photo.small_file_id,
|
|
"small_photo_unique_id": ctx.forward_from_chat.photo.small_photo_unique_id,
|
|
"big_file_id": ctx.forward_from_chat.photo.big_file_id,
|
|
"big_photo_unique_id": ctx.forward_from_chat.photo.big_photo_unique_id,
|
|
}
|
|
elif (
|
|
not ctx.forward_sender_name
|
|
and not ctx.forward_from
|
|
and ctx.forward_from_chat
|
|
or ctx.forward_sender_name
|
|
or not ctx.forward_from
|
|
):
|
|
return ""
|
|
else:
|
|
return (
|
|
{
|
|
"small_file_id": ctx.forward_from.photo.small_file_id,
|
|
"small_photo_unique_id": ctx.forward_from.photo.small_photo_unique_id,
|
|
"big_file_id": ctx.forward_from.photo.big_file_id,
|
|
"big_photo_unique_id": ctx.forward_from.photo.big_photo_unique_id,
|
|
}
|
|
if ctx.forward_from.photo
|
|
else ""
|
|
)
|
|
|
|
elif ctx.from_user and ctx.from_user.photo:
|
|
return {
|
|
"small_file_id": ctx.from_user.photo.small_file_id,
|
|
"small_photo_unique_id": ctx.from_user.photo.small_photo_unique_id,
|
|
"big_file_id": ctx.from_user.photo.big_file_id,
|
|
"big_photo_unique_id": ctx.from_user.photo.big_photo_unique_id,
|
|
}
|
|
elif (
|
|
ctx.from_user
|
|
or ctx.sender_chat
|
|
and not ctx.sender_chat.photo
|
|
or not ctx.sender_chat
|
|
):
|
|
return ""
|
|
else:
|
|
return {
|
|
"small_file_id": ctx.sender_chat.photo.small_file_id,
|
|
"small_photo_unique_id": ctx.sender_chat.photo.small_photo_unique_id,
|
|
"big_file_id": ctx.sender_chat.photo.big_file_id,
|
|
"big_photo_unique_id": ctx.sender_chat.photo.big_photo_unique_id,
|
|
}
|
|
|
|
|
|
async def get_text_or_caption(ctx: Message):
|
|
if ctx.text:
|
|
return ctx.text
|
|
elif ctx.caption:
|
|
return ctx.caption
|
|
else:
|
|
return ""
|
|
|
|
|
|
async def pyrogram_to_quotly(messages, is_reply):
|
|
if not isinstance(messages, list):
|
|
messages = [messages]
|
|
payload = {
|
|
"type": "quote",
|
|
"format": "png",
|
|
"backgroundColor": "#1b1429",
|
|
"messages": [],
|
|
}
|
|
|
|
for message in messages:
|
|
the_message_dict_to_append = {}
|
|
if message.entities:
|
|
the_message_dict_to_append["entities"] = [
|
|
{
|
|
"type": entity.type.name.lower(),
|
|
"offset": entity.offset,
|
|
"length": entity.length,
|
|
}
|
|
for entity in message.entities
|
|
]
|
|
elif message.caption_entities:
|
|
the_message_dict_to_append["entities"] = [
|
|
{
|
|
"type": entity.type.name.lower(),
|
|
"offset": entity.offset,
|
|
"length": entity.length,
|
|
}
|
|
for entity in message.caption_entities
|
|
]
|
|
else:
|
|
the_message_dict_to_append["entities"] = []
|
|
the_message_dict_to_append["chatId"] = await get_message_sender_id(message)
|
|
the_message_dict_to_append["text"] = await get_text_or_caption(message)
|
|
the_message_dict_to_append["avatar"] = True
|
|
the_message_dict_to_append["from"] = {}
|
|
the_message_dict_to_append["from"]["id"] = await get_message_sender_id(message)
|
|
the_message_dict_to_append["from"]["name"] = await get_message_sender_name(
|
|
message
|
|
)
|
|
the_message_dict_to_append["from"][
|
|
"username"
|
|
] = await get_message_sender_username(message)
|
|
the_message_dict_to_append["from"]["type"] = message.chat.type.name.lower()
|
|
the_message_dict_to_append["from"]["photo"] = await get_message_sender_photo(
|
|
message
|
|
)
|
|
if message.reply_to_message and is_reply:
|
|
the_message_dict_to_append["replyMessage"] = {
|
|
"name": await get_message_sender_name(message.reply_to_message),
|
|
"text": await get_text_or_caption(message.reply_to_message),
|
|
"chatId": await get_message_sender_id(message.reply_to_message),
|
|
}
|
|
else:
|
|
the_message_dict_to_append["replyMessage"] = {}
|
|
payload["messages"].append(the_message_dict_to_append)
|
|
r = await fetch.post("https://bot.lyo.su/quote/generate.png", json=payload)
|
|
if not r.is_error:
|
|
return r.read()
|
|
else:
|
|
raise QuotlyException(r.json())
|
|
|
|
|
|
def isArgInt(txt) -> list:
|
|
count = txt
|
|
try:
|
|
count = int(count)
|
|
return [True, count]
|
|
except ValueError:
|
|
return [False, 0]
|
|
|
|
|
|
@app.on_message(filters.command(["q", "qr"]) & filters.reply)
|
|
async def msg_quotly_cmd(self: Client, ctx: Message):
|
|
is_reply = bool(ctx.command[0].endswith("r"))
|
|
if len(ctx.text.split()) > 1:
|
|
check_arg = isArgInt(ctx.command[1])
|
|
if check_arg[0]:
|
|
if check_arg[1] < 2 or check_arg[1] > 10:
|
|
return await ctx.reply_msg("Invalid range", del_in=6)
|
|
try:
|
|
messages = [
|
|
i
|
|
for i in await self.get_messages(
|
|
chat_id=ctx.chat.id,
|
|
message_ids=range(
|
|
ctx.reply_to_message.id,
|
|
ctx.reply_to_message.id + (check_arg[1] + 5),
|
|
),
|
|
replies=-1,
|
|
)
|
|
if not i.empty and not i.media
|
|
]
|
|
except Exception:
|
|
return await ctx.reply_text("🤷🏻♂️")
|
|
try:
|
|
make_quotly = await pyrogram_to_quotly(messages, is_reply=is_reply)
|
|
bio_sticker = BytesIO(make_quotly)
|
|
bio_sticker.name = "misskatyquote_sticker.webp"
|
|
return await ctx.reply_sticker(bio_sticker)
|
|
except Exception:
|
|
return await ctx.reply_msg("🤷🏻♂️")
|
|
try:
|
|
messages_one = await self.get_messages(
|
|
chat_id=ctx.chat.id, message_ids=ctx.reply_to_message.id, replies=-1
|
|
)
|
|
messages = [messages_one]
|
|
except Exception:
|
|
return await ctx.reply_msg("🤷🏻♂️")
|
|
try:
|
|
make_quotly = await pyrogram_to_quotly(messages, is_reply=is_reply)
|
|
bio_sticker = BytesIO(make_quotly)
|
|
bio_sticker.name = "misskatyquote_sticker.webp"
|
|
return await ctx.reply_sticker(bio_sticker)
|
|
except Exception as e:
|
|
return await ctx.reply_msg(f"ERROR: {e}")
|