__init__.py 572 B

123456789101112131415161718192021222324
  1. from __future__ import absolute_import
  2. import tornado.websocket
  3. class BaseWebSocketHandler(tornado.websocket.WebSocketHandler):
  4. # listeners = [], should be created in derived class
  5. def open(self):
  6. listeners = self.listeners
  7. listeners.append(self)
  8. def on_message(self, message):
  9. pass
  10. def on_close(self):
  11. listeners = self.listeners
  12. if self in listeners:
  13. listeners.remove(self)
  14. @classmethod
  15. def send_message(cls, message):
  16. for l in cls.listeners:
  17. l.write_message(message)