diff --git a/bot/helper/download_tools.py b/bot/helper/download_tools.py index a28aded..a72928c 100644 --- a/bot/helper/download_tools.py +++ b/bot/helper/download_tools.py @@ -32,7 +32,7 @@ class DownloadHelper: download = aria2.add_magnet(link, {'dir': DOWNLOAD_DIR + str(self.__listener.message.message_id)}) self.__is_torrent = True else: - self.__listener.onDownloadError("No download URL or URL malformed") + self.__listener.onDownloadError("No download URL or URL malformed", None, None) return download_dict[self.__listener.message.message_id] = DownloadStatus(download.gid, self.__listener.message.message_id) diff --git a/bot/mirror.py b/bot/mirror.py index 8b8a966..6fcb2d9 100644 --- a/bot/mirror.py +++ b/bot/mirror.py @@ -33,11 +33,11 @@ class MirrorListener(listeners.MirrorListeners): def onDownloadError(self, error, progress_status_list: list, index: int): LOGGER.error(error) - + if progress_status_list is not None: + fs_utils.clean_download(progress_status_list[index].path()) msg = f"@{self.message.from_user.username} your download has been cancelled due to: {error}" sendMessage(msg, self.context, self.update) del download_dict[self.message.message_id] - fs_utils.clean_download(progress_status_list[index].path()) def onUploadStarted(self, progress_status_list: list, index: int): pass