evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

prevent cleaning of uploading/archiving downloads

This commit is contained in:
jaskaranSM 2020-01-02 10:25:45 +05:30
parent f11e9a206c
commit b84ffcc416
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
bot/modules/cancel_mirror.py
View File

@ -41,16 +41,17 @@ def cancel_mirror(bot, update):
@run_async @run_async
def cancel_all(update, bot): def cancel_all(update, bot):
with download_dict_lock: with download_dict_lock:
count = 0
for dlDetails in list(download_dict.values()): for dlDetails in list(download_dict.values()):
if not dlDetails.status() == "Uploading" or dlDetails.status() == "Archiving": if not dlDetails.status() == "Uploading" or dlDetails.status() == "Archiving":
aria2.pause([dlDetails.download()]) aria2.pause([dlDetails.download()])
count += 1
continue continue
if dlDetails.status() == "Queued": if dlDetails.status() == "Queued":
count += 1
dlDetails._listener.onDownloadError("Download Manually Cancelled By user.") dlDetails._listener.onDownloadError("Download Manually Cancelled By user.")
delete_all_messages() delete_all_messages()
sendMessage('Cancelled all downloads!', update, bot) sendMessage(f'Cancelled {count} downloads!', update, bot)
sleep(0.5) # Wait a Second For Aria2 To free Resources.
clean_download(DOWNLOAD_DIR)
cancel_mirror_handler = CommandHandler(BotCommands.CancelMirror, cancel_mirror, cancel_mirror_handler = CommandHandler(BotCommands.CancelMirror, cancel_mirror,