def put(self, note_id, user_id): note = session.query(Note).get(note_id) if note is None: return Message.instance_not_exist() user = session.query(User).get(user_id) if user is None: return Message.instance_not_exist() is_moderator = False for moderator in note.users: if moderator.id == auth.current_user().id: is_moderator = True if note.owner_id != auth.current_user().id and not is_moderator: return Message.auth_failed() max_moderators = 5 if note.number_of_moderators >= max_moderators: return Message.message('Can not add a moderator', 400) note.number_of_moderators += 1 note.users.append(user) session.commit() return Message.successful('add moderator to note')
def delete(self, tag_id): tag = session.query(Tag).get(tag_id) if tag is None: return Message.instance_not_exist() session.delete(tag) session.commit() return Message.successful('deleted')
def post(self): request_data = request.get_json() try: name = request_data['name'] text = request_data['text'] if not isinstance(auth.current_user(), User): raise Exception owner_id = auth.current_user().id except KeyError: return Message.value_error() except Exception: return Message.creation_error() if note_schema.validate(data=request_data, session=session): return Message.creation_error() note = Note( name=name, text=text, owner_id=owner_id) if note is None: return Message.creation_error() session.add(note) session.commit() return Message.successful('created', 201)
def put(self, note_id): request_data = request.get_json() note = session.query(Note).get(note_id) if note is None: return Message.instance_not_exist() is_moderator = False for moderator in note.users: if moderator.id == auth.current_user().id: is_moderator = True if note.owner_id != auth.current_user().id and not is_moderator: return Message.auth_failed() if 'name' in request_data: note.name = request_data['name'] if 'text' in request_data: note.text = request_data['text'] if 'tag_id' in request_data: tag_id = request_data['tag_id'] tag = session.query(Tag).get(tag_id) note.tags.append(tag) session.commit() return Message.successful('updated')
def test_delete_tag(self): tag = create_tag(self) response = self.client.delete(f"/api/tag/{tag.id}/") self.assertEqual((response.get_json(), 200), Message.successful('deleted'))
def test_create_user_succesful(self): response = self.client.post("/api/user/create/", data=json.dumps( dict(username="******", email="*****@*****.**", password="******")), content_type='application/json') self.assertEqual((response.get_json(), 201), Message.successful('created', 201))
def delete(self, user_id): user = session.query(User).get(user_id) if user is None: return Message.instance_not_exist() if user.email != auth.current_user().email: return Message.auth_failed() session.delete(user) session.commit() return Message.successful('deleted')
def put(self, tag_id): request_data = request.get_json() tag = session.query(Tag).get(tag_id) if tag is None: return Message.instance_not_exist() if 'name' in request_data: tag.name = request_data['name'] session.commit() return Message.successful('updated')
def delete(self, note_id): note = session.query(Note).get(note_id) if note is None: return Message.instance_not_exist() if note.owner_id != auth.current_user().id: return Message.auth_failed() session.delete(note) session.commit() return Message.successful('deleted')
def test_delete_note(self): note = create_note(self) response = self.client.delete( f"/api/note/{note.id}/", headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 200), Message.successful('deleted'))
def test_create_tag_succesful(self): response = self.client.post( "/api/tag/", data=json.dumps(dict(name="tag1", )), content_type='application/json', headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 201), Message.successful('created', 201))
def test_put_tag(self): tag = create_tag(self) response = self.client.put( f"/api/tag/{tag.id}/", data=json.dumps(dict(name="tag1_edited")), content_type='application/json', headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 200), Message.successful('updated'))
def test_note_tag(self): tag = create_tag(self) note = create_note(self) response = self.client.put( f"/api/add/note/{note.id}/tag/{tag.id}/", content_type='application/json', headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 200), Message.successful('add tag to note'))
def put(self, note_id, tag_id): note = session.query(Note).get(note_id) if note is None: return Message.instance_not_exist() if note.owner_id != auth.current_user().id: return Message.auth_failed() tag = session.query(Tag).get(tag_id) if tag is None: return Message.instance_not_exist() note.tags.append(tag) session.commit() return Message.successful('add tag to note')
def put(self, user_id): request_data = request.get_json() user = session.query(User).get(user_id) if user is None: return Message.instance_not_exist() if user.email != auth.current_user().email: return Message.auth_failed() if 'username' in request_data: user.username = request_data['username'] if 'email' in request_data: user.email = request_data['email'] if 'password' in request_data: user.password = request_data["password"] session.commit() return Message.successful('updated')
def test_create_note_succesful(self): user = create_user(self) response = self.client.post( "/api/note/", data=json.dumps( dict(name="note1", text="note1 blablablablabla", owner_id=user.id)), content_type='application/json', headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 201), Message.successful('created', 201))
def test_note_user(self): user = User(username=f"user{11}", email=f"user{11}@test.com", password=f"user{11}test") self.session.add(user) self.session.commit() note = create_note(self) response = self.client.put( f"/api/add/note/{note.id}/user/{user.id}/", content_type='application/json', headers={ 'Authorization': 'Basic ' + base64.b64encode(b'[email protected]:user1test').decode('utf-8') }) self.assertEqual((response.get_json(), 200), Message.successful('add moderator to note'))
def post(self): request_data = request.get_json() try: username = request_data['username'] email = request_data['email'] password = request_data['password'] except KeyError: return Message.value_error() user = User(username=username, email=email, password=password) if user is None: return Message.creation_error() session.add(user) try: session.commit() except IntegrityError: session.rollback() return Message.creation_error() return Message.successful('created', 201)
def post(self): request_data = request.get_json() try: name = request_data['name'] except KeyError: return Message.value_error() if tag_schema.validate(data=request_data, session=session): return Message.value_error() tag = Tag(name=name) if tag is None: return Message.creation_error() session.add(tag) try: session.commit() except IntegrityError: session.rollback() return Message.creation_error() return Message.successful('created', 201)