app_websocket.py

#!/usr/bin/python
#  -*- coding: utf-8 -*-
import logging
import tornado.escape
import tornado.ioloop
import tornado.options
import tornado.web
import tornado.websocket
import os.path
import uuid

from tornado.options import define, options

define("port", default=8889, help="run on the given port", type=int)


class Application(tornado.web.Application):
    def __init__(self):
        handlers = [
            (r"/", MainHandler),
            (r"/chatsocket", ChatSocketHandler)
        ]
        settings = dict(
            cookie_secret='insert_good_cookie',
            template_path=os.path.join(os.path.dirname(__file__), "templates"),
            static_path=os.path.join(os.path.dirname(__file__), "static"),
            xsrf_cookies=True,
            debug=True,
        )
        tornado.web.Application.__init__(self, handlers, **settings)


class MainHandler(tornado.web.RequestHandler):
    def get(self):
        self.render('index.html', messages=ChatSocketHandler.cache)


class ChatSocketHandler(tornado.websocket.WebSocketHandler):
    waiters = set()
    cache = []
    cache_size = 100

    # def get_compression_options(self):
    #     return {}

    def open(self):
        """
        Выполняется в момент соединения клиента и сервера
        в данном случае в момент загрузки web страницы.
        """
        logging.info('websocket client +1')
        ChatSocketHandler.waiters.add(self)

    def on_close(self):
        """ Выполняется в момент закрытия или обновления web страницы. """
        logging.info('websocket client -1')
        ChatSocketHandler.waiters.remove(self)

    @classmethod
    def send_updates(cls, chat):
        """ Всем клиентам, по порядку, отдаются сообщения """
        logging.info("sending message to %d waiters", len(cls.waiters))
        for waiter in cls.waiters:
            try:
                waiter.write_message(chat)
            except:
                logging.error('Error sending message', exc_info=True)

    @classmethod
    def update_cache(cls, chat):
        """
        Псевдо хранение сообщений в оперативной памяти
        с пулом в 100 последних сообщений
        """
        cls.cache.append(chat)
        if len(cls.cache) > cls.cache_size:
            cls.cache = cls.cache[-cls.cache_size:]

    def on_message(self, message):
        """ Выполняется когда приходит сообщение от клиента """
        logging.info('got message %r', message)
        parsed = tornado.escape.json_decode(message)
        chat = {
            "id": str(uuid.uuid4()),
            "body": parsed['body'],
        }
        chat['html'] = tornado.escape.to_basestring(
            self.render_string('message.html', message=chat)
        )
        # put it in cache and give to listeners
        ChatSocketHandler.update_cache(chat)
        ChatSocketHandler.send_updates(chat)


def main():
    logging.info("Server started")
    tornado.options.parse_command_line()
    app = Application()
    app.listen(options.port)
    tornado.ioloop.IOLoop.instance().start()


if __name__ == '__main__':
    main()