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 get(self):
user_list = session.query(User).all()
if not user_list:
return Message.instance_not_exist()
return user_list_schema.dump(user_list), 200
def get(self, user_id):
user = session.query(User).get(user_id)
if user is None:
return Message.instance_not_exist()
user_note_list = user.notes
return note_list_schema.dump(user_note_list)
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 get(self):
note_list = session.query(Note).all()
if not note_list:
return Message.instance_not_exist()
return note_list_schema.dump(note_list), 200
def get(self, tag_id):
tag = session.query(Tag).get(tag_id)
if tag is None:
return Message.instance_not_exist()
return tag_schema.dump(tag), 200
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 get(self):
tag_list = session.query(Tag).all()
if not tag_list:
return Message.instance_not_exist()
return tag_list_schema.dump(tag_list), 200
def test_get_tag_404(self):
tag = create_tag(self)
response = self.client.get(f"/api/tag/{tag.id + 2}/")
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
def test_userNotesList_404(self):
user = create_user(self)
response = self.client.get(f"/api/user/{user.id+2}/notes/")
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
def test_get_note_404(self):
note = create_note(self)
response = self.client.get(f"/api/note/{note.id + 2}/")
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
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 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_tag_404(self):
tag = create_tag(self)
response = self.client.delete(
f"/api/tag/{tag.id + 2}/",
headers={
'Authorization':
'Basic ' +
base64.b64encode(b'[email protected]:user1test').decode('utf-8')
})
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
def test_put_tag_404(self):
tag = create_tag(self)
response = self.client.put(
f"/api/tag/{tag.id + 2}/",
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(), 404),
Message.instance_not_exist())
def test_note_404_tag(self):
tag = create_tag(self)
note = create_note(self)
response = self.client.put(
f"/api/add/note/{note.id+2}/tag/{tag.id}/",
content_type='application/json',
headers={
'Authorization':
'Basic ' +
base64.b64encode(b'[email protected]:user1test').decode('utf-8')
})
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
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_note_404_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+2}/user/{user.id}/",
content_type='application/json',
headers={
'Authorization':
'Basic ' +
base64.b64encode(b'[email protected]:user1test').decode('utf-8')
})
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
def test_tagList_404(self):
response = self.client.get(f"/api/tag/all/")
self.assertEqual((response.get_json(), 404),
Message.instance_not_exist())
def get(self, note_id):
note = session.query(Note).get(note_id)
if note is None:
return Message.instance_not_exist()
return note_schema.dump(note), 200
def verify_password(login, password):
user = session.query(User).filter(User.email == login).first()
if user is None:
return Message.instance_not_exist()
if user and check_password_hash(user.password, password):
return user
def get(self, user_id):
user = session.query(User).get(user_id)
if user is None:
return Message.instance_not_exist()
return user_schema.dump(user), 200
