mirror of
https://github.com/anonymousx97/social-dl.git
synced 2025-02-20 11:13:19 +08:00
ae04775a3c
Fixes: • Fixed instagram set sauce bug. • Fixed Tasks getting overwritten bug. • Fixed Get_output() wasn't being cancelled on cmd cancellation. Add/Misc: • Add Task IDs in Custom message object. • Clean up/Improve logic for .shell .
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
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=message.task_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):
|
|
message = Message.parse_message(message)
|
|
func = Config.CMD_DICT[message.cmd]
|
|
coro = func(bot, message)
|
|
try:
|
|
task = asyncio.Task(coro, name=message.task_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):
|
|
message = Message.parse_message(message)
|
|
coro = dl(bot, message)
|
|
try:
|
|
task = asyncio.Task(coro, name=message.task_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",
|
|
)
|