diff --git a/app/corpora/events.py b/app/corpora/events.py index 8ad98fd3..28d3bad7 100644 --- a/app/corpora/events.py +++ b/app/corpora/events.py @@ -5,7 +5,6 @@ from flask import current_app, request from flask_login import login_required from .CQiClient.CQiClient import CQiClient import logging -import time analysis_clients = {} @@ -15,7 +14,6 @@ analysis_clients = {} @login_required def init_corpus_analysis(corpus_id): ''' TODO: Check if current_user is allowed to subscribe to this ''' - time.sleep(5) # wait for IP of container only for dev socketio.start_background_task(observe_corpus_analysis_connection, current_app._get_current_object(), corpus_id, @@ -33,13 +31,17 @@ def observe_corpus_analysis_connection(app, corpus_id, session_id): logger = logging.getLogger(__name__) with app.app_context(): corpus = Corpus.query.filter_by(id=corpus_id).first() - logger.warning('MÖÖÖÖP') - analysis_client = CQiClient(host=corpus.analysis_container_ip) - analysis_client.ctrl_connect('opaque', 'opaque') + while corpus.status != 'analysing': + db.session.refresh(corpus) + socketio.sleep(3) + analysis_server = '{}_analysis_container{}'.format(corpus.creator.username, corpus.id) + analysis_client = CQiClient(host=analysis_server) analysis_clients[session_id] = analysis_client + analysis_client.ctrl_connect('opaque', 'opaque') while session_id in connected_sessions: logger.warning(analysis_client.ctrl_ping()) socketio.sleep(3) analysis_client.ctrl_bye() + analysis_clients.pop(session_id, None) corpus.status = 'stop analysis' db.session.commit()