def setUp(self): from sonofjscompute import models self.app = create_app() self.client = self.app.test_client() models.create_redis(self.app, db=1) self.redis = models.redis self.addCleanup(self._teardown_redis)
from flask_sockets import Sockets, log_request from json import dumps from sonofjscompute import create_app from sonofjscompute.models import Task app = create_app() sockets = Sockets(app) @sockets.route('/mosaic/') def echo_socket(ws): ws.receive() task = Task.get_last() while task.is_processing(): result = task.get_task_result() ws.send(dumps({'url':result})) ws.close()