TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-28 00:56:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add download support for non-CDN files

Browse Source
This commit is contained in:
Dan 2017-12-20 16:21:56 +01:00
parent 0fbc5e1a90
commit 0f2ef8e681
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pyrogram/client/client.py
View File

@ -806,8 +806,10 @@ class Client:
else:
return r
def get_file(self, id: int, access_hash: int, version: int):
def get_file(self, id: int, access_hash: int, version: int = 0):
# TODO: Refine
# TODO: Use proper file name and extension
limit = 512 * 1024
offset = 0
@ -827,6 +829,28 @@ class Client:
)
)
if isinstance(r, types.upload.File):
with open("_".join([str(id), str(access_hash), str(version)]) + ".jpg", "wb") as f:
while True:
chunk = r.bytes
if not chunk:
break
f.write(chunk)
offset += limit
r = session.send(
functions.upload.GetFile(
location=types.InputDocumentFileLocation(
id=id,
access_hash=access_hash,
version=version
),
offset=offset,
limit=limit
)
)
if isinstance(r, types.upload.FileCdnRedirect):
ctr = CTR(r.encryption_key, r.encryption_iv)