From 9de365531a6f1e0cbc5159be85b0cd3cb2c50491 Mon Sep 17 00:00:00 2001
From: Steve Owens <steve@doitnext.com>
Date: Sun, 11 Dec 2016 14:44:45 -0800
Subject: [PATCH] Address run_forever() terminates too early bug

https://github.com/Lawouach/WebSocket-for-Python/issues/152
---
 ws4py/websocket.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ws4py/websocket.py b/ws4py/websocket.py
index 00c56b7..16ac434 100644
--- a/ws4py/websocket.py
+++ b/ws4py/websocket.py
@@ -404,9 +404,7 @@ def terminate(self):
         good and cleanup resources by unsetting
         the `environ` and `stream` attributes.
         """
-        s = self.stream
-
-        self.client_terminated = self.server_terminated = True
+        s = self.stream       
 
         try:
             if s.closing is None:
@@ -414,6 +412,7 @@ def terminate(self):
             else:
                 self.closed(s.closing.code, s.closing.reason)
         finally:
+            self.client_terminated = self.server_terminated = True
             self.close_connection()
 
             # Cleaning up resources