Change logic of how connected sessions are identified.

app/corpora/__init__.py

@@ -2,5 +2,4 @@ from flask import Blueprint
 
 corpora = Blueprint('corpora', __name__)
 
-
-from . import views
+from . import events, views

app/corpora/events.py

@@ -0,0 +1,30 @@
+from app import socketio
+from app.events import connected_sessions
+from flask import current_app, request
+from flask_login import login_required
+import logging
+
+
+@socketio.on('init_corpus_analysis')
+@login_required
+def init_corpus_analysis(corpus_id):
+    ''' TODO: Check if current_user is allowed to subscribe to this '''
+    socketio.start_background_task(init_corpus_analysis_,
+                                   current_app._get_current_object(),
+                                   corpus_id,
+                                   request.sid)
+
+
+@socketio.on('query_event')
+def recv_query(message):
+    logger = logging.getLogger(__name__)
+    logger.warning(message)
+
+
+def init_corpus_analysis_(app, corpus_id, session_id):
+    logger = logging.getLogger(__name__)
+    with app.app_context():
+        while session_id in connected_sessions:
+            logger.warning('Run container, run!')
+            socketio.sleep(3)
+    logger.warning('Stop container, stop!')