8ad7073d80
Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-authored-by: Danish <danish@ultroid.tech> Co-authored-by: Mahesh Chauhan <84653476+vasusen-code@users.noreply.github.com> Co-authored-by: Aditya <xditya@ultroid.tech> Co-authored-by: CyrusXD <79554993+Ashutosh1478@users.noreply.github.com>
119 lines
3.6 KiB
Python
119 lines
3.6 KiB
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2021-2022 TeamUltroid
|
|
#
|
|
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
|
|
# PLease read the GNU Affero General Public License in
|
|
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
|
|
"""
|
|
✘ Commands Available -
|
|
|
|
• `{i}mediainfo <reply to media>`
|
|
To get info about it.
|
|
|
|
• `{i}rotate <degree/angle> <reply to media>`
|
|
Rotate any video/photo/media..
|
|
Note : for video it should be angle of 90's
|
|
"""
|
|
import os
|
|
import time
|
|
from datetime import datetime as dt
|
|
|
|
from pyUltroid.functions.misc import rotate_image
|
|
|
|
from . import (
|
|
LOGS,
|
|
bash,
|
|
downloader,
|
|
get_string,
|
|
make_html_telegraph,
|
|
mediainfo,
|
|
ultroid_cmd,
|
|
)
|
|
|
|
try:
|
|
import cv2
|
|
except ImportError:
|
|
LOGS.info("WARNING: 'cv2' not found!")
|
|
cv2 = None
|
|
|
|
|
|
@ultroid_cmd(pattern="mediainfo$")
|
|
async def mi(e):
|
|
r = await e.get_reply_message()
|
|
if not (r and r.media):
|
|
return await e.eor(get_string("cvt_3"), time=5)
|
|
xx = mediainfo(r.media)
|
|
murl = r.media.stringify()
|
|
url = await make_html_telegraph("Mediainfo", "Ultroid", f"<code>{murl}</code>")
|
|
ee = await e.eor(f"**[{xx}]({url})**\n\n`Loading More...`", link_preview=False)
|
|
taime = time.time()
|
|
if hasattr(r.media, "document"):
|
|
file = r.media.document
|
|
mime_type = file.mime_type
|
|
filename = r.file.name
|
|
if not filename:
|
|
if "audio" in mime_type:
|
|
filename = "audio_" + dt.now().isoformat("_", "seconds") + ".ogg"
|
|
elif "video" in mime_type:
|
|
filename = "video_" + dt.now().isoformat("_", "seconds") + ".mp4"
|
|
dl = await downloader(
|
|
"resources/downloads/" + filename,
|
|
file,
|
|
ee,
|
|
taime,
|
|
f"`**[{xx}]({url})**\n\n`Loading More...",
|
|
)
|
|
naam = dl.name
|
|
else:
|
|
naam = await r.download_media()
|
|
out, er = await bash(f"mediainfo '{naam}' --Output=HTML")
|
|
if er:
|
|
LOGS.info(er)
|
|
return await ee.edit(f"**[{xx}]({url})**", link_preview=False)
|
|
try:
|
|
urll = await make_html_telegraph("Mediainfo", "Ultroid", out)
|
|
except Exception as er:
|
|
LOGS.exception(er)
|
|
return
|
|
await ee.edit(
|
|
f"**[{xx}]({url})**\n\n[{get_string('mdi_1')}]({urll})", link_preview=False
|
|
)
|
|
os.remove(naam)
|
|
|
|
|
|
@ultroid_cmd(pattern="rotate( (.*)|$)")
|
|
async def rotate_(ult):
|
|
match = ult.pattern_match.group(1).strip()
|
|
if not ult.is_reply:
|
|
return await ult.eor("`Reply to a media...`")
|
|
if match:
|
|
try:
|
|
match = int(match)
|
|
except ValueError:
|
|
match = None
|
|
if not match:
|
|
return await ult.eor("`Please provide a valid angle to rotate media..`")
|
|
reply = await ult.get_reply_message()
|
|
msg = await ult.eor(get_string("com_1"))
|
|
photo = reply.game.photo if reply.game else None
|
|
if photo or reply.photo or reply.sticker:
|
|
media = await ult.client.download_media(photo or reply)
|
|
img = cv2.imread(media)
|
|
new_ = rotate_image(img, match)
|
|
file = "ult.png"
|
|
cv2.imwrite(file, new_)
|
|
elif reply.video:
|
|
media = await reply.download_media()
|
|
file = media + ".mp4"
|
|
await bash(
|
|
f'ffmpeg -i "{media}" -c copy -metadata:s:v:0 rotate={match} "{file}" -y'
|
|
)
|
|
else:
|
|
return await msg.edit("`Unsupported Media..\nReply to Photo/Video`")
|
|
if os.path.exists(file):
|
|
await ult.client.send_file(
|
|
ult.chat_id, file=file, video_note=bool(reply.video_note), reply_to=reply.id
|
|
)
|
|
os.remove(media)
|
|
await msg.try_delete()
|