mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2026-01-02 18:44:51 +00:00
111 lines
4.5 KiB
Python
111 lines
4.5 KiB
Python
"""
|
|
* @author yasir <yasiramunandar@gmail.com>
|
|
* @date 2022-12-01 09:12:27
|
|
* @lastModified 2022-12-01 09:32:31
|
|
* @projectName MissKatyPyro
|
|
* Copyright @YasirPedia All rights reserved
|
|
"""
|
|
import datetime
|
|
import os
|
|
import time
|
|
import traceback
|
|
from asyncio import gather, sleep
|
|
from logging import getLogger
|
|
|
|
from pyrogram import enums, filters
|
|
from pyrogram.errors import FloodWait
|
|
from pyrogram.types import InlineKeyboardMarkup
|
|
|
|
from misskaty import BOT_USERNAME, app
|
|
from misskaty.core.decorator.errors import capture_err
|
|
from misskaty.core.decorator.ratelimiter import ratelimiter
|
|
from misskaty.core.message_utils import *
|
|
from misskaty.helper import gen_ik_buttons, get_duration, is_url, progress_for_pyrogram, screenshot_flink, take_ss
|
|
from misskaty.vars import COMMAND_HANDLER
|
|
|
|
LOGGER = getLogger(__name__)
|
|
|
|
__MODULE__ = "MediaTool"
|
|
__HELP__ = """"
|
|
/genss [reply to video] - Generate Screenshot From Video. (Support TG Media and Direct URL)
|
|
/mediainfo [link/reply to TG Video] - Get Mediainfo From File.
|
|
"""
|
|
|
|
|
|
@app.on_message(filters.command(["genss"], COMMAND_HANDLER))
|
|
@capture_err
|
|
@ratelimiter
|
|
async def genss(client, m):
|
|
if not m.from_user:
|
|
return
|
|
replied = m.reply_to_message
|
|
if len(m.command) == 2 and is_url(m.command[1]):
|
|
snt = await kirimPesan(m, "Give me some time to process your request!! 😴", quote=True)
|
|
|
|
duration = await get_duration(m.command[1])
|
|
if isinstance(duration, str):
|
|
return await editPesan(snt, "😟 Sorry! I cannot open the file.")
|
|
btns = gen_ik_buttons()
|
|
await editPesan(snt, f"Now choose how many result for screenshot? 🥳.\n\nTotal duration: `{datetime.timedelta(seconds=duration)}` (`{duration}s`)", reply_markup=InlineKeyboardMarkup(btns))
|
|
elif replied and replied.media:
|
|
vid = [replied.video, replied.document]
|
|
media = next((v for v in vid if v is not None), None)
|
|
if media is None:
|
|
return await kirimPesan(m, "Reply to a Telegram Video or document as video to generate screenshoot!", quote=True)
|
|
process = await kirimPesan(m, "<code>Processing, please wait..</code>", quote=True)
|
|
|
|
c_time = time.time()
|
|
the_real_download_location = await replied.download(
|
|
progress=progress_for_pyrogram,
|
|
progress_args=("Trying to download, please wait..", process, c_time),
|
|
)
|
|
if the_real_download_location is not None:
|
|
try:
|
|
await editPesan(process, f"File video berhasil didownload dengan path <code>{the_real_download_location}</code>.")
|
|
await sleep(2)
|
|
images = await take_ss(the_real_download_location)
|
|
await editPesan(process, "Mencoba mengupload, hasil generate screenshot..")
|
|
await client.send_chat_action(chat_id=m.chat.id, action=enums.ChatAction.UPLOAD_PHOTO)
|
|
|
|
try:
|
|
await gather(
|
|
*[
|
|
m.reply_document(images, reply_to_message_id=m.id),
|
|
m.reply_photo(images, reply_to_message_id=m.id),
|
|
]
|
|
)
|
|
except FloodWait as e:
|
|
await sleep(e.value)
|
|
await gather(
|
|
*[
|
|
m.reply_document(images, reply_to_message_id=m.id),
|
|
m.reply_photo(images, reply_to_message_id=m.id),
|
|
]
|
|
)
|
|
await kirimPesan(
|
|
m,
|
|
f"☑️ Uploaded [1] screenshoot.\n\n{m.from_user.first_name} (<code>{m.from_user.id}</code>)\n#️⃣ #ssgen #id{m.from_user.id}\n\nSS Generate by @{BOT_USERNAME}",
|
|
reply_to_message_id=m.id,
|
|
)
|
|
await process.delete()
|
|
try:
|
|
os.remove(images)
|
|
os.remove(the_real_download_location)
|
|
except:
|
|
pass
|
|
except Exception:
|
|
exc = traceback.format_exc()
|
|
await kirimPesan(m, f"Gagal generate screenshot.\n\n{exc}")
|
|
try:
|
|
os.remove(images)
|
|
os.remove(the_real_download_location)
|
|
except:
|
|
pass
|
|
else:
|
|
await kirimPesan(m, "Reply to a Telegram media to get screenshots from media..")
|
|
|
|
|
|
@app.on_callback_query(filters.regex(r"^scht"))
|
|
@ratelimiter
|
|
async def _(c, m):
|
|
asyncio.create_task(screenshot_flink(c, m))
|