From bd7ab23a8fd89940219e5e45394da86803ad04ef Mon Sep 17 00:00:00 2001 From: painor <30215770+painor@users.noreply.github.com> Date: Mon, 19 Aug 2019 20:11:13 +0100 Subject: [PATCH] Support async progress_callback in upload_file (#1259) --- telethon/client/uploads.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/telethon/client/uploads.py b/telethon/client/uploads.py index ed5ad343..6b133053 100644 --- a/telethon/client/uploads.py +++ b/telethon/client/uploads.py @@ -573,7 +573,9 @@ class UploadMethods: self._log[__name__].debug('Uploaded %d/%d', part_index + 1, part_count) if progress_callback: - progress_callback(stream.tell(), file_size) + r = progress_callback(stream.tell(), file_size) + if inspect.isawaitable(r): + await r else: raise RuntimeError( 'Failed to upload file part {}.'.format(part_index))