Pdisk Video Search bot

# (c) @AbirHasan2005

import os
import asyncio
import traceback
from dotenv import (
    load_dotenv
)
from pyrogram import (
    Client,
    filters,
    idle
)
from pyrogram.types import (
    Message,
    InlineKeyboardMarkup,
    InlineKeyboardButton
)
from pyrogram.errors import (
    MessageNotModified
)
from core.search_video import search_pdisk_videos

if os.path.exists("configs.env"):
    load_dotenv("configs.env")


class Configs(object):
    API_ID = int(os.environ.get("API_ID", 0))
    API_HASH = os.environ.get("API_HASH", "")
    BOT_TOKEN = os.environ.get("BOT_TOKEN", "")
    PDISK_USERNAME = os.environ.get("PDISK_USERNAME", "")
    PDISK_PASSWORD = os.environ.get("PDISK_PASSWORD", "")
    MAX_RESULTS = int(os.environ.get("MAX_RESULTS", 5))
    AUTH_CHATS = list(set(int(x) for x in os.environ.get("AUTH_CHATS", "0").split()))
    # Which PDisk Domain?
    PDISK_DOMAINS = [
        "https://www.cofilink.com/",
        "https://www.pdisk1.net/",
        "https://www.pdisk.net/"
    ]
    PDISK_DOMAIN = os.environ.get("PDISK_DOMAIN", PDISK_DOMAINS[2])


PDiskBot = Client(
    session_name=":memory:",
    api_id=Configs.API_ID,
    api_hash=Configs.API_HASH,
    bot_token=Configs.BOT_TOKEN
)



@PDiskBot.on_message(filters.command("start") & ~filters.edited)
async def start_handler(_, m: Message):
     await m.reply_text("**Hi,\n\n 🤖 I Am (Not at all 😏) Simple Bot \n\n🔀 I Can Search For Movies From Pdisk\n\n📁 You Can Send ME Any Movie_Name\n\n🤩 Then Stand Back And See The Magic\n\n👨🏻‍💻 I Was Developed @Being_ram_esh\n\n💬 U Can Also Add Me To Groups**", quote=True,
     reply_markup=InlineKeyboardMarkup([

                           [
             InlineKeyboardButton("Join Our Group", url="https://t.me/joinchat/RVUzWLee4lxiYWNl"),
             InlineKeyboardButton("Creator", url="https://t.me/being_ram_esh"),
             ]
                                
             ]))


@PDiskBot.on_message(~filters.edited, group=-1)
async def text_handler(_, m: Message):
   
    editable = await m.reply_text("**Searching🔎🔎**", quote=True,
                                 )
    
    response = await search_pdisk_videos(m.text.split(" ", 1)[-1], Configs.PDISK_USERNAME, Configs.PDISK_PASSWORD)
    if isinstance(response, Exception):
        trac
        try: await editable.edit("Failed to search!",
                                 reply_markup=InlineKeyboardMarkup([
                                     [InlineKeyboardButton("Support Group", url="https://t.me/cineworldzoneo")]
                                 ]))
        except MessageNotModified: pass
    elif not response["data"]["list"]:
        try: await editable.edit("Movie Not Found!\n\n",
                                 reply_markup=InlineKeyboardMarkup([
                                     [InlineKeyboardButton("Request Group", url="https://t.me/joinchat/RVUzWLee4lxiYWNl")]
                                 ]))
        except MessageNotModified: pass
    else:
        data = response["data"]["list"]
        text = ""
        count = 0
        for i in range(len(data)):
            if count > Configs.MAX_RESULTS:
                        break
            count += 1
            text += f"**Title:** `{data[i]['title']}`\n" \
                    f"**PDisk Link:** {Configs.PDISK_DOMAIN + 'share-video?videoid=' + data[i]['share_link'].split('=', 1)[-1]}\n\n"
                    
        try: await editable.edit(text, disable_web_page_preview=True)
        except MessageNotModified: pass


async def run():
    await PDiskBot.start()
    print("\n\nBot Started!\n\n")
    await idle()
    await PDiskBot.stop()
    print("\n\nBot Stopped!\n\n")


if __name__ == "__main__":
    asyncio.get_event_loop().run_until_complete(run())