Fix websocket handler bug on Python3.5 with no uvloop

2017-05-15 10:50:13 +08:00 · 2017-05-15 10:50:13 +08:00 · 2f84cdd708
commit 2f84cdd708
parent 7cc02e84ed
1 changed files with 6 additions and 1 deletions
--- a/sanic/app.py
+++ b/sanic/app.py
@ -206,7 +206,12 @@ class Sanic:
        def response(handler):
            async def websocket_handler(request, *args, **kwargs):
                request.app = self
-                protocol = request.transport.get_protocol()
+                try:
+                    protocol = request.transport.get_protocol()
+                except AttributeError:
+                    # On Python3.5 the Transport classes in asyncio do not
+                    # have a get_protocol() method as in uvloop
+                    protocol = request.transport._protocol
                ws = await protocol.websocket_handshake(request)

                # schedule the application handler