You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
2.3 KiB
55 lines
2.3 KiB
from socketio import Namespace as _Namespace
|
|
|
|
|
|
class Namespace(_Namespace):
|
|
def __init__(self, namespace=None):
|
|
super().__init__(namespace)
|
|
self.socketio = None
|
|
|
|
def _set_socketio(self, socketio):
|
|
self.socketio = socketio
|
|
|
|
def trigger_event(self, event, *args):
|
|
"""Dispatch an event to the proper handler method.
|
|
|
|
In the most common usage, this method is not overloaded by subclasses,
|
|
as it performs the routing of events to methods. However, this
|
|
method can be overridden if special dispatching rules are needed, or if
|
|
having a single method that catches all events is desired.
|
|
"""
|
|
handler_name = 'on_' + (event or '')
|
|
if not hasattr(self, handler_name):
|
|
# there is no handler for this event, so we ignore it
|
|
return
|
|
handler = getattr(self, handler_name)
|
|
try:
|
|
return self.socketio._handle_event(handler, event, self.namespace,
|
|
*args)
|
|
except TypeError:
|
|
if event == 'disconnect':
|
|
# legacy disconnect events do not have the reason argument
|
|
return self.socketio._handle_event(
|
|
handler, event, self.namespace, *args[:-1])
|
|
else:
|
|
raise
|
|
|
|
def emit(self, event, data=None, room=None, include_self=True,
|
|
namespace=None, callback=None):
|
|
"""Emit a custom event to one or more connected clients."""
|
|
return self.socketio.emit(event, data, room=room,
|
|
include_self=include_self,
|
|
namespace=namespace or self.namespace,
|
|
callback=callback)
|
|
|
|
def send(self, data, room=None, include_self=True, namespace=None,
|
|
callback=None):
|
|
"""Send a message to one or more connected clients."""
|
|
return self.socketio.send(data, room=room, include_self=include_self,
|
|
namespace=namespace or self.namespace,
|
|
callback=callback)
|
|
|
|
def close_room(self, room, namespace=None):
|
|
"""Close a room."""
|
|
return self.socketio.close_room(room=room,
|
|
namespace=namespace or self.namespace)
|