|
|
|
|
|
import os |
|
|
import time |
|
|
import base64 |
|
|
import requests |
|
|
from pyrogram import * |
|
|
from pyrogram.types import * |
|
|
from akn.utils.handler import * |
|
|
from akn.utils.logger import LOGS |
|
|
from akn.utils.prefixprem import command |
|
|
from config import * |
|
|
|
|
|
from typing import List, Dict, Tuple |
|
|
|
|
|
def generate_quote(messages: List[Dict]) -> Tuple[bool, str]: |
|
|
json = { |
|
|
"type": "quote", |
|
|
"format": "webp", |
|
|
"backgroundColor": "#260746/#6100c2", |
|
|
"width": 512, |
|
|
"height": 768, |
|
|
"scale": 2, |
|
|
"messages": messages, |
|
|
} |
|
|
try: |
|
|
response = requests.post("https://bot.lyo.su/quote/generate", json=json).json() |
|
|
image = base64.b64decode(str(response["result"]["image"]).encode("utf-8")) |
|
|
file_name = f"Quote_{int(time.time())}.webp" |
|
|
with open(file_name, "wb") as f: |
|
|
f.write(image) |
|
|
return True, file_name |
|
|
except Exception as e: |
|
|
return False, str(e) |
|
|
|
|
|
def get_entities(message: Message) -> List[Dict]: |
|
|
entities = [] |
|
|
if message.entities: |
|
|
for entity in message.entities: |
|
|
entities.append( |
|
|
{ |
|
|
"type": entity.type.name.lower(), |
|
|
"offset": entity.offset, |
|
|
"length": entity.length, |
|
|
} |
|
|
) |
|
|
return entities |
|
|
|
|
|
@Akeno( |
|
|
~filters.scheduled |
|
|
& command(["q"]) |
|
|
& filters.me |
|
|
& ~filters.forwarded |
|
|
) |
|
|
async def quotely(client: Client, message: Message): |
|
|
if not message.reply_to_message: |
|
|
return await message.reply_text("Reply to a message to quote it.") |
|
|
if message.reply_to_message.media: |
|
|
if message.reply_to_message.caption: |
|
|
message.reply_to_message.text = message.reply_to_message.caption |
|
|
else: |
|
|
return await message.reply_text("Reply to a text message to quote it.") |
|
|
cmd = None |
|
|
if len(message.command) > 1: |
|
|
cmd = message.command[1].lower() |
|
|
pro = await message.reply_text("__Generating quote...__") |
|
|
msg_data = [] |
|
|
if cmd and cmd == "r": |
|
|
await pro.edit_text("__Generating quote with reply...__") |
|
|
reply_msg_id = message.reply_to_message.reply_to_message_id |
|
|
if reply_msg_id: |
|
|
reply_msg = await client.get_messages(message.chat.id, reply_msg_id) |
|
|
if reply_msg and reply_msg.text: |
|
|
replied_name = reply_msg.from_user.first_name |
|
|
if reply_msg.from_user.last_name: |
|
|
replied_name += f" {reply_msg.from_user.last_name}" |
|
|
reply_message = { |
|
|
"chatId": reply_msg.from_user.id, |
|
|
"entities": get_entities(reply_msg), |
|
|
"name": replied_name, |
|
|
"text": reply_msg.text, |
|
|
} |
|
|
else: |
|
|
reply_message = {} |
|
|
else: |
|
|
reply_message = {} |
|
|
else: |
|
|
reply_message = {} |
|
|
name = message.reply_to_message.from_user.first_name |
|
|
if message.reply_to_message.from_user.last_name: |
|
|
name += f" {message.reply_to_message.from_user.last_name}" |
|
|
emoji_status = None |
|
|
if message.reply_to_message.from_user.emoji_status: |
|
|
emoji_status = str(message.reply_to_message.from_user.emoji_status.custom_emoji_id) |
|
|
msg_data.append( |
|
|
{ |
|
|
"entities": get_entities(message.reply_to_message), |
|
|
"avatar": True, |
|
|
"from": { |
|
|
"id": message.reply_to_message.from_user.id, |
|
|
"name": name, |
|
|
"emoji_status": emoji_status, |
|
|
}, |
|
|
"text": message.reply_to_message.text, |
|
|
"replyMessage": reply_message, |
|
|
} |
|
|
) |
|
|
status, path = generate_quote(msg_data) |
|
|
if not status: |
|
|
return await message.reply_text(f"`{path}`") |
|
|
await message.reply_sticker(path) |
|
|
await pro.delete() |
|
|
os.remove(path) |
|
|
|
|
|
module = modules_help.add_module("quote", __file__) |
|
|
module.add_command("q", "Generate a quote sticker of the replied message.") |
|
|
module.add_command("q r", "Generate a quote sticker of the replied message with it's reply message.") |
