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

Wait for threads to stop when closing the session

Browse Source
This commit is contained in:
Dan 2017-12-23 23:43:20 +01:00
parent 0231ce58d5
commit 5128f6310b
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyrogram/session/session.py
View File

@ -124,11 +124,6 @@ class Session:
self.current_salt = FutureSalt(0, 0, self._send(functions.Ping(0)).new_server_salt)
self.current_salt = self._send(functions.GetFutureSalts(1)).salts[0]
if self.next_salt_thread is not None:
self.next_salt_thread.join()
self.next_salt_thread_event.clear()
self.next_salt_thread = Thread(target=self.next_salt, name="NextSaltThread")
self.next_salt_thread.start()
@ -147,11 +142,6 @@ class Session:
)
).text
if self.ping_thread is not None:
self.ping_thread.join()
self.ping_thread_event.clear()
self.ping_thread = Thread(target=self.ping, name="PingThread")
self.ping_thread.start()
@ -170,8 +160,16 @@ class Session:
def stop(self):
self.is_connected.clear()
self.ping_thread_event.set()
self.next_salt_thread_event.set()
self.ping_thread.join()
self.next_salt_thread.join()
self.ping_thread_event.clear()
self.next_salt_thread_event.clear()
self.connection.close()
for i in range(self.WORKERS):