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