diff --git a/program/music.py b/program/music.py index 9dc59fa..a71fd7c 100644 --- a/program/music.py +++ b/program/music.py @@ -2,16 +2,16 @@ # Commit Start Date 20/10/2021 # Finished On 28/10/2021 -import re import asyncio +import re -from pyrogram import Client -from driver.veez import call_py, user, bot -from driver.queues import QUEUE, add_to_queue +from config import ASSISTANT_NAME, BOT_USERNAME, IMG_1, IMG_2 from driver.filters import command, other_filters +from driver.queues import QUEUE, add_to_queue +from driver.veez import call_py, user +from pyrogram import Client from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message -from config import BOT_USERNAME, GROUP_SUPPORT, IMG_1, IMG_2, UPDATES_CHANNEL, ASSISTANT_NAME from pytgcalls import StreamType from pytgcalls.types.input_stream import AudioPiped from youtubesearchpython import VideosSearch @@ -51,78 +51,70 @@ async def ytdl(link): @Client.on_message(command(["play", f"play@{BOT_USERNAME}"]) & other_filters) -async def play(_, m: Message): - +async def play(c: Client, m: Message): + replied = m.reply_to_message + chat_id = m.chat.id keyboard = InlineKeyboardMarkup( [ [ - InlineKeyboardButton( - text="• Mᴇɴᴜ", callback_data="cbmenu" - ), - InlineKeyboardButton( - text="• Cʟᴏsᴇ", callback_data="cls" - ), + InlineKeyboardButton(text="• Mᴇɴᴜ", callback_data="cbmenu"), + InlineKeyboardButton(text="• Cʟᴏsᴇ", callback_data="cls"), ] ] ) - try: - popo = await bot.get_me() - papa = popo - pepe = papa.id + aing = await c.get_me() except Exception as e: return await m.reply_text(f"error:\n\n{e}") - chat_title = m.chat.title - a = await bot.get_chat_member(m.chat.id, pepe) + a = await c.get_chat_member(chat_id, aing.id) if a.status != "administrator": - await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**") + await m.reply_text( + f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**" + ) return if not a.can_manage_voice_chats: await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Manage voice chat__") + "missing required permission:" + "\n\n» ❌ __Manage voice chat__" + ) return if not a.can_delete_messages: await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Delete messages__") + "missing required permission:" + "\n\n» ❌ __Delete messages__" + ) return if not a.can_invite_users: - await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Add users__") + await m.reply_text("missing required permission:" + "\n\n» ❌ __Add users__") return if not a.can_restrict_members: - await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Ban users__") + await m.reply_text("missing required permission:" + "\n\n» ❌ __Ban users__") return try: - uber = await user.get_me() - grab = uber - good = grab.id - b = await bot.get_chat_member(m.chat.id, good) + ubot = await user.get_me() + b = await c.get_chat_member(chat_id, ubot.id) if b.status == "kicked": - await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**") + await m.reply_text( + f"@{ASSISTANT_NAME} **is banned in group** {m.chat.title}\n\n» **unban the userbot first if you want to use this bot.**" + ) return except UserNotParticipant: if m.chat.username: try: - await user.join_chat(f"{m.chat.username}") + await user.join_chat(m.chat.username) except Exception as e: await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") return - else: - try: - pope = await bot.export_chat_invite_link(m.chat.id) - pepo = await bot.revoke_chat_invite_link(m.chat.id, pope) - await user.join_chat(pepo.invite_link) - except UserAlreadyParticipant: - pass - except Exception as e: - return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") - replied = m.reply_to_message - chat_id = m.chat.id + else: + try: + pope = await c.export_chat_invite_link(chat_id) + pepo = await c.revoke_chat_invite_link(chat_id, pope) + await user.join_chat(pepo.invite_link) + except UserAlreadyParticipant: + pass + except Exception as e: + return await m.reply_text( + f"❌ **userbot failed to join**\n\n**reason**:{e}" + ) + if replied: if replied.audio or replied.voice: suhu = await replied.reply("📥 **downloading audio...**") @@ -232,7 +224,9 @@ async def play(_, m: Message): if chat_id in QUEUE: pos = add_to_queue(chat_id, songname, ytlink, url, "Audio", 0) await suhu.delete() - requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})" + requester = ( + f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})" + ) await m.reply_photo( photo=f"{IMG_1}", caption=f"💡 **Track added to the queue**\n\n🏷 **Name:** [{songname}]({url})\n💭 **Chat:** `{chat_id}`\n🎧 **Request by:** {requester}\n🔢 **At position »** `{pos}`", @@ -261,78 +255,71 @@ async def play(_, m: Message): # stream is used for live streaming only -@Client.on_message(command(["stream", f"stream@{BOT_USERNAME}"]) & other_filters) -async def stream(_, m: Message): +@Client.on_message(command(["stream", f"stream@{BOT_USERNAME}"]) & other_filters) +async def stream(c: Client, m: Message): + chat_id = m.chat.id keyboard = InlineKeyboardMarkup( [ [ - InlineKeyboardButton( - text="• Mᴇɴᴜ", callback_data="cbmenu" - ), - InlineKeyboardButton( - text="• Cʟᴏsᴇ", callback_data="cls" - ), + InlineKeyboardButton(text="• Mᴇɴᴜ", callback_data="cbmenu"), + InlineKeyboardButton(text="• Cʟᴏsᴇ", callback_data="cls"), ] ] ) - try: - popo = await bot.get_me() - papa = popo - pepe = papa.id + aing = await c.get_me() except Exception as e: return await m.reply_text(f"error:\n\n{e}") - chat_title = m.chat.title - a = await bot.get_chat_member(m.chat.id, pepe) + a = await c.get_chat_member(chat_id, aing.id) if a.status != "administrator": - await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**") + await m.reply_text( + f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**" + ) return if not a.can_manage_voice_chats: await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Manage voice chat__") + "missing required permission:" + "\n\n» ❌ __Manage voice chat__" + ) return if not a.can_delete_messages: await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Delete messages__") + "missing required permission:" + "\n\n» ❌ __Delete messages__" + ) return if not a.can_invite_users: - await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Add users__") + await m.reply_text("missing required permission:" + "\n\n» ❌ __Add users__") return if not a.can_restrict_members: - await m.reply_text( - "missing required permission:" - + "\n\n» ❌ __Ban users__") + await m.reply_text("missing required permission:" + "\n\n» ❌ __Ban users__") return try: - uber = await user.get_me() - grab = uber - good = grab.id - b = await bot.get_chat_member(m.chat.id, good) + ubot = await user.get_me() + b = await c.get_chat_member(chat_id, ubot.id) if b.status == "kicked": - await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**") + await m.reply_text( + f"@{ASSISTANT_NAME} **is banned in group** {m.chat.title}\n\n» **unban the userbot first if you want to use this bot.**" + ) return except UserNotParticipant: if m.chat.username: try: - await user.join_chat(f"{m.chat.username}") + await user.join_chat(m.chat.username) except Exception as e: await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") return - else: - try: - pope = await bot.export_chat_invite_link(m.chat.id) - pepo = await bot.revoke_chat_invite_link(m.chat.id, pope) - await user.join_chat(pepo.invite_link) - except UserAlreadyParticipant: - pass - except Exception as e: - return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") - chat_id = m.chat.id + else: + try: + pope = await c.export_chat_invite_link(chat_id) + pepo = await c.revoke_chat_invite_link(chat_id, pope) + await user.join_chat(pepo.invite_link) + except UserAlreadyParticipant: + pass + except Exception as e: + return await m.reply_text( + f"❌ **userbot failed to join**\n\n**reason**:{e}" + ) + if len(m.command) < 2: await m.reply("» give me a live-link/m3u8 url/youtube link to stream.") else: @@ -370,7 +357,9 @@ async def stream(_, m: Message): ) add_to_queue(chat_id, "Radio", livelink, link, "Audio", 0) await suhu.delete() - requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})" + requester = ( + f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})" + ) await m.reply_photo( photo=f"{IMG_2}", caption=f"💡 **[Radio live]({link}) stream started.**\n\n💭 **Chat:** `{chat_id}`\n💡 **Status:** `Playing`\n🎧 **Request by:** {requester}",