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

Allow mirroring tg file again once file download is complete 

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2020-03-22 16:47:40 +05:30
parent 7d74cf29d8
commit 9369c7dc98
2 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
bot/__main__.py
View File

@ -1,14 +1,16 @@
from telegram.ext import CommandHandler, run_async
from bot import dispatcher, LOGGER, updater, botStartTime
from bot.helper.ext_utils import fs_utils
from .helper.ext_utils.bot_utils import get_readable_file_size, get_readable_time
import signal
import time
from bot.helper.telegram_helper.message_utils import *
import shutil import shutil
from .helper.telegram_helper.filters import CustomFilters import signal
from telegram.ext import CommandHandler, run_async
from bot import dispatcher, updater, botStartTime
from bot.helper.ext_utils import fs_utils
from bot.helper.telegram_helper.bot_commands import BotCommands from bot.helper.telegram_helper.bot_commands import BotCommands
from .modules import authorize, list, cancel_mirror, mirror_status, mirror from bot.helper.telegram_helper.message_utils import *
from .helper.ext_utils.bot_utils import get_readable_file_size, get_readable_time
from .helper.telegram_helper.filters import CustomFilters
from .modules import authorize, list, cancel_mirror, mirror_status, mirror, clone
@run_async @run_async
def stats(bot, update): def stats(bot, update):

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

@ -3,7 +3,7 @@ import time
from pyrogram import Client from pyrogram import Client
from bot import LOGGER, bot, download_dict, download_dict_lock, TELEGRAM_API, \ from bot import LOGGER, download_dict, download_dict_lock, TELEGRAM_API, \
TELEGRAM_HASH, USER_SESSION_STRING TELEGRAM_HASH, USER_SESSION_STRING
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
@ -55,7 +55,8 @@ class TelegramDownloadHelper(DownloadHelper):
return 0 return 0
def __onDownloadComplete(self): def __onDownloadComplete(self):
with global_lock:
GLOBAL_GID.remove(self.gid)
self.__listener.onDownloadComplete() self.__listener.onDownloadComplete()
def __download(self, message, path): def __download(self, message, path):
@ -64,9 +65,6 @@ class TelegramDownloadHelper(DownloadHelper):
self.__onDownloadComplete() self.__onDownloadComplete()
def add_download(self, message, path): def add_download(self, message, path):
if message.chat.type == "private":
_message = self.__user_bot.get_messages(bot.get_me().id, message.message_id)
else:
_message = self.__user_bot.get_messages(message.chat.id, message.message_id) _message = self.__user_bot.get_messages(message.chat.id, message.message_id)
media = None media = None
media_array = [_message.document, _message.video, _message.audio] media_array = [_message.document, _message.video, _message.audio]