sporada/nopaque
1
0
Fork 0
mirror of https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git synced 2025-11-04 12:22:47 +00:00
Code Issues Projects Releases Wiki Activity

Proper socketio_login/admin_required decorators

Browse Source
This commit is contained in:
Patrick Jentsch
2020-05-20 09:36:59 +02:00
parent f86301112b
commit 8d201a29ae
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/decorators.py
View File

@@ -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_login import current_user
from flask_socketio import disconnect from flask_socketio import disconnect
from functools import wraps from functools import wraps
@@ -39,15 +39,17 @@ def socketio_admin_required(f):
if current_user.is_administrator: if current_user.is_administrator:
return f(*args, **kwargs) return f(*args, **kwargs)
else: else:
disconnect() response = {'code': 401, 'desc': 'Unauthorized'}
socketio.emit(request.event['message'], response, room=request.sid)
return wrapped return wrapped
def socketio_login_required(f): def socketio_login_required(f):
@wraps(f) @wraps(f)
def wrapped(*args, **kwargs): def wrapped(*args, **kwargs):
if not current_user.is_authenticated: if current_user.is_authenticated:
disconnect()
else:
return f(*args, **kwargs) return f(*args, **kwargs)
else:
response = {'code': 401, 'desc': 'Unauthorized'}
socketio.emit(request.event['message'], response, room=request.sid)
return wrapped return wrapped