Update Akeno/plugins/youtube.py

Akeno/plugins/youtube.py

@@ -3,6 +3,7 @@ import time
 
 import requests
 from pyrogram.types import Message
+from pyrogram import Client, filters
 from youtube_search import YoutubeSearch
 from yt_dlp import YoutubeDL
 
@@ -23,7 +24,7 @@ custom_loading = "<emoji id=5974235702701853774>🗿</emoji>"
     & filters.me
     & ~filters.forwarded
 )
-async def youtube_search_audio(_, message: Message):
+async def youtube_search_audio(client: Client, message: Message):
     if len(message.command) < 2:
         return await message.reply_text(
             "Give a valid youtube search to download audio."
@@ -37,7 +38,7 @@ async def youtube_search_audio(_, message: Message):
     status, url = YoutubeDriver.check_url(okk)
     if not status:
         return await pro.edit_text(url)
-    if client.me_is_premium:
+    if client.me.is_premium:
         await pro.edit_text(f"{custom_loading}__Downloading audio ...__")
     else:
         await pro.edit_text(f"__Downloading audio ...__")
@@ -46,7 +47,7 @@ async def youtube_search_audio(_, message: Message):
             yt_data = ytdl.extract_info(url, False)
             yt_file = ytdl.prepare_filename(yt_data)
             ytdl.process_info(yt_data)
-        if client.me_is_premium:
+        if client.me.is_premium:
             upload_text = f"**{custom_loading}𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
         else:
             upload_text = f"**𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
@@ -83,7 +84,7 @@ async def youtube_search_audio(_, message: Message):
     & filters.me
     & ~filters.forwarded
 )
-async def youtube_audio(_, message: Message):
+async def youtube_audio(client: Client, message: Message):
     if len(message.command) < 2:
         return await message.reply_text(
             "Give a valid youtube link to download audio."
@@ -93,7 +94,7 @@ async def youtube_audio(_, message: Message):
     status, url = YoutubeDriver.check_url(query)
     if not status:
         return await pro.edit_text(url)
-    if client.me_is_premium:
+    if client.me.is_premium:
         await pro.edit_text(f"{custom_loading}__Downloading audio ...__")
     else:
         await pro.edit_text(f"__Downloading audio ...__")
@@ -102,7 +103,7 @@ async def youtube_audio(_, message: Message):
             yt_data = ytdl.extract_info(url, False)
             yt_file = ytdl.prepare_filename(yt_data)
             ytdl.process_info(yt_data)
-        if client.me_is_premium:
+        if client.me.is_premium:
             upload_text = f"**{custom_loading}𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
         else:
             upload_text = f"**𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
@@ -153,7 +154,7 @@ async def ytvideo_search(client: Client, message: Message):
     status, url = YoutubeDriver.check_url(okk)
     if not status:
         return await pro.edit_text(url)
-    if client.me_is_premium:
+    if client.me.is_premium:
         await pro.edit_text(f"{custom_loading}__Downloading audio ...__")
     else:
         await pro.edit_text(f"__Downloading audio ...__")
@@ -162,7 +163,7 @@ async def ytvideo_search(client: Client, message: Message):
             yt_data = ytdl.extract_info(url, True)
             yt_file = yt_data["id"]
 
-        if client.me_is_premium:
+        if client.me.is_premium:
             upload_text = f"**{custom_loading}𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
         else:
             upload_text = f"**𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
@@ -207,7 +208,7 @@ async def ytvideo(client: Client, message: Message):
     status, url = YoutubeDriver.check_url(query)
     if not status:
         return await pro.edit_text(url)
-    if client.me_is_premium:
+    if client.me.is_premium:
         await pro.edit_text(f"{custom_loading}__Downloading audio ...__")
     else:
         await pro.edit_text(f"__Downloading audio ...__")
@@ -216,7 +217,7 @@ async def ytvideo(client: Client, message: Message):
             yt_data = ytdl.extract_info(url, True)
             yt_file = yt_data["id"]
 
-        if client.me_is_premium:
+        if client.me.is_premium:
             upload_text = f"**{custom_loading}𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
         else:
             upload_text = f"**𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
@@ -272,4 +273,4 @@ module.add_command("yta", "Download the youtube link video in .mp3 format!.")
 module.add_command("ytv", "Download the youtube link video in .mp3 format!.")
 module.add_command("ytsa", "Download the youtube search video in .mp3 format!.")
 module.add_command("ytva", "Download the youtube search video in .mp4 format!")
-module.add_command("ytlink", "Search for a video on youtube")
+module.add_command("ytlink", "Search for a video on youtube")