def get(self, user_id):
auth_user = self.auth.get_user_by_session()
user_auth_key = ndb.Key('UserModel', auth_user['user_id'])
user_key = ndb.Key('UserModel', int(user_id))
query = ndb.OR(ndb.AND(MessageModel.sender == user_auth_key, MessageModel.receiver == user_key),
ndb.AND(MessageModel.sender == user_key, MessageModel.receiver == user_auth_key))
messages = MessageModel.query(query).order(-MessageModel.datetime).fetch()
return Response(messages)
def get(self):
max = 500
query = MessageModel.query().order(-MessageModel.date)
messages = query.fetch(max)
aaData = []
for message in messages:
aaData.append({
"ip": message.ip,
"date": message.date.isoformat(),
"name": message.name,
"city": message.city,
"message": message.message,
"email": message.email,
"id": message.key.id()
})
return {'aaData': aaData}
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}
