Организация кода в Tornado. WebSocket

import redis
import momoko
 
...
 
# БД для быстрой проверки вхождений
c = redis.StrictRedis()
 
# Пулл подключенных по вебсокету пользователей
client_dict = {}
 
...
 
class MessagesHandler(tornado.websocket.WebSocketHandler):
    def __init__(self, *args, **kwargs):
        super(MessagesHandler, self).__init__(*args, **kwargs)
 
    @property
    def db(self):
        return self.application.db
 
    def check_origin(self, origin):
        return True
 
    def open(self):
        # Получаю сессию из кук, получаю инфу о юзере, 
        # Создаю черный список и список диалогов в redis 
        # для дальнейшей быстрой проверки по вебсокету при отправке сообщений 
        
    @tornado.gen.coroutine
    def on_message(self, mess):
        # Все сообщения идут в json виде.
        # Валидация jsonschema
    
        if 'message' in received_json_data:
            # Беру инфу из json
            # Собщение может быть текстом, фотографией, локацией и еще несколько варинтов
 
            try:
                # Проверка, что юзер которому хотим отпраить сообщение онлайн по вебсокету
                if len(client_dict[recipient_id]) == 0:
                    self.write_message(to_json_error('Empty clients list. Write to administrator', ticket))
                    return
            except KeyError:
                # Recipient is OFFLINE
                # Сообщение записывается в Postgres после нескольких проверок 
                # (блеклист, соответствие диалогу и т.д.)
                response = yield self.send_to_DB_and_check(..., check=True)
                # ...
                return
 
            # Recipient is ONLINE
            # Проверка в redis (блеклист, соответствие диалогу и т.д.)
 
            # RDB
            # Запись сообщения в Postgres. Просто INSERT, так как провеки все сделаны
            response = yield self.send_to_DB_and_check(..., check=False)
            
            # Отсылка сообщения получателю по вебсокету
            for k, recipient_wsconnection in client_dict[recipient_id].items():
                recipient_wsconnection.write_message(...)
                
            # Отсылка сообщения что все ок отправителю
            self.write_message(...)
            
        # Другой тип сообщения для сигнализации что пользователь начал ввод сообщений
        elif 'typing' in received_json_data:
            # Тут проверок нет практически  
            # Но код занимает строчек 30
 
        # Сообщение для тестирование (эмуляция присылания сообщений от рандомных юзеров)            
        elif 'test' in received_json_data:
            # Тут где-то 170 строк кода
            
            # В сообщении присылается кол-во сообщений, кторое юзер хочет получить
            # Генерация требуемого кол-ва юзеров естественно с проверками на блокировку и т.д.
                # Проверка сущестует ли диалог, если нет то создается
            # Отправка сообщений юзерам с записью в БД
 
    # Проверка в Postgres  
    @tornado.gen.coroutine
    def send_to_DB_and_check(self, thread_id, sender_id, recipient_id, message, time, type_mes, check=False):
        if check == True:
            # Несколько SQL запросов
        else:
            # Один SQL запрос
        return True, data
        
    def on_close(self):
        # Для экономии места удаляем инфу из redis для отключающегося юзера
 
...

if 'message' ...

if 'typing' ...

if 'test' ...