From 65c71add56406deccbae0708949a4291825a97da Mon Sep 17 00:00:00 2001 From: Anas <77075674+anasty17@users.noreply.github.com> Date: Thu, 10 Jun 2021 08:11:40 +0300 Subject: [PATCH] fix status msg for mega (#155) * fix * fix --- bot/helper/mirror_utils/download_utils/mega_downloader.py | 5 ++++- bot/modules/mirror.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/bot/helper/mirror_utils/download_utils/mega_downloader.py b/bot/helper/mirror_utils/download_utils/mega_downloader.py index e902565..8639948 100644 --- a/bot/helper/mirror_utils/download_utils/mega_downloader.py +++ b/bot/helper/mirror_utils/download_utils/mega_downloader.py @@ -1,7 +1,7 @@ from bot import LOGGER, MEGA_API_KEY, download_dict_lock, download_dict, MEGA_EMAIL_ID, MEGA_PASSWORD import threading from mega import (MegaApi, MegaListener, MegaRequest, MegaTransfer, MegaError) -from bot.helper.telegram_helper.message_utils import update_all_messages, sendMarkup, deleteMessage, sendMessage +from bot.helper.telegram_helper.message_utils import * import os from bot.helper.ext_utils.bot_utils import new_thread, get_mega_link_type, get_readable_file_size from bot.helper.mirror_utils.status_utils.mega_download_status import MegaDownloadStatus @@ -168,6 +168,7 @@ class MegaDownloadHelper: node = folder_api.authorizeNode(mega_listener.node) if STOP_DUPLICATE_MEGA: msg = sendMessage('Check the File/Folder if already in drive...', listener.bot, listener.update) + LOGGER.info(f'Check the File/Folder if already in drive') mname = node.getName() if listener.isTar == True: mname = mname + ".tar" @@ -185,6 +186,7 @@ class MegaDownloadHelper: deleteMessage(listener.bot, msg) if MEGA_LIMIT is not None: msg2 = sendMessage('Check the File/Folder size...', listener.bot, listener.update) + LOGGER.info(f'Check the File/Folder size') limit = MEGA_LIMIT limit = limit.split(' ', maxsplit=1) limitint = int(limit[0]) @@ -203,6 +205,7 @@ class MegaDownloadHelper: return else: deleteMessage(listener.bot, msg2) + sendStatusMessage(listener.update, listener.bot) with download_dict_lock: download_dict[listener.uid] = MegaDownloadStatus(mega_listener, listener) os.makedirs(path) diff --git a/bot/modules/mirror.py b/bot/modules/mirror.py index 2924b1d..43aa98b 100644 --- a/bot/modules/mirror.py +++ b/bot/modules/mirror.py @@ -298,7 +298,6 @@ def _mirror(bot, update, isTar=False, extract=False): else: mega_dl = MegaDownloadHelper() mega_dl.add_download(link, f'{DOWNLOAD_DIR}/{listener.uid}/', listener) - sendStatusMessage(update, bot) else: ariaDlManager.add_download(link, f'{DOWNLOAD_DIR}/{listener.uid}/', listener, name) sendStatusMessage(update, bot)