Move NotFoundError into the lookup method

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

View File

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