Skip to content
Snippets Groups Projects
Commit 8d201a29 authored by Patrick Jentsch's avatar Patrick Jentsch
Browse files

Proper socketio_login/admin_required decorators

parent f8630111
No related branches found
No related tags found
No related merge requests found
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment