mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Make sure to clean temp files

Browse Source
This commit is contained in:
Dan 2018-02-22 14:13:59 +01:00
parent 26adfc2ef4
commit 9c9a034f66
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pyrogram/client/client.py
View File

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