Merge pull request #69 from EriHoss/stop_getfile_sessions

Make sure started Session is always stopped upon exceptions in Client.get_file
2018-05-01 11:45:51 +02:00 · 2018-05-01 11:45:51 +02:00 · 11aa46665e
commit 11aa46665e
parent 20146ded28 11d0eb6ea1
1 changed files with 10 additions and 6 deletions
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@ -2988,13 +2988,17 @@ class Client:
            )

            session.start()
-
-            session.send(
-                functions.auth.ImportAuthorization(
-                    id=exported_auth.id,
-                    bytes=exported_auth.bytes
+            try:
+                session.send(
+                    functions.auth.ImportAuthorization(
+                        id=exported_auth.id,
+                        bytes=exported_auth.bytes
+                    )
                )
-            )
+            except Exception as e:
+                session.stop()
+                raise e
+
        else:
            session = Session(
                dc_id,