yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-03 14:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add a retry mechanism when uploading chunks

Browse source
This commit is contained in:
Dan 2019-02-05 17:10:00 +01:00
parent a2263ad8ce
commit 712b390f77
1 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
pyrogram/client/client.py
View file

@ -1439,21 +1439,25 @@ class Client(Methods, BaseClient):
md5_sum = "".join([hex(i)[2:].zfill(2) for i in md5_sum.digest()]) md5_sum = "".join([hex(i)[2:].zfill(2) for i in md5_sum.digest()])
break break
if is_big: for _ in range(3):
rpc = functions.upload.SaveBigFilePart( if is_big:
file_id=file_id, rpc = functions.upload.SaveBigFilePart(
file_part=file_part, file_id=file_id,
file_total_parts=file_total_parts, file_part=file_part,
bytes=chunk file_total_parts=file_total_parts,
) bytes=chunk
else: )
rpc = functions.upload.SaveFilePart( else:
file_id=file_id, rpc = functions.upload.SaveFilePart(
file_part=file_part, file_id=file_id,
bytes=chunk file_part=file_part,
) bytes=chunk
)
assert session.send(rpc), "Couldn't upload file" if session.send(rpc):
break
else:
raise AssertionError("Telegram didn't accept chunk #{} of {}".format(file_part, path))
if is_missing_part: if is_missing_part:
return return