social-dl/app/social_dl.py

import asyncio
import traceback

from app import Config, bot
from app.core import filters
from app.core.MediaHandler import ExtractAndSendMedia
from app.core.message import Message


@bot.add_cmd(cmd="dl")
async def dl(bot, message):
    reply = await bot.send_message(
        chat_id=message.chat.id, text="`trying to download...`"
    )
    coro = ExtractAndSendMedia.process(message)
    task = asyncio.Task(coro, name=f"{message.chat.id}-{message.id}")
    media = await task
    if media.exceptions:
        exceptions = "\n".join(media.exceptions)
        await bot.log(
            traceback=exceptions,
            func="DL",
            chat=message.chat.title or message.chat.first_name,
            name="traceback.txt",
        )
        return await reply.edit(f"Media Download Failed.")
    if media.media_objects:
        await message.delete()
    await reply.delete()


@bot.on_message(filters.user_filter)
@bot.on_edited_message(filters.user_filter)
async def cmd_dispatcher(bot, message):
    parsed_message = Message.parse_message(message)
    func = Config.CMD_DICT[parsed_message.cmd]
    coro = func(bot, parsed_message)
    try:
        task = asyncio.Task(coro, name=f"{message.chat.id}-{message.id}")
        await task
    except asyncio.exceptions.CancelledError:
        await bot.log(text=f"<b>#Cancelled</b>:\n<code>{message.text}</code>")
    except BaseException:
        await bot.log(
            traceback=str(traceback.format_exc()),
            chat=message.chat.title or message.chat.first_name,
            func=func.__name__,
            name="traceback.txt",
        )


@bot.on_message(filters.chat_filter)
async def dl_dispatcher(bot, message):
    parsed_message = Message.parse_message(message)
    coro = dl(bot, parsed_message)
    try:
        task = asyncio.Task(coro, name=f"{message.chat.id}-{message.id}")
        await task
    except asyncio.exceptions.CancelledError:
        await bot.log(text=f"<b>#Cancelled</b>:\n<code>{message.text}</code>")
    except BaseException:
        await bot.log(
            traceback=str(traceback.format_exc()),
            chat=message.chat.title or message.chat.first_name,
            func=dl.__name__,
            name="traceback.txt",
        )