4e65f218f5
Co-authored-by: Aditya <me@xditya.me> Co-authored-by: Danish <danish@ultroid.tech> Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-authored-by: sppidy <sppidy@users.noreply.github.com> Co-authored-by: Arnab Paryali <Arnabxd@users.noreply.github.com> Co-authored-by: divkix <divkix@users.noreply.github.com> Co-authored-by: hellboi_atul <hellboi-atul@users.noreply.github.com> Co-authored-by: Programming Error <error@notavailable.live> Co-authored-by: New-dev0 <New-dev0@users.noreply.github.com>
403 lines
13 KiB
Python
403 lines
13 KiB
Python
# Ultroid - UserBot
|
||
# Copyright (C) 2021 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}circle`
|
||
Reply to a audio song or gif to get video note.
|
||
|
||
• `{i}ls`
|
||
Get all the Files inside a Directory.
|
||
|
||
• `{i}bots`
|
||
Shows the number of bots in the current chat with their perma-link.
|
||
|
||
• `{i}hl <a link>`
|
||
Embeds the link with a whitespace as message.
|
||
|
||
• `{i}id`
|
||
Reply a Sticker to Get Its Id
|
||
Reply a User to Get His Id
|
||
Without Replying You Will Get the Chat's Id
|
||
|
||
• `{i}sg <reply to a user><username/id>`
|
||
Get His Name History of the replied user.
|
||
|
||
• `{i}tr <dest lang code> <(reply to) a message>`
|
||
Get translated message.
|
||
|
||
"""
|
||
|
||
import os
|
||
import time
|
||
from asyncio.exceptions import TimeoutError
|
||
from pathlib import Path
|
||
|
||
import cv2
|
||
from googletrans import Translator
|
||
from hachoir.metadata import extractMetadata
|
||
from hachoir.parser import createParser
|
||
from telethon.errors.rpcerrorlist import YouBlockedUserError
|
||
from telethon.tl.types import ChannelParticipantAdmin, ChannelParticipantsBots
|
||
from telethon.tl.types import DocumentAttributeVideo as video
|
||
from telethon.utils import pack_bot_file_id
|
||
|
||
from . import *
|
||
from . import humanbytes as hb
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="tr",
|
||
)
|
||
async def _(event):
|
||
if len(event.text) > 3:
|
||
if not event.text[3] == " ":
|
||
return
|
||
input = event.text[4:6]
|
||
txt = event.text[7:]
|
||
xx = await eor(event, "`Translating...`")
|
||
if event.reply_to_msg_id:
|
||
previous_message = await event.get_reply_message()
|
||
text = previous_message.message
|
||
lan = input or "en"
|
||
elif input:
|
||
text = txt
|
||
lan = input or "en"
|
||
else:
|
||
return await eod(xx, f"`{hndlr}tr LanguageCode` as reply to a message", time=5)
|
||
translator = Translator()
|
||
try:
|
||
tt = translator.translate(text, dest=lan)
|
||
output_str = f"**TRANSLATED** from {tt.src} to {lan}\n{tt.text}"
|
||
await eor(xx, output_str)
|
||
except Exception as exc:
|
||
await eod(xx, str(exc), time=10)
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="id ?(.*)",
|
||
)
|
||
async def _(event):
|
||
if event.reply_to_msg_id:
|
||
await event.get_input_chat()
|
||
r_msg = await event.get_reply_message()
|
||
if r_msg.media:
|
||
bot_api_file_id = pack_bot_file_id(r_msg.media)
|
||
await eor(
|
||
event,
|
||
"**Current Chat ID:** `{}`\n**From User ID:** `{}`\n**Bot API File ID:** `{}`".format(
|
||
str(event.chat_id),
|
||
str(r_msg.sender_id),
|
||
bot_api_file_id,
|
||
),
|
||
)
|
||
else:
|
||
await eor(
|
||
event,
|
||
"**Chat ID:** `{}`\n**User ID:** `{}`".format(
|
||
str(event.chat_id),
|
||
str(r_msg.sender_id),
|
||
),
|
||
)
|
||
elif event.pattern_match.group(1):
|
||
ids = await get_user_id(event.pattern_match.group(1))
|
||
return await eor(
|
||
event,
|
||
"**Chat ID:** `{}`\n**User ID:** `{}`".format(
|
||
str(event.chat_id),
|
||
str(ids),
|
||
),
|
||
)
|
||
else:
|
||
await eor(event, "**Current Chat ID:** `{}`".format(str(event.chat_id)))
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="bots ?(.*)",
|
||
)
|
||
async def _(ult):
|
||
await ult.edit("`...`")
|
||
if ult.is_private:
|
||
user = await ult.get_chat()
|
||
if not user.bot:
|
||
return await ult.edit("`Seariously ?`")
|
||
|
||
mentions = "**Bots in this Chat**: \n"
|
||
input_str = ult.pattern_match.group(1)
|
||
to_write_chat = await ult.get_input_chat()
|
||
chat = None
|
||
if not input_str:
|
||
chat = to_write_chat
|
||
else:
|
||
mentions = f"**Bots in **{input_str}: \n"
|
||
try:
|
||
chat = await ultroid_bot.get_entity(input_str)
|
||
except Exception as e:
|
||
await eor(ult, str(e))
|
||
return None
|
||
try:
|
||
async for x in ultroid_bot.iter_participants(
|
||
chat,
|
||
filter=ChannelParticipantsBots,
|
||
):
|
||
if isinstance(x.participant, ChannelParticipantAdmin):
|
||
mentions += "\n ⚜️ [{}](tg://user?id={}) `{}`".format(
|
||
x.first_name,
|
||
x.id,
|
||
x.id,
|
||
)
|
||
else:
|
||
mentions += "\n [{}](tg://user?id={}) `{}`".format(
|
||
x.first_name,
|
||
x.id,
|
||
x.id,
|
||
)
|
||
except Exception as e:
|
||
mentions += " " + str(e) + "\n"
|
||
await eor(ult, mentions)
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="hl",
|
||
)
|
||
async def _(ult):
|
||
try:
|
||
input = ult.text.split(" ", maxsplit=1)[1]
|
||
except IndexError:
|
||
return await eod(ult, "`Input some link`", time=5)
|
||
await eor(ult, "[ㅤㅤㅤㅤㅤㅤㅤ](" + input + ")", link_preview=False)
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="circle$",
|
||
)
|
||
async def _(e):
|
||
a = await e.get_reply_message()
|
||
if a is None:
|
||
return await eor(e, "Reply to a gif or audio")
|
||
if a.document and a.document.mime_type == "audio/mpeg":
|
||
z = await eor(e, "**Cʀᴇᴀᴛɪɴɢ Vɪᴅᴇᴏ Nᴏᴛᴇ**")
|
||
toime = time.time()
|
||
try:
|
||
bbbb = await a.download_media(thumb=-1)
|
||
im = cv2.imread(bbbb)
|
||
dsize = (320, 320)
|
||
output = cv2.resize(im, dsize, interpolation=cv2.INTER_AREA)
|
||
cv2.imwrite("img.png", output)
|
||
thumb = "img.png"
|
||
os.remove(bbbb)
|
||
except TypeError:
|
||
bbbb = "resources/extras/ultroid.jpg"
|
||
im = cv2.imread(bbbb)
|
||
dsize = (320, 320)
|
||
output = cv2.resize(im, dsize, interpolation=cv2.INTER_AREA)
|
||
cv2.imwrite("img.png", output)
|
||
thumb = "img.png"
|
||
c = await downloader(
|
||
"resources/downloads/" + a.file.name,
|
||
a.media.document,
|
||
z,
|
||
toime,
|
||
"Dᴏᴡɴʟᴏᴀᴅɪɴɢ...",
|
||
)
|
||
await z.edit("**Dᴏᴡɴʟᴏᴀᴅᴇᴅ...\nNᴏᴡ Cᴏɴᴠᴇʀᴛɪɴɢ...**")
|
||
await bash(
|
||
f'ffmpeg -i "{c.name}" -preset ultrafast -acodec libmp3lame -ac 2 -ab 144 -ar 44100 comp.mp3'
|
||
)
|
||
await bash(
|
||
f'ffmpeg -y -i "{thumb}" -i comp.mp3 -preset ultrafast -c:a copy circle.mp4'
|
||
)
|
||
taime = time.time()
|
||
foile = await uploader("circle.mp4", "circle.mp4", taime, z, "Uᴘʟᴏᴀᴅɪɴɢ...")
|
||
metadata = extractMetadata(createParser("circle.mp4"))
|
||
duration = metadata.get("duration").seconds
|
||
attributes = [video(duration=duration, w=320, h=320, round_message=True)]
|
||
await e.client.send_file(
|
||
e.chat_id,
|
||
foile,
|
||
thumb=thumb,
|
||
reply_to=a,
|
||
attributes=attributes,
|
||
)
|
||
await z.delete()
|
||
os.system("rm resources/downloads/*")
|
||
os.system("rm circle.mp4 comp.mp3 img.png")
|
||
elif a.document and a.document.mime_type == "video/mp4":
|
||
z = await eor(e, "**Cʀᴇᴀᴛɪɴɢ Vɪᴅᴇᴏ Nᴏᴛᴇ**")
|
||
c = await a.download_media("resources/downloads/")
|
||
await e.client.send_file(
|
||
e.chat_id,
|
||
c,
|
||
video_note=True,
|
||
thumb="resources/extras/ultroid.jpg",
|
||
reply_to=a,
|
||
)
|
||
await z.delete()
|
||
os.remove(c)
|
||
else:
|
||
return await eor(e, "**Reply to a gif or audio file only**")
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="ls ?(.*)",
|
||
)
|
||
async def _(e):
|
||
path = Path(e.pattern_match.group(1))
|
||
if not path:
|
||
path = Path(".")
|
||
else:
|
||
if not os.path.isdir(path):
|
||
return await eod(e, "`Incorrect Directory.`")
|
||
if not os.listdir(path):
|
||
return await eod(e, "`This Directory is Empty.`")
|
||
files = path.iterdir()
|
||
pyfiles = []
|
||
jsons = []
|
||
vdos = []
|
||
audios = []
|
||
pics = []
|
||
others = []
|
||
otherfiles = []
|
||
folders = []
|
||
text = []
|
||
apk = []
|
||
exe = []
|
||
zip_ = []
|
||
book = []
|
||
for file in sorted(files):
|
||
if os.path.isdir(file):
|
||
folders.append("📂 " + str(file))
|
||
elif str(file).endswith(".py"):
|
||
pyfiles.append("🐍 " + str(file))
|
||
elif str(file).endswith(".json"):
|
||
jsons.append("🔮 " + str(file))
|
||
elif str(file).endswith((".mkv", ".mp4", ".avi", ".gif")):
|
||
vdos.append("🎥 " + str(file))
|
||
elif str(file).endswith((".mp3", ".ogg", ".m4a")):
|
||
audios.append("🔊 " + str(file))
|
||
elif str(file).endswith((".jpg", ".jpeg", ".png", ".webp")):
|
||
pics.append("🖼 " + str(file))
|
||
elif str(file).endswith((".txt", ".text", ".log")):
|
||
text.append("📄 " + str(file))
|
||
elif str(file).endswith((".apk", ".xapk")):
|
||
apk.append("📲 " + str(file))
|
||
elif str(file).endswith(".exe"):
|
||
exe.append("⚙ " + str(file))
|
||
elif str(file).endswith((".zip", ".rar")):
|
||
zip_.append("🗜 " + str(file))
|
||
elif str(file).endswith((".pdf", ".epub")):
|
||
book.append("📗 " + str(file))
|
||
elif "." in str(file)[1:]:
|
||
others.append("🏷 " + str(file))
|
||
else:
|
||
otherfiles.append("📒 " + str(file))
|
||
omk = [
|
||
*sorted(folders),
|
||
*sorted(pyfiles),
|
||
*sorted(jsons),
|
||
*sorted(zip_),
|
||
*sorted(vdos),
|
||
*sorted(pics),
|
||
*sorted(audios),
|
||
*sorted(apk),
|
||
*sorted(exe),
|
||
*sorted(book),
|
||
*sorted(text),
|
||
*sorted(others),
|
||
*sorted(otherfiles),
|
||
]
|
||
text = ""
|
||
fls, fos = 0, 0
|
||
flc, foc = 0, 0
|
||
for i in omk:
|
||
emoji = i.split()[0]
|
||
name = i.split(maxsplit=1)[1]
|
||
nam = name.split("/")[-1]
|
||
if os.path.isdir(name):
|
||
size = 0
|
||
for path, dirs, files in os.walk(name):
|
||
for f in files:
|
||
fp = os.path.join(path, f)
|
||
size += os.path.getsize(fp)
|
||
if hb(size):
|
||
text += emoji + f" `{nam}`" + " `" + hb(size) + "`\n"
|
||
fos += size
|
||
else:
|
||
text += emoji + f" `{nam}`" + "\n"
|
||
foc += 1
|
||
else:
|
||
if hb(int(os.path.getsize(name))):
|
||
text += (
|
||
emoji + f" `{nam}`" + " `" + hb(int(os.path.getsize(name))) + "`\n"
|
||
)
|
||
fls += int(os.path.getsize(name))
|
||
else:
|
||
text += emoji + f" `{nam}`" + "\n"
|
||
flc += 1
|
||
tfos, tfls, ttol = hb(fos), hb(fls), hb(fos + fls)
|
||
if not hb(fos):
|
||
tfos = "0 B"
|
||
if not hb(fls):
|
||
tfls = "0 B"
|
||
if not hb(fos + fls):
|
||
ttol = "0 B"
|
||
text += f"\n\n`Folders` : `{foc}` : `{tfos}`\n`Files` : `{flc}` : `{tfls}`\n`Total` : `{flc+foc}` : `{ttol}`"
|
||
await eor(e, text)
|
||
|
||
|
||
@ultroid_cmd(
|
||
pattern="sg ?(.*)",
|
||
)
|
||
async def lastname(steal):
|
||
mat = steal.pattern_match.group(1)
|
||
if not (steal.is_reply or mat):
|
||
await eor(steal, "`Use this command with reply or give Username/id...`")
|
||
return
|
||
if mat:
|
||
user_id = await get_user_id(mat)
|
||
message = await steal.get_reply_message()
|
||
if message:
|
||
user_id = message.sender.id
|
||
chat = "@SangMataInfo_bot"
|
||
id = f"/search_id {user_id}"
|
||
lol = await eor(steal, "`Processing !...`")
|
||
try:
|
||
async with ultroid_bot.conversation(chat) as conv:
|
||
try:
|
||
msg = await conv.send_message(id)
|
||
response = await conv.get_response()
|
||
respond = await conv.get_response()
|
||
responds = await conv.get_response()
|
||
except YouBlockedUserError:
|
||
await lol.edit("Please unblock @sangmatainfo_bot and try again")
|
||
return
|
||
if (
|
||
response.text.startswith("No records found")
|
||
or respond.text.startswith("No records found")
|
||
or responds.text.startswith("No records found")
|
||
):
|
||
await lol.edit("No records found for this user")
|
||
await steal.client.delete_messages(conv.chat_id, [msg.id, response.id])
|
||
return
|
||
else:
|
||
if response.text.startswith("🔗"):
|
||
await lol.edit(respond.message)
|
||
await lol.reply(responds.message)
|
||
elif respond.text.startswith("🔗"):
|
||
await lol.edit(response.message)
|
||
await lol.reply(responds.message)
|
||
else:
|
||
await lol.edit(respond.message)
|
||
await lol.reply(response.message)
|
||
await steal.client.delete_messages(
|
||
conv.chat_id,
|
||
[msg.id, responds.id, respond.id, response.id],
|
||
)
|
||
except TimeoutError:
|
||
return await lol.edit("Error: @SangMataInfo_bot is not responding!.")
|