From f0c00c88013ba88a5c9e98b43f1446945cbbbe3d Mon Sep 17 00:00:00 2001
From: Eric Blundell <eblundell800@gmail.com>
Date: Tue, 20 Mar 2018 16:30:48 -0500
Subject: [PATCH] move first mkstemp to exception safe location in get_file

---
 pyrogram/client/client.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index b63d0b4f..529873ff 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -2247,10 +2247,9 @@ class Client:
                 version=version
             )
 
-        fd, file_name = tempfile.mkstemp()
-
         limit = 1024 * 1024
         offset = 0
+        file_name = None
 
         try:
             r = session.send(
@@ -2261,6 +2260,8 @@ class Client:
                 )
             )
 
+            fd, file_name = tempfile.mkstemp()
+
             if isinstance(r, types.upload.File):
                 with os.fdopen(fd, "wb") as f:
                     while True:
@@ -2374,10 +2375,11 @@ class Client:
         except Exception as e:
             log.error(e, exc_info=True)
 
-            try:
-                os.remove(file_name)
-            except OSError:
-                pass
+            if file_name:
+                try:
+                    os.remove(file_name)
+                except OSError:
+                    pass
         else:
             return file_name
         finally: