49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
|
from __future__ import absolute_import
|
||
|
|
import queue
|
||
|
|
import threading
|
||
|
|
import time
|
||
|
|
|
||
|
|
try:
|
||
|
|
from simple_websocket import Server, ConnectionClosed
|
||
|
|
_websocket_available = True
|
||
|
|
except ImportError: # pragma: no cover
|
||
|
|
_websocket_available = False
|
||
|
|
|
||
|
|
|
||
|
|
class WebSocketWSGI(object): # pragma: no cover
|
||
|
|
"""
|
||
|
|
This wrapper class provides a threading WebSocket interface that is
|
||
|
|
compatible with eventlet's implementation.
|
||
|
|
"""
|
||
|
|
def __init__(self, handler, server):
|
||
|
|
self.app = handler
|
||
|
|
|
||
|
|
def __call__(self, environ, start_response):
|
||
|
|
self.ws = Server(environ)
|
||
|
|
return self.app(self)
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
return self.ws.close()
|
||
|
|
|
||
|
|
def send(self, message):
|
||
|
|
try:
|
||
|
|
return self.ws.send(message)
|
||
|
|
except ConnectionClosed:
|
||
|
|
raise IOError()
|
||
|
|
|
||
|
|
def wait(self):
|
||
|
|
try:
|
||
|
|
return self.ws.receive()
|
||
|
|
except ConnectionClosed:
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
_async = {
|
||
|
|
'thread': threading.Thread,
|
||
|
|
'queue': queue.Queue,
|
||
|
|
'queue_empty': queue.Empty,
|
||
|
|
'event': threading.Event,
|
||
|
|
'websocket': WebSocketWSGI if _websocket_available else None,
|
||
|
|
'sleep': time.sleep,
|
||
|
|
}
|