zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-17 23:14:11 +02:00
Code Issues Projects Releases Packages Wiki Activity

Replace inline if with .get

Browse source 
Not sure why I was doing that
This commit is contained in:
zandercymatics 2024-08-21 08:46:47 -06:00
parent 4f4dac2728
commit 729ce4c9f7
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
5 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/registrar/models/user.py
View file

@ -385,12 +385,12 @@ class User(AbstractUser):
def is_org_user(self, request):
has_organization_feature_flag = flag_is_active(request, "organization_feature")
portfolio = request.session["portfolio"] if "portfolio" in request.session else None
portfolio = request.session.get("portfolio")
return has_organization_feature_flag and self.has_base_portfolio_permission(portfolio)
def get_user_domain_ids(self, request):
"""Returns either the domains ids associated with this user on UserDomainRole or Portfolio"""
portfolio = request.session["portfolio"] if "portfolio" in request.session else None
portfolio = request.session.get("portfolio")
if self.is_org_user(request) and self.has_view_all_domains_permission(portfolio):
return DomainInformation.objects.filter(portfolio=portfolio).values_list(
"domain_id", flat=True