2
0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00

fix open file leak in client.download_media

This commit is contained in:
Eric Blundell 2018-03-20 09:02:17 -05:00
parent 0694480a46
commit bd1234f227

View File

@ -2248,6 +2248,8 @@ class Client:
elif isinstance(file_out, str): elif isinstance(file_out, str):
f = open(file_out, 'wb') f = open(file_out, 'wb')
close_file = True
elif hasattr(file_out, 'write'): elif hasattr(file_out, 'write'):
f = file_out f = file_out
@ -2367,7 +2369,7 @@ class Client:
else: else:
return file_out return file_out
finally: finally:
if close_file and f and hasattr(f, 'close'): if close_file and f is not None:
f.close() f.close()
session.stop() session.stop()