move socketio decorators

This commit is contained in:
Patrick Jentsch
2024-05-04 14:55:05 +02:00
parent 882987ba68
commit e1cfd394fa
9 changed files with 44 additions and 70 deletions

View File

@ -1,12 +1,12 @@
from flask_login import current_user
from flask_socketio import join_room, leave_room
from app import hashids, socketio
from app.extensions.flask_socketio_extras import login_required
from app.decorators import socketio_login_required
from app.models import User
@socketio.on('GET /users/<user_id>')
@login_required
@socketio_login_required
def get_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.get(user_id)
@ -22,7 +22,7 @@ def get_user(user_hashid):
@socketio.on('SUBSCRIBE /users/<user_id>')
@login_required
@socketio_login_required
def subscribe_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.get(user_id)
@ -35,7 +35,7 @@ def subscribe_user(user_hashid):
@socketio.on('UNSUBSCRIBE /users/<user_id>')
@login_required
@socketio_login_required
def unsubscribe_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.get(user_id)

View File

@ -1,12 +1,12 @@
from flask_login import current_user
from flask_socketio import join_room
from app import hashids, socketio
from app.extensions.flask_socketio_extras import admin_required, login_required
from app.decorators import socketio_admin_required, socketio_login_required
from app.models import User
@socketio.on('GET /users')
@admin_required
@socketio_admin_required
def get_users():
users = User.query.filter_by().all()
return {
@ -20,14 +20,14 @@ def get_users():
@socketio.on('SUBSCRIBE /users')
@admin_required
@socketio_admin_required
def subscribe_users():
join_room('/users')
return {'options': {'status': 200, 'statusText': 'OK'}}
@socketio.on('GET /users/<user_id>')
@login_required
@socketio_login_required
def get_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.get(user_id)
@ -46,7 +46,7 @@ def get_user(user_hashid):
@socketio.on('SUBSCRIBE /users/<user_id>')
@login_required
@socketio_login_required
def subscribe_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.get(user_id)
@ -59,7 +59,7 @@ def subscribe_user(user_hashid):
@socketio.on('GET /public_users')
@login_required
@socketio_login_required
def get_public_users():
users = User.query.filter_by(is_public=True).all()
return {
@ -76,14 +76,14 @@ def get_public_users():
@socketio.on('SUBSCRIBE /users')
@admin_required
@socketio_admin_required
def subscribe_users():
join_room('/public_users')
return {'options': {'status': 200, 'statusText': 'OK'}}
@socketio.on('GET /public_users/<user_id>')
@login_required
@socketio_login_required
def get_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.filter_by(id=user_id, is_public=True).first()
@ -102,7 +102,7 @@ def get_user(user_hashid):
@socketio.on('SUBSCRIBE /public_users/<user_id>')
@login_required
@socketio_login_required
def subscribe_user(user_hashid):
user_id = hashids.decode(user_hashid)
user = User.query.filter_by(id=user_id, is_public=True).first()