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/') @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/') @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/') @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'} # } # }