zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-31 23:16:36 +02:00
Code Issues Projects Releases Packages Wiki Activity

Try to add domain counting logic

Browse source
This commit is contained in:
Rebecca Hsieh 2024-10-21 16:42:02 -07:00
parent e7a6eb30e0
commit 199f669fad
No known key found for this signature in database
3 changed files with 49 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -471,3 +471,11 @@ class User(AbstractUser):
return DomainRequest.objects.filter(portfolio=portfolio).values_list("id", flat=True)
else:
return UserDomainRole.objects.filter(user=self).values_list("id", flat=True)
# def get_user_domain_count(self, request):
# """Returns the count of domains associated with this user on UserDomainRole or Portfolio"""
# portfolio = request.session.get("portfolio")
# if self.is_org_user(request) and self.has_view_all_domains_portfolio_permission(portfolio):
# return DomainInformation.objects.filter(portfolio=portfolio).count()
# else:
# return UserDomainRole.objects.filter(user=self).count()