LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-16 19:41:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support asynchronous progress_callback's

Browse Source
This commit is contained in:
Lonami 2019-07-04 15:34:51 +02:00 committed by GitHub
parent 3f19f6fd50
commit 0d9e639f4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
telethon/client/downloads.py
View File

@ -3,6 +3,7 @@ import io
import os import os
import pathlib import pathlib
import typing import typing
import inspect
from .. import utils, helpers, errors, hints from .. import utils, helpers, errors, hints
from ..requestiter import RequestIter from ..requestiter import RequestIter
@ -434,7 +435,9 @@ class DownloadMethods:
input_location, request_size=part_size, dc_id=dc_id): input_location, request_size=part_size, dc_id=dc_id):
f.write(chunk) f.write(chunk)
if progress_callback: if progress_callback:
progress_callback(f.tell(), file_size) r = progress_callback(f.tell(), file_size)
if inspect.isawaitable(r):
await r
f.flush() f.flush()
if in_memory: if in_memory: