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

Make use of tempfile.NamedTemporaryFile in getfile, use context managers

Browse source
This commit is contained in:
Eric Blundell 2018-03-21 04:07:55 -05:00
parent b45960212b
commit 0f4e29584a
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pyrogram/client/client.py
View file

@ -2250,7 +2250,7 @@ class Client:
limit = 1024 * 1024 limit = 1024 * 1024
offset = 0 offset = 0
fd, file_name = tempfile.mkstemp() file_name = None
try: try:
r = session.send( r = session.send(
@ -2262,7 +2262,9 @@ class Client:
) )
if isinstance(r, types.upload.File): if isinstance(r, types.upload.File):
with os.fdopen(fd, "wb") as f: with tempfile.NamedTemporaryFile('wb', delete=False) as f:
file_name = f.name
while True: while True:
chunk = r.bytes chunk = r.bytes
@ -2299,7 +2301,8 @@ class Client:
cdn_session.start() cdn_session.start()
try: try:
with os.fdopen(fd, "wb") as f: with tempfile.NamedTemporaryFile('wb', delete=False) as f:
file_name = f.name
while True: while True:
r2 = cdn_session.send( r2 = cdn_session.send(
functions.upload.GetCdnFile( functions.upload.GetCdnFile(