hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-09 05:19:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support arbitrary file objects for .download_file() and update README

Browse Source
This commit is contained in:
Lonami Exo
2017-06-09 11:12:56 +02:00
parent 40f4c2533a
commit 40616ba704
3 changed files with 28 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
telethon/telegram_bare_client.py
View File

@@ -283,11 +283,11 @@ class TelegramBareClient:
def download_file(self,
input_location,
file_path,
file,
part_size_kb=None,
file_size=None,
progress_callback=None):
"""Downloads the given InputFileLocation to file_path.
"""Downloads the given InputFileLocation to file (a stream or str).
If 'progress_callback' is not None, it should be a function that
takes two parameters, (bytes_downloaded, total_bytes). Note that
@@ -303,11 +303,15 @@ class TelegramBareClient:
if part_size % 1024 != 0:
raise ValueError('The part size must be evenly divisible by 1024.')
# Ensure that we'll be able to download the media
utils.ensure_parent_dir_exists(file_path)
if isinstance(file, str):
# Ensure that we'll be able to download the media
utils.ensure_parent_dir_exists(file)
f = open(file, 'wb')
else:
f = file
offset_index = 0
with open(file_path, 'wb') as file:
try:
offset_index = 0
while True:
offset = offset_index * part_size
result = self.invoke(
@@ -319,8 +323,11 @@ class TelegramBareClient:
if not result.bytes:
return result.type # Return some extra information
file.write(result.bytes)
f.write(result.bytes)
if progress_callback:
progress_callback(file.tell(), file_size)
progress_callback(f.tell(), file_size)
finally:
if isinstance(file, str):
f.close()
# endregion