File size: 3,601 Bytes
db78256 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
from .. import (
task_dict,
task_dict_lock,
user_data,
queued_up,
queued_dl,
queue_dict_lock,
)
from ..core.config_manager import Config
from ..helper.ext_utils.bot_utils import new_task
from ..helper.ext_utils.status_utils import get_task_by_gid
from ..helper.telegram_helper.bot_commands import BotCommands
from ..helper.telegram_helper.message_utils import send_message
from ..helper.ext_utils.task_manager import start_dl_from_queued, start_up_from_queued
@new_task
async def remove_from_queue(_, message):
user_id = (message.from_user or message.sender_chat).id
msg = message.text.split()
status = msg[1] if len(msg) > 1 and msg[1] in ["fd", "fu"] else ""
if status and len(msg) > 2 or not status and len(msg) > 1:
gid = msg[2] if status else msg[1]
task = await get_task_by_gid(gid)
if task is None:
await send_message(message, f"GID: <code>{gid}</code> Not Found.")
return
elif reply_to_id := message.reply_to_message_id:
async with task_dict_lock:
task = task_dict.get(reply_to_id)
if task is None:
await send_message(message, "This is not an active task!")
return
elif len(msg) in {1, 2}:
msg = f"""Reply to an active Command message which was used to start the download/upload.
<code>/{BotCommands.ForceStartCommand[0]}</code> fd (to remove it from download queue) or fu (to remove it from upload queue) or nothing to start remove it from both download and upload queue.
Also send <code>/{BotCommands.ForceStartCommand[0]} GID</code> fu|fd or obly gid to force start by removeing the task rom queue!
Examples:
<code>/{BotCommands.ForceStartCommand[1]}</code> GID fu (force upload)
<code>/{BotCommands.ForceStartCommand[1]}</code> GID (force download and upload)
By reply to task cmd:
<code>/{BotCommands.ForceStartCommand[1]}</code> (force download and upload)
<code>/{BotCommands.ForceStartCommand[1]}</code> fd (force download)
"""
await send_message(message, msg)
return
if (
Config.OWNER_ID != user_id
and task.listener.user_id != user_id
and (user_id not in user_data or not user_data[user_id].get("SUDO"))
):
await send_message(message, "This task is not for you!")
return
listener = task.listener
msg = ""
async with queue_dict_lock:
if status == "fu":
listener.force_upload = True
if listener.mid in queued_up:
await start_up_from_queued(listener.mid)
msg = "Task have been force started to upload!"
else:
msg = "Force upload enabled for this task!"
elif status == "fd":
listener.force_download = True
if listener.mid in queued_dl:
await start_dl_from_queued(listener.mid)
msg = "Task have been force started to download only!"
else:
msg = "This task not in download queue!"
else:
listener.force_download = True
listener.force_upload = True
if listener.mid in queued_up:
await start_up_from_queued(listener.mid)
msg = "Task have been force started to upload!"
elif listener.mid in queued_dl:
await start_dl_from_queued(listener.mid)
msg = "Task have been force started to download and upload will start once download finish!"
else:
msg = "This task not in queue!"
if msg:
await send_message(message, msg)
|