You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
950 B
Python
35 lines
950 B
Python
5 years ago
|
from gevent.pywsgi import WSGIServer
|
||
|
|
|
||
|
|
from .handler import WebSocketHandler
|
||
|
|
from .logging import create_logger
|
||
|
|
|
||
|
|
|
||
|
|
class WebSocketServer(WSGIServer):
|
||
|
|
handler_class = WebSocketHandler
|
||
|
|
debug_log_format = (
|
||
|
|
'-' * 80 + '\n' +
|
||
|
|
'%(levelname)s in %(module)s [%(pathname)s:%(lineno)d]:\n' +
|
||
|
|
'%(message)s\n' +
|
||
|
|
'-' * 80
|
||
|
|
)
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
self.debug = kwargs.pop('debug', False)
|
||
|
|
self.pre_start_hook = kwargs.pop('pre_start_hook', None)
|
||
|
|
self._logger = None
|
||
|
|
self.clients = {}
|
||
|
|
|
||
|
|
super(WebSocketServer, self).__init__(*args, **kwargs)
|
||
|
|
|
||
|
|
def handle(self, socket, address):
|
||
|
|
handler = self.handler_class(socket, address, self)
|
||
|
|
handler.handle()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def logger(self):
|
||
|
|
if not self._logger:
|
||
|
|
self._logger = create_logger(
|
||
|
|
__name__, self.debug, self.debug_log_format)
|
||
|
|
|
||
|
|
return self._logger
|