Reraise exceptions in a Py2/Py3 compatible way

Fixes #58
miguelgrinberg · Dec 9, 2017 · 4a33200 · 4a33200
1 parent a1eafcd
commit 4a33200
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/engineio/asyncio_socket.py b/engineio/asyncio_socket.py
@@ -1,5 +1,6 @@
 import asyncio
 import six
+import sys
 import time
 
 from . import exceptions
@@ -76,8 +77,9 @@ async def handle_get_request(self, environ):
         try:
             packets = await self.poll()
         except exceptions.QueueEmpty:
+            exc = sys.exc_info()
             await self.close(wait=False)
-            raise
+            six.reraise(*exc)
         return packets
 
     async def handle_post_request(self, environ):

diff --git a/engineio/socket.py b/engineio/socket.py
@@ -1,4 +1,5 @@
 import six
+import sys
 import time
 
 from . import exceptions
@@ -91,8 +92,9 @@ def handle_get_request(self, environ, start_response):
         try:
             packets = self.poll()
         except exceptions.QueueEmpty:
+            exc = sys.exc_info()
             self.close(wait=False)
-            raise
+            six.reraise(*exc)
         return packets
 
     def handle_post_request(self, environ):