mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-11-15 09:15:41 +00:00
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from app import db, socketio
|
|
from app.events import connected_sessions
|
|
from app.models import Corpus
|
|
from flask import current_app, request
|
|
from flask_login import login_required
|
|
from .CQiClient.CQiClient import CQiClient
|
|
import logging
|
|
|
|
|
|
@socketio.on('init_corpus_analysis')
|
|
@login_required
|
|
def init_corpus_analysis(corpus_id):
|
|
logger = logging.getLogger(__name__)
|
|
logger.warning('init_corpus_analysis')
|
|
''' TODO: Check if current_user is allowed to subscribe to this '''
|
|
socketio.start_background_task(observe_corpus_analysis_connection,
|
|
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 observe_corpus_analysis_connection(app, corpus_id, session_id):
|
|
logger = logging.getLogger(__name__)
|
|
corpus = Corpus.query.filter_by(id=corpus_id).first()
|
|
with app.app_context():
|
|
cqi_client = CQiClient(host=corpus.analysis_container_ip)
|
|
cqi_client.ctrl_connect('opaque', 'opaque')
|
|
while session_id in connected_sessions:
|
|
logger.warning(cqi_client.ctrl_ping())
|
|
logger.warning('Run container, run!')
|
|
socketio.sleep(3)
|
|
corpus.status = 'stop analysis'
|
|
db.session.commit()
|
|
logger.warning('Stop container, stop!')
|