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

Use bot session_string

This commit is contained in:
BruteWoorse 2021-03-10 00:49:55 +08:00 committed by GitHub
parent 72441215ac
commit d023e2c179
Signed by untrusted user: GitHub
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,10 +23,10 @@ class TelegramDownloadHelper(DownloadHelper):
self.__name = "" self.__name = ""
self.__gid = '' self.__gid = ''
self.__start_time = time.time() self.__start_time = time.time()
self.__user_bot = Client(api_id=TELEGRAM_API, self._bot = Client(api_id=TELEGRAM_API,
api_hash=TELEGRAM_HASH, api_hash=TELEGRAM_HASH,
session_name=USER_SESSION_STRING) session_name=USER_SESSION_STRING)
self.__user_bot.start() self._bot.start()
self.__is_cancelled = False self.__is_cancelled = False
@property @property
@ -53,7 +53,7 @@ class TelegramDownloadHelper(DownloadHelper):
def __onDownloadProgress(self, current, total): def __onDownloadProgress(self, current, total):
if self.__is_cancelled: if self.__is_cancelled:
self.__onDownloadError('Cancelled by user!') self.__onDownloadError('Cancelled by user!')
self.__user_bot.stop_transmission() self._bot.stop_transmission()
return return
with self.__resource_lock: with self.__resource_lock:
self.downloaded_bytes = current self.downloaded_bytes = current
@ -76,7 +76,7 @@ class TelegramDownloadHelper(DownloadHelper):
self.__listener.onDownloadComplete() self.__listener.onDownloadComplete()
def __download(self, message, path): def __download(self, message, path):
download = self.__user_bot.download_media(message, download = self._bot.download_media(message,
progress=self.__onDownloadProgress, file_name=path) progress=self.__onDownloadProgress, file_name=path)
if download is not None: if download is not None:
self.__onDownloadComplete() self.__onDownloadComplete()
@ -85,7 +85,7 @@ class TelegramDownloadHelper(DownloadHelper):
self.__onDownloadError('Internal error occurred') self.__onDownloadError('Internal error occurred')
def add_download(self, message, path): def add_download(self, message, path):
_message = self.__user_bot.get_messages(message.chat.id, message.message_id) _message = self._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]
for i in media_array: for i in media_array: