2
0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +00:00

Make sure to clean temp files

This commit is contained in:
Dan 2018-02-22 14:13:59 +01:00
parent 26adfc2ef4
commit 9c9a034f66

View File

@ -344,6 +344,11 @@ class Client:
finally:
done.set()
try:
os.remove("./{}".format(tmp_file_name))
except FileNotFoundError:
pass
log.debug("{} stopped".format(name))
def updates_worker(self):
@ -1879,7 +1884,7 @@ class Client:
version=version
)
file_name = str(MsgId())
file_name = "download_{}.temp".format(MsgId())
limit = 1024 * 1024
offset = 0
@ -2341,13 +2346,14 @@ class Client:
)
)
def download_media(self, message: types.Message, file_name: str = None):
def download_media(self, message: types.Message, file_name: str = None, block: bool = True):
done = Event()
media = message.media if isinstance(message, types.Message) else message
self.download_queue.put((media, file_name, done))
done.wait()
if block:
done.wait()
def add_contacts(self, contacts: list):
"""Use this method to add contacts to your Telegram address book.