LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 09:26:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support async progress_callback in upload_file (#1259)

Browse Source
This commit is contained in:
painor 2019-08-19 20:11:13 +01:00 committed by Lonami
parent 42874de2b2
commit bd7ab23a8f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
telethon/client/uploads.py
View File

@ -573,7 +573,9 @@ class UploadMethods:
self._log[__name__].debug('Uploaded %d/%d', self._log[__name__].debug('Uploaded %d/%d',
part_index + 1, part_count) part_index + 1, part_count)
if progress_callback: if progress_callback:
progress_callback(stream.tell(), file_size) r = progress_callback(stream.tell(), file_size)
if inspect.isawaitable(r):
await r
else: else:
raise RuntimeError( raise RuntimeError(
'Failed to upload file part {}.'.format(part_index)) 'Failed to upload file part {}.'.format(part_index))