Fix bugs with email

This commit is contained in:
zandercymatics 2024-10-24 10:26:59 -06:00
parent d9ec108f58
commit 87d51c1002
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
4 changed files with 59 additions and 27 deletions

View file

@ -2,6 +2,8 @@ from django.db import models
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 .utility.time_stamped_model import TimeStampedModel
@ -131,6 +133,17 @@ class Portfolio(TimeStampedModel):
def get_federal_type(cls, federal_agency):
return federal_agency.federal_type if federal_agency else None
@property
def portfolio_admin_users(self):
"""Gets all users with the role organization_admin for this particular portfolio.
Returns a queryset of User."""
admin_ids = self.portfolio_users.filter(
roles__overlap=[
UserPortfolioRoleChoices.ORGANIZATION_ADMIN,
],
).values_list("user__id", flat=True)
return User.objects.filter(id__in=admin_ids)
# == Getters for domains == #
def get_domains(self, order_by=None):
"""Returns all DomainInformations associated with this portfolio"""