Update src/registrar/models/user.py

Co-authored-by: Matt-Spence <matthewswSpence@gmail.com>
This commit is contained in:
zandercymatics 2024-08-27 12:27:35 -06:00 committed by GitHub
parent bef49a85c1
commit 89c2e4543a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -206,12 +206,11 @@ class User(AbstractUser):
if not portfolio:
return False
portfolio_perms = self.portfolio_permissions.filter(portfolio=portfolio, user=self).first()
if not portfolio_perms:
user_portfolio_perms = self.portfolio_permissions.filter(portfolio=portfolio, user=self).first()
if not user_portfolio_perms:
return False
portfolio_permissions = portfolio_perms._get_portfolio_permissions()
return portfolio_permission in portfolio_permissions
return portfolio_permission in user_portfolio_perms._get_portfolio_permissions()
def has_base_portfolio_permission(self, portfolio):
return self._has_portfolio_permission(portfolio, UserPortfolioPermissionChoices.VIEW_PORTFOLIO)