diff --git a/app/__init__.py b/app/__init__.py index 8a6b4325b05f524077898a7d5ec0754e131f858f..e7a21381d94d8a46fd7e00fef8c93a3ab1d56236 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -32,8 +32,8 @@ def create_app(config_name): app, message_queue=app.config['NOPAQUE_SOCKETIO_MESSAGE_QUEUE_URI']) with app.app_context(): - from . import socketio_events - from . import sqlalchemy_events + from .events import socketio as socketio_events + from .events import sqlalchemy as sqlalchemy_events from .admin import admin as admin_blueprint from .auth import auth as auth_blueprint from .corpora import corpora as corpora_blueprint diff --git a/app/corpora/events.py b/app/corpora/events.py index e0a880620eaf82739578b812ba4f1f3937a46745..a420993b6129ee0d0943b078e721ff3910571c16 100644 --- a/app/corpora/events.py +++ b/app/corpora/events.py @@ -5,7 +5,7 @@ from socket import gaierror from werkzeug.utils import secure_filename from .. import db, socketio from ..decorators import socketio_login_required -from ..socketio_events import socketio_sessions +from ..events.socketio import sessions as socketio_sessions from ..models import Corpus import cqi import math diff --git a/app/socketio_events.py b/app/events/socketio.py similarity index 92% rename from app/socketio_events.py rename to app/events/socketio.py index 614668ad504e67c033fbbcf1739e85e2aff35aa7..539680f82f9d2f927e185365fb7ade1334699684 100644 --- a/app/socketio_events.py +++ b/app/events/socketio.py @@ -1,9 +1,9 @@ from flask import request from flask_login import current_user from flask_socketio import join_room, leave_room -from . import socketio -from .decorators import socketio_login_required -from .models import User +from .. import socketio +from ..decorators import socketio_login_required +from ..models import User ''' @@ -11,7 +11,7 @@ from .models import User ' of all connected sessions, which can be used to determine the runtimes of ' associated background tasks. ''' -socketio_sessions = [] +sessions = [] ############################################################################### @@ -24,7 +24,7 @@ def socketio_connect(): ' The Socket.IO module creates a session id (sid) for each request. ' On connect the sid is saved in the connected sessions list. ''' - socketio_sessions.append(request.sid) + sessions.append(request.sid) @socketio.on('disconnect') @@ -34,7 +34,7 @@ def socketio_disconnect(): ' list. ''' try: - socketio_sessions.remove(request.sid) + sessions.remove(request.sid) except ValueError: pass diff --git a/app/sqlalchemy_events.py b/app/events/sqlalchemy.py similarity index 98% rename from app/sqlalchemy_events.py rename to app/events/sqlalchemy.py index 54737a5220b1bd1cd62572dc4e208eb3ace9ef5b..256b64534c8bcc4472b803efcbb7304fc1187f95 100644 --- a/app/sqlalchemy_events.py +++ b/app/events/sqlalchemy.py @@ -1,6 +1,6 @@ from datetime import datetime -from . import db, socketio -from .models import Corpus, CorpusFile, Job, JobInput, JobResult +from .. import db, socketio +from ..models import Corpus, CorpusFile, Job, JobInput, JobResult ###############################################################################