63 lines
2.7 KiB
Python
63 lines
2.7 KiB
Python
from telegram.ext import CommandHandler, run_async
|
|
from bot.helper.telegram_helper.message_utils import *
|
|
from bot import download_dict, aria2, dispatcher, download_dict_lock, DOWNLOAD_DIR
|
|
from bot.helper.telegram_helper.filters import CustomFilters
|
|
from bot.helper.ext_utils.fs_utils import clean_download
|
|
from bot.helper.telegram_helper.bot_commands import BotCommands
|
|
from time import sleep
|
|
|
|
|
|
@run_async
|
|
def cancel_mirror(bot, update):
|
|
mirror_message = update.message.reply_to_message
|
|
with download_dict_lock:
|
|
keys = download_dict.keys()
|
|
dl = download_dict[mirror_message.message_id]
|
|
if mirror_message is None or mirror_message.message_id not in keys:
|
|
if '/mirror' in mirror_message.text or '/tarmirror' in mirror_message.text:
|
|
msg = 'Message has already been cancelled'
|
|
else:
|
|
msg = 'Please reply to the /mirror message which was used to start the download to cancel it!'
|
|
sendMessage(msg, bot, update)
|
|
return
|
|
if dl.status() == "Uploading":
|
|
sendMessage("Upload in Progress, Don't Cancel it.", bot, update)
|
|
return
|
|
elif dl.status() == "Archiving":
|
|
sendMessage("Archival in Progress, Don't Cancel it.", bot, update)
|
|
return
|
|
elif dl.status() != "Queued":
|
|
download = dl.download()
|
|
if len(download.followed_by_ids) != 0:
|
|
downloads = aria2.get_downloads(download.followed_by_ids)
|
|
aria2.pause(downloads)
|
|
aria2.pause([download])
|
|
else:
|
|
dl._listener.onDownloadError("Download stopped by user!")
|
|
sleep(1) # Wait a Second For Aria2 To free Resources.
|
|
clean_download(f'{DOWNLOAD_DIR}{mirror_message.message_id}/')
|
|
|
|
|
|
@run_async
|
|
def cancel_all(update, bot):
|
|
with download_dict_lock:
|
|
count = 0
|
|
for dlDetails in list(download_dict.values()):
|
|
if not dlDetails.status() == "Uploading" or dlDetails.status() == "Archiving":
|
|
aria2.pause([dlDetails.download()])
|
|
count += 1
|
|
continue
|
|
if dlDetails.status() == "Queued":
|
|
count += 1
|
|
dlDetails._listener.onDownloadError("Download Manually Cancelled By user.")
|
|
delete_all_messages()
|
|
sendMessage(f'Cancelled {count} downloads!', update, bot)
|
|
|
|
|
|
cancel_mirror_handler = CommandHandler(BotCommands.CancelMirror, cancel_mirror,
|
|
filters=CustomFilters.authorized_chat | CustomFilters.authorized_user)
|
|
cancel_all_handler = CommandHandler(BotCommands.CancelAllCommand, cancel_all,
|
|
filters=CustomFilters.owner_filter)
|
|
dispatcher.add_handler(cancel_all_handler)
|
|
dispatcher.add_handler(cancel_mirror_handler)
|