132 lines
4.9 KiB
Python
132 lines
4.9 KiB
Python
# Copyright (c) 2019. Tobias Kurze
|
|
"""
|
|
This module provides functions related to authentication through the API.
|
|
For example: listing of available auth providers or registration of users.
|
|
|
|
Login through API does not start a new session, but instead returns JWT.
|
|
"""
|
|
import logging
|
|
|
|
from flask_jwt_extended import jwt_required
|
|
from flask_restx import fields, Resource
|
|
from sqlalchemy import exc
|
|
|
|
from backend import db, app, LrcException
|
|
from backend.api import api_room
|
|
from backend.models.room_model import Room
|
|
from backend.models.recorder_model import Recorder
|
|
|
|
|
|
logger = logging.getLogger("lrc.api.room")
|
|
|
|
room_model = api_room.model('Room', {
|
|
'id': fields.String(required=False, description='The room\'s identifier'),
|
|
'created_at': fields.DateTime(required=False, description='Creation date of the room info'),
|
|
'name': fields.String(required=True, description='The room\'s name'),
|
|
'alternate_name': fields.String(required=False, description='The room\'s alternate name'),
|
|
'comment': fields.String(required=False, description='The room\'s comment'),
|
|
'number': fields.String(required=True, description='The room\'s number'),
|
|
'building_name': fields.String(required=False, description='The building\'s name'),
|
|
'building_number': fields.String(required=False, description='The building\'s number'),
|
|
'recorder': fields.Nested(api_room.model('room_recorder',
|
|
{'id': fields.Integer(), 'name': fields.String(),
|
|
'ip': fields.String(), 'network_name': fields.String()}),
|
|
allow_null=True,
|
|
skip_none=False,
|
|
required=False,
|
|
description='Room recorder.'),
|
|
|
|
})
|
|
|
|
|
|
@api_room.route('/<int:id>')
|
|
@api_room.response(404, 'Room not found')
|
|
@api_room.param('id', 'The room identifier')
|
|
class RoomResource(Resource):
|
|
@jwt_required
|
|
@api_room.doc('get_room')
|
|
@api_room.marshal_with(room_model, skip_none=False)
|
|
def get(self, id):
|
|
"""Fetch a user given its identifier"""
|
|
room = Room.query.get(id)
|
|
if room is not None:
|
|
return room
|
|
api_room.abort(404)
|
|
|
|
@jwt_required
|
|
@api_room.doc('delete_todo')
|
|
@api_room.response(204, 'Todo deleted')
|
|
def delete(self, id):
|
|
'''Delete a task given its identifier'''
|
|
room = Room.query.get(id)
|
|
if room is not None:
|
|
db.session.delete(room)
|
|
db.session.commit()
|
|
return '', 204
|
|
api_room.abort(404)
|
|
|
|
@jwt_required
|
|
@api_room.doc('update_room')
|
|
@api_room.expect(room_model)
|
|
def put(self, id):
|
|
app.logger.debug(api_room.payload)
|
|
'''Update a task given its identifier'''
|
|
if "recorder_id" in api_room.payload:
|
|
if api_room.payload["recorder_id"] is None:
|
|
api_room.payload["recorder"] = None
|
|
else:
|
|
recorder = Recorder.query.get(api_room.payload["recorder_id"])
|
|
if recorder is not None:
|
|
api_room.payload["recorder"] = recorder
|
|
else:
|
|
return "specified recorder (id: {}) does not exist!".format(api_room.payload["recorder_id"]), 404
|
|
room = Room.query.get(id)
|
|
if room is not None:
|
|
room.recorder = api_room.payload["recorder"]
|
|
else:
|
|
num_rows_matched = Room.query.filter_by(id=id).update(api_room.payload)
|
|
db.session.commit()
|
|
return "ok"
|
|
api_room.abort(404)
|
|
|
|
|
|
@api_room.route('')
|
|
class RoomList(Resource):
|
|
@jwt_required
|
|
@api_room.doc('rooms')
|
|
@api_room.marshal_list_with(room_model, skip_none=False)
|
|
def get(self):
|
|
"""
|
|
List all rooms
|
|
:return: rooms
|
|
"""
|
|
return Room.get_all()
|
|
|
|
@jwt_required
|
|
@api_room.doc('create_room')
|
|
@api_room.expect(room_model)
|
|
@api_room.marshal_with(room_model, skip_none=False, code=201)
|
|
def post(self):
|
|
if "recorder_id" in api_room.payload:
|
|
if api_room.payload["recorder_id"] is None:
|
|
api_room.payload["recorder"] = None
|
|
else:
|
|
recorder = Recorder.query.get(api_room.payload["recorder_id"])
|
|
if recorder is not None:
|
|
api_room.payload["recorder"] = recorder
|
|
else:
|
|
return "specified recorder (id: {}) does not exist!".format(api_room.payload["recorder_id"]), 404
|
|
del api_room.payload["recorder_id"]
|
|
try:
|
|
room = Room(**api_room.payload)
|
|
db.session.add(room)
|
|
try:
|
|
db.session.commit()
|
|
return room
|
|
except exc.IntegrityError as e:
|
|
db.session.rollback()
|
|
return str(e.detail), 400
|
|
except LrcException as e:
|
|
logger.error(e)
|
|
return api_room.abort(400, str(e))
|