def get(self):
message_id = request.values.get('id', None)
if message_id is None:
abort(400, message="Bad Request. Message 'id' was not provided.")
message = MessageModel.get_by_id(int(message_id))
if message is None:
abort(404, message="Message '%s' was not found." % message_id)
return {'message': message.to_dict()}
def delete(self):
message_id = request.values.get('id', None)
if message_id is None:
abort(400, message="Please provide a message id.")
try:
id = int(message_id)
messages = MessageModel.get_by_id(id)
except ValueError:
if message_id == 'all':
messages = MessageModel.query().fetch(500)
else:
abort(400, message="Id should be a number.")
if messages is None:
abort(404, message="Message '%s' was not found." % message_id)
if type(messages) is list:
for message in messages:
message.key.delete()
else:
messages.key.delete()
return {'message': "Message '%s' was deleted successfully." % message_id}
