def get(self, id):
operator = OperatorModel.find_by_id(id)
if operator:
return operator.json()
return {'message': "There is no operator with given id"}, 404
def delete(self, id):
operator = OperatorModel.find_by_id(id)
if operator:
operator.delete_from_db()
return {'message': 'Operator deleted'}
return {'message': 'There is no operator with this id'}, 404
def put(self, id):
operator = OperatorModel.find_by_id(id)
if operator:
data = self.parser.parse_args()
for attribute, value in data.items():
if value:
setattr(operator, attribute, value)
try:
operator.save_to_db()
except Exception:
return {'message': 'An error occured inserting an operator'}, \
500
return operator.json()
return {'message': 'There is no operator with this id'}, 404