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

Fix linting

Browse source
This commit is contained in:
Erin Song 2025-01-23 11:48:07 -08:00
parent 607336406f
commit 0522138cc4
No known key found for this signature in database
2 changed files with 16 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/registrar/models/portfolio.py
View file

@ -4,6 +4,7 @@ from registrar.models.domain_request import DomainRequest
from registrar.models.federal_agency import FederalAgency
from registrar.models.user import User
from registrar.models.utility.portfolio_helper import UserPortfolioRoleChoices
from django.db.models import Q
from .utility.time_stamped_model import TimeStampedModel
@ -144,12 +145,20 @@ class Portfolio(TimeStampedModel):
).values_list("user__id", flat=True)
return User.objects.filter(id__in=admin_ids)
def portfolio_users_with_permissions(self, permissions=[]):
def portfolio_users_with_permissions(self, permissions=[], include_admin=False):
"""Gets all users with specified additional permissions for this particular portfolio.
Returns a queryset of User."""
portfolio_users = self.portfolio_users
if permissions:
portfolio_users = portfolio_users.filter(additional_permissions__overlap=permissions)
if include_admin:
portfolio_users = portfolio_users.filter(
Q(additional_permissions__overlap=permissions) |
Q(roles__overlap=[
UserPortfolioRoleChoices.ORGANIZATION_ADMIN,
]),
)
else:
portfolio_users = portfolio_users.filter(additional_permissions__overlap=permissions)
user_ids = portfolio_users.values_list("user__id", flat=True)
return User.objects.filter(id__in=user_ids)