nopaque/app/users/events.py

from flask_login import current_user
from flask_socketio import join_room, leave_room
from app import hashids, socketio
from app.decorators import socketio_login_required
from app.models import User


@socketio.on('GET /users/<user_id>')
@socketio_login_required
def get_user(user_hashid, backrefs=False, relationships=False):
    user_id = hashids.decode(user_hashid)
    user = User.query.get(user_id)
    if user is None:
        return {'status': 404, 'statusText': 'Not found'}
    if not (user == current_user or current_user.is_administrator()):
        return {'status': 403, 'statusText': 'Forbidden'}
    return {
        'body': user.to_json_serializeable(
            backrefs=backrefs,
            relationships=relationships
        ),
        'status': 200,
        'statusText': 'OK',
    }


@socketio.on('SUBSCRIBE /users/<user_id>')
@socketio_login_required
def subscribe_user(user_hashid):
    user_id = hashids.decode(user_hashid)
    user = User.query.get(user_id)
    if user is None:
        return {'status': 404, 'statusText': 'Not found'}
    if not (user == current_user or current_user.is_administrator()):
        return {'status': 403, 'statusText': 'Forbidden'}
    join_room(f'/users/{user.hashid}')
    return {'status': 200, 'statusText': 'OK'}


@socketio.on('UNSUBSCRIBE /users/<user_id>')
@socketio_login_required
def unsubscribe_user(user_hashid):
    user_id = hashids.decode(user_hashid)
    user = User.query.get(user_id)
    if user is None:
        return {'status': 404, 'statusText': 'Not found'}
    if not (user == current_user or current_user.is_administrator()):
        return {'status': 403, 'statusText': 'Forbidden'}
    leave_room(f'/users/{user.hashid}')
    return {'status': 200, 'statusText': 'OK'}


# @socketio.on('GET User')
# @socketio_login_required
# def n_get_user(user_hashid):
#     # This constructs a JSON response which can easily be converted to a Response object
#     # Ref: https://developer.mozilla.org/en-US/docs/Web/API/Response/Response
#     user_id = hashids.decode(user_hashid)
#     user = User.query.get(user_id)
#     if user is None:
#         return {'options': {'status': 404, 'statusText': 'Not found'}}
#     if not (user == current_user or current_user.is_administrator()):
#         return {'options': {'status': 403, 'statusText': 'Forbidden'}}
#     body = {
#         'id': user.hashid,
#         # ...
#         'relationships': {
#             'corpora': {corpus.hashid for corpus in user.corpora},
#             'jobs': [job.hashid for job in user.jobs]
#         }
#     }
#     return {
#         'body': user.to_json_serializable(),
#         'options': {
#             'status': 200,
#             'statusText': 'OK',
#             'headers': {'Content-Type: application/json'}
#         }
#     }