def post(self, message_key_url): action = self.request.get('action') if action == 'delete': message = Message.get_message(message_key_url) if message is not None: if message.to_user.user_id() == users.get_current_user().user_id(): Message.delete(message_key_url) self.redirect("/inbox") else: self.error(403) else: self.error(404)
def test_model_message(self): """Test message""" contact_type = ContactType(name='test') contact_type.save() contact = Contact(name='test', contact_type=contact_type) contact.save() self.user = DjangoUser(username=self.username, password='') self.user.set_password(self.password) self.user.save() stream = MessageStream(name='test') stream.save() obj = Message(title='test', body='test', author=contact, stream=stream) obj.save() self.assertEquals('test', obj.title) self.assertNotEquals(obj.id, None) obj.delete()
def message_endpoint(): """ Submits the given message to the intended recipient. add to database. Args: Return: """ resp = None if request.method == 'POST': try: message = Message.post(request) session.add(message) session.commit() except IntegrityError as ie: app.logger.error("Data integrity error: {}".format(ie)) return make_response( jsonify(message='Resource not created.', error=ie), 500) except InvalidUsage: raise resp = make_response(jsonify(message=message.serialize), 201) resp.headers['Content-Type'] += '; charset=utf-8' resp.headers['Location'] = '/v1/messages/{}'.format(message.id) return resp elif request.method == 'GET': try: messages = Message.get(request) except (NotFound, InvalidUsage, NotAcceptable): raise resp = make_response(jsonify(messages), 200) resp.headers['Content-Type'] += '; charset=utf-8' return resp elif request.method == 'DELETE': try: delete_count = Message.delete(request) except (NotFound, InternalServerError): raise resp = make_response(jsonify(delete_count), 204) resp.headers['Content-Type'] += '; charset=utf-8' return resp
def delete_message(): Comment.delete().where( Comment.message_id == request.form['message_id']).execute() Message.delete().where(Message.id == request.form['message_id']).execute() return jsonify(True)