Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add initial support for photos download

Browse Source
This commit is contained in:
Dan 2017-12-23 12:26:26 +01:00
parent ef7ee13a56
commit 54fa3af4d5
1 changed files with 24 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
pyrogram/client/client.py
View File

@ -806,9 +806,29 @@ class Client:
else:
return r
def get_file(self, id: int, access_hash: int, version: int = 0):
def get_file(self,
id: int = None,
access_hash: int = None,
volume_id: int = None,
local_id: int = None,
secret: int = None,
version: int = 0):
# TODO: Refine
# TODO: Use proper file name and extension
# TODO: Remove redundant code
if volume_id: # Photos are accessed by volume_id, local_id, secret
location = types.InputFileLocation(
volume_id=volume_id,
local_id=local_id,
secret=secret
)
else: # Any other file can be more easily accessed by id and access_hash
location = types.InputDocumentFileLocation(
id=id,
access_hash=access_hash,
version=version
)
limit = 512 * 1024
offset = 0
@ -819,11 +839,7 @@ class Client:
try:
r = session.send(
functions.upload.GetFile(
location=types.InputDocumentFileLocation(
id=id,
access_hash=access_hash,
version=version
),
location=location,
offset=offset,
limit=limit
)
@ -842,11 +858,7 @@ class Client:
r = session.send(
functions.upload.GetFile(
location=types.InputDocumentFileLocation(
id=id,
access_hash=access_hash,
version=version
),
location=location,
offset=offset,
limit=limit
)
@ -869,11 +881,7 @@ class Client:
while True:
r2 = cdn_session.send(
functions.upload.GetCdnFile(
location=types.InputDocumentFileLocation(
id=id,
access_hash=access_hash,
version=version
),
location=location,
file_token=r.file_token,
offset=offset,
limit=limit