Upload 4 files

Akeno/plugins/__init__.py

@@ -1 +1,18 @@
-from config import *
+import glob
+from os.path import basename, dirname, isfile
+
+from Akeno.utils.logger import LOGS
+
+
+def __list_all_modules():
+    mod_paths = glob.glob(dirname(__file__) + "/*.py")
+    all_modules = [
+        basename(f)[:-3]
+        for f in mod_paths
+        if isfile(f) and f.endswith(".py") and not f.endswith("__init__.py")
+    ]
+    return all_modules
+
+ALL_MODULES = sorted(__list_all_modules())
+LOGS.info("Modules to load: %s", str(ALL_MODULES))
+__all__ = ALL_MODULES + ["ALL_MODULES"]

Akeno/plugins/id.py

@@ -1,12 +1,16 @@
-import os
 import asyncio
+import os
 import random
 import time
+
 from pyrogram import Client, filters
+from pyrogram.errors import *
 from pyrogram.types import Message
+
 from Akeno.utils.handler import *
 from config import *
 
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["id"], CMD_HANDLER)
@@ -84,7 +88,7 @@ async def get_id(bot: Client, message: Message):
                 f"**Forwarded User ID**: `{message.reply_to_message.forward_from.id}`\n"
             )
         else:
-            user_detail = ( 
+            user_detail = (
                 f"**User ID**: `{message.reply_to_message.from_user.id if message.reply_to_message.from_user else None}`\n"
                 f"**Sender Chat ID**: `{message.reply_to_message.sender_chat.id if message.reply_to_message.sender_chat else None}`\n"
                 f"**Sender Chat Title**: `{message.reply_to_message.sender_chat.title if message.reply_to_message.sender_chat else None}`\n"
@@ -92,6 +96,11 @@ async def get_id(bot: Client, message: Message):
             )
         user_detail += f"**Message ID**: `{message.reply_to_message.id}`\n\n"
         user_detail += file_id
-        await message.edit_text(user_detail)
+        try:
+            await message.reply_text(user_detail)
+        except ChannelInvalid:
+            await message.reply_text("Channel Invalid")
+        except Exception as e:
+            await message.reply_text(f"Error: {e}")
     else:
-        await message.edit_text(f"**Chat ID**: `{message.chat.id}`")
+        await message.reply_text(f"**Chat ID**: `{message.chat.id}`")

Akeno/plugins/info.py

@@ -30,20 +30,12 @@ async def who_is(client: Client, message: Message):
     try:
         user = await client.get_users(user_id)
         username = f"@{user.username}" if user.username else "-"
-        first_name = f"{user.first_name}" if user.first_name else "-"
-        last_name = f"{user.last_name}" if user.last_name else "-"
-        fullname = (
-            f"{user.first_name} {user.last_name}" if user.last_name else user.first_name
-        )
-        user_details = (await client.get_chat(user.id)).bio
-        bio = f"{user_details}" if user_details else "-"
-        h = f"{user.status}"
-        if h.startswith("UserStatus"):
-            y = h.replace("UserStatus.", "")
-            status = y.capitalize()
-        else:
-            status = "-"
-        dc_id = f"{user.dc_id}" if user.dc_id else "-"
+        first_name = user.first_name or "-"
+        last_name = user.last_name or "-"
+        fullname = f"{user.first_name} {user.last_name}" if user.last_name else user.first_name
+        user_details = (await client.get_chat(user.id)).bio or "-"
+        status = user.status.replace("UserStatus.", "").capitalize() if user.status.startswith("UserStatus") else "-"
+        dc_id = user.dc_id or "-"
         common = await client.get_common_chats(user.id)
         out_str = f"""<b>USER INFORMATION:</b>
 
@@ -83,4 +75,4 @@ async def who_is(client: Client, message: Message):
         return await ex.edit(f"**INFO:** `{e}`")
 
 module = modules_help.add_module("info", __file__)
-module.add_command("info", "to info view users.") 
+module.add_command("info", "to info view users.")

Akeno/plugins/youtube.py

@@ -3,6 +3,7 @@ import time
 
 import requests
 from pyrogram.types import Message
+from youtube_search import YoutubeSearch
 from yt_dlp import YoutubeDL
 
 from Akeno.utils.database import db
@@ -14,6 +15,60 @@ from Akeno.utils.scripts import progress
 from config import *
 
 
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["ytsa"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def youtube_search_audio(_, message: Message):
+    if len(message.command) < 2:
+        return await message.reply_text(
+            "Give a valid youtube search to download audio."
+        )
+    query = await input_user(message)
+    results = YoutubeSearch(query, max_results=5).to_dict()
+    watch = results[0]["url_suffix"]
+    url_suffix = watch.split("/")[1]
+    okk = f"https://youtube.com/{url_suffix}"
+    pro = await message.reply_text("Checking ...")
+    status, url = YoutubeDriver.check_url(okk)
+    if not status:
+        return await pro.edit_text(url)
+    await pro.edit_text("🎼 __Downloading audio ...__")
+    try:
+        with YoutubeDL(YoutubeDriver.song_options()) as ytdl:
+            yt_data = ytdl.extract_info(url, False)
+            yt_file = ytdl.prepare_filename(yt_data)
+            ytdl.process_info(yt_data)
+        upload_text = f"**⬆️ 𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
+        await pro.edit_text(upload_text)
+        response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
+        with open(f"{yt_file}.jpg", "wb") as f:
+            f.write(response.content)
+        await message.reply_audio(
+            f"{yt_file}.mp3",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            performer="[Akeno UB]",
+            title=yt_data["title"],
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
+        )
+        await pro.delete()
+    except Exception as e:
+        return await pro.edit_text(f"**🍀 Audio not Downloaded:** `{e}`")
+    try:
+        os.remove(f"{yt_file}.jpg")
+        os.remove(f"{yt_file}.mp3")
+    except:
+        pass
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["yta"], CMD_HANDLER)
@@ -64,6 +119,58 @@ async def youtube_audio(_, message: Message):
     except:
         pass
 
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["ytva"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def ytvideo_search(client: Client, message: Message):
+    if len(message.command) < 2:
+        return await message.reply_text(
+            "Give a valid youtube search to download video."
+        )
+    query = await input_user(message)
+    results = YoutubeSearch(query, max_results=5).to_dict()
+    watch = results[0]["url_suffix"]
+    url_suffix = watch.split("/")[1]
+    okk = f"https://youtube.com/{url_suffix}"
+    pro = await message.reply_text("Checking ...")
+    status, url = YoutubeDriver.check_url(okk)
+    if not status:
+        return await pro.edit_text(url)
+    await pro.edit_text("🎼 __Downloading video ...__")
+    try:
+        with YoutubeDL(YoutubeDriver.video_options()) as ytdl:
+            yt_data = ytdl.extract_info(url, True)
+            yt_file = yt_data["id"]
+
+        upload_text = f"**⬆️ 𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 𝖲𝗈𝗇𝗀 ...** \n\n**𝖳𝗂𝗍𝗅𝖾:** `{yt_data['title'][:50]}`\n**𝖢𝗁𝖺𝗇𝗇𝖾𝗅:** `{yt_data['channel']}`"
+        await pro.edit_text(upload_text)
+        response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
+        with open(f"{yt_file}.jpg", "wb") as f:
+            f.write(response.content)
+        await message.reply_video(
+            f"{yt_file}.mp4",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
+        )
+        await pro.delete()
+    except Exception as e:
+        return await pro.edit_text(f"**🍀 Video not Downloaded:** `{e}`")
+    try:
+        os.remove(f"{yt_file}.jpg")
+        os.remove(f"{yt_file}.mp4")
+    except:
+        pass
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["ytv"], CMD_HANDLER)
@@ -135,6 +242,8 @@ async def ytlink(_, message: Message):
     await pro.edit_text(text, disable_web_page_preview=True)
 
 module = modules_help.add_module("youtube", __file__)
-module.add_command("yta", "Download the youtube video in .mp3 format!.")
-module.add_command("ytv", "Download the youtube video in .mp4 format!")
+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")