Replace the default function decorators with new socketio specific ones

This commit is contained in:
Patrick Jentsch
2020-03-26 16:14:09 +01:00
parent 257600a2a8
commit eba6af3b1e
3 changed files with 38 additions and 24 deletions

View File

@ -1,9 +1,10 @@
from app import db, logger, socketio
from app.decorators import socketio_login_required
from app.events import connected_sessions
from app.models import Corpus, User
from .cqi import CQiClient
from flask import current_app, request
from flask_login import current_user, login_required
from flask_login import current_user
import math
@ -23,13 +24,13 @@ corpus_analysis_clients_status = {}
@socketio.on('send_analysis_status_cli')
@login_required
@socketio_login_required
def update_status(response):
update_analysis_status(response['status'])
@socketio.on('corpus_analysis_init')
@login_required
@socketio_login_required
def init_corpus_analysis(corpus_id):
logger.warning('Initiating corpus analysis.')
socketio.start_background_task(corpus_analysis_session_handler,
@ -39,7 +40,7 @@ def init_corpus_analysis(corpus_id):
@socketio.on('corpus_analysis_query')
@login_required
@socketio_login_required
def corpus_analysis_query(query):
update_analysis_status('running')
logger.warning('Recieved a query.')
@ -113,7 +114,7 @@ def get_analysis_status():
@socketio.on('inspect_match')
@login_required
@socketio_login_required
def inspect_match(message):
client = corpus_analysis_clients.get(request.sid)
if client is None: