pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move NotFoundError into the lookup method

Browse Source
This commit is contained in:
leigh-mil 2019-03-15 14:33:38 -04:00
parent 9a97b75d71
commit 32a705d1d3
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/domain/invitations.py
View File

@ -111,6 +111,10 @@ class Invitations(object):
@classmethod @classmethod
def lookup_by_portfolio_and_user(cls, portfolio, user): def lookup_by_portfolio_and_user(cls, portfolio, user):
portfolio_role = PortfolioRoles.get(portfolio.id, user.id) portfolio_role = PortfolioRoles.get(portfolio.id, user.id)
if portfolio_role.latest_invitation is None:
raise NotFoundError("invitation")
return portfolio_role.latest_invitation return portfolio_role.latest_invitation
@classmethod @classmethod

3
atst/routes/portfolios/task_orders.py
View File

@ -126,9 +126,6 @@ def resend_invite(portfolio_id, task_order_id, form=None):
invitation = Invitations.lookup_by_portfolio_and_user(portfolio, officer) invitation = Invitations.lookup_by_portfolio_and_user(portfolio, officer)
if not invitation:
raise NotFoundError("invitation")
if invitation.status is not InvitationStatus.PENDING: if invitation.status is not InvitationStatus.PENDING:
raise NoAccessError("invitation") raise NoAccessError("invitation")