diff --git a/app/decorators.py b/app/decorators.py index 0a361424cff705f7facde00ee0165585701837ab..5fa664e0e339f442b2c8a2e479342737bc98e603 100644 --- a/app/decorators.py +++ b/app/decorators.py @@ -1,4 +1,4 @@ -from flask import abort, current_app +from flask import abort, current_app, request from flask_login import current_user from flask_socketio import disconnect from functools import wraps @@ -39,15 +39,17 @@ def socketio_admin_required(f): if current_user.is_administrator: return f(*args, **kwargs) else: - disconnect() + response = {'code': 401, 'desc': 'Unauthorized'} + socketio.emit(request.event['message'], response, room=request.sid) return wrapped def socketio_login_required(f): @wraps(f) def wrapped(*args, **kwargs): - if not current_user.is_authenticated: - disconnect() - else: + if current_user.is_authenticated: return f(*args, **kwargs) + else: + response = {'code': 401, 'desc': 'Unauthorized'} + socketio.emit(request.event['message'], response, room=request.sid) return wrapped