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

fix tg status

This commit is contained in:
Anas 2021-07-09 02:49:06 +03:00 committed by GitHub
parent dedf1a330a
commit cdfbdc48f8
Signed by untrusted user: GitHub
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
bot/helper/mirror_utils/download_utils/telegram_downloader.py
View File

@ -4,15 +4,14 @@ import time
from bot import LOGGER, download_dict, download_dict_lock, app, STOP_DUPLICATE_MIRROR from bot import LOGGER, download_dict, download_dict_lock, app, STOP_DUPLICATE_MIRROR
from .download_helper import DownloadHelper from .download_helper import DownloadHelper
from ..status_utils.telegram_download_status import TelegramDownloadStatus from ..status_utils.telegram_download_status import TelegramDownloadStatus
from bot.helper.telegram_helper.message_utils import * from bot.helper.telegram_helper.message_utils import sendMarkup, sendStatusMessage
from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper from bot.helper.mirror_utils.upload_utils.gdriveTools import GoogleDriveHelper
from time import sleep
from bot.helper.ext_utils.bot_utils import get_readable_file_size
global_lock = threading.Lock() global_lock = threading.Lock()
GLOBAL_GID = set() GLOBAL_GID = set()
logging.getLogger("pyrogram").setLevel(logging.WARNING) logging.getLogger("pyrogram").setLevel(logging.WARNING)
class TelegramDownloadHelper(DownloadHelper): class TelegramDownloadHelper(DownloadHelper):
def __init__(self, listener): def __init__(self, listener):
super().__init__() super().__init__()
@ -100,26 +99,21 @@ class TelegramDownloadHelper(DownloadHelper):
name = filename name = filename
path = path + name path = path + name
if download or STOP_DUPLICATE_MIRROR: if download:
LOGGER.info(f"Checking File/Folder if already in Drive...") if STOP_DUPLICATE_MIRROR:
sleep(1) LOGGER.info(f"Checking File/Folder if already in Drive...")
if self.__listener.isTar: if self.__listener.isTar:
name = name + ".tar" name = name + ".tar"
if self.__listener.extract: if self.__listener.extract:
smsg = None smsg = None
else: else:
gd = GoogleDriveHelper() gd = GoogleDriveHelper()
smsg, button = gd.drive_list(name) smsg, button = gd.drive_list(name)
if smsg: if smsg:
self.__onDownloadError('File/Folder is already available in Drive.\n\n') self.__onDownloadError('File/Folder is already available in Drive.\n\n')
sendMarkup("Here are the search results:", sendMarkup("Here are the search results:", self.__listener.bot, self.__listener.update, button)
self.__listener.bot, return
self.__listener.update, sendStatusMessage(self.__listener.update, self.__listener.bot)
button
)
self.__is_cancelled = True
return
self.__onDownloadStart(name, media.file_size, media.file_id) self.__onDownloadStart(name, media.file_size, media.file_id)
LOGGER.info(f'Downloading Telegram file with id: {media.file_id}') LOGGER.info(f'Downloading Telegram file with id: {media.file_id}')
threading.Thread(target=self.__download, args=(_message, path)).start() threading.Thread(target=self.__download, args=(_message, path)).start()