darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-30 06:21:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

[WIP] changed dao query to use .first and return specific not found

Browse Source 
error message from rest view method.
This commit is contained in:
Adam Shimali
2016-02-25 12:10:04 +00:00
parent 883ad02c25
commit 99c9cd608c
3 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/dao/invited_user_dao.py
View File

@@ -9,7 +9,7 @@ def save_invited_user(invited_user):
def get_invited_user(service_id, invited_user_id):
return InvitedUser.query.filter_by(service_id=service_id, id=invited_user_id).one()
return InvitedUser.query.filter_by(service_id=service_id, id=invited_user_id).first()
def get_invited_users_for_service(service_id):

4
app/invite/rest.py
View File

@@ -39,4 +39,8 @@ def get_invited_users_by_service(service_id):
@invite.route('/<invited_user_id>', methods=['GET'])
def get_invited_user_by_service_and_id(service_id, invited_user_id):
invited_user = get_invited_user(service_id, invited_user_id)
if not invited_user:
message = 'Invited user not found for service id: {} and invited user id: {}'.format(service_id,
invited_user_id)
return jsonify(result='error', message=message), 404
return jsonify(data=invited_user_schema.dump(invited_user).data), 200