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

Fix logic for only admin

Browse source
This commit is contained in:
Rebecca Hsieh 2024-10-24 17:13:18 -07:00
parent 6bf34c8d3b
commit 9cc44d4d70
No known key found for this signature in database
1 changed files with 27 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -6,7 +6,7 @@ from django.db import models
from django.db.models import Q
from registrar.models import DomainInformation, UserDomainRole
from registrar.models.utility.portfolio_helper import UserPortfolioPermissionChoices
from registrar.models.utility.portfolio_helper import UserPortfolioPermissionChoices, UserPortfolioRoleChoices
from .domain_invitation import DomainInvitation
from .portfolio_invitation import PortfolioInvitation
@ -496,20 +496,38 @@ class User(AbstractUser):
return active_requests_count
# def is_only_admin_of_portfolio(self, portfolio):
# """Check if the user is the only admin of the given portfolio."""
# UserPortfolioPermission = apps.get_model("registrar", "UserPortfolioPermission")
# # Grab admin permission ability we want
# admin_permission = UserPortfolioPermissionChoices.EDIT_PORTFOLIO
# # Get all users with admin permission for this portfolio
# admins = UserPortfolioPermission.objects.filter(portfolio=portfolio, roles__contains=[admin_permission])
# # Check if there is more than one admin
# if admins.count() == 1 and admins.first().user == self:
# # The user is the only admin
# return True
# # There are other admins OR the user is not the only one
# return False
def is_only_admin_of_portfolio(self, portfolio):
"""Check if the user is the only admin of the given portfolio."""
UserPortfolioPermission = apps.get_model("registrar", "UserPortfolioPermission")
# Grab admin permission ability we want
admin_permission = UserPortfolioPermissionChoices.EDIT_PORTFOLIO
admin_permission = UserPortfolioRoleChoices.ORGANIZATION_ADMIN
# Get all users with admin permission for this portfolio
admins = UserPortfolioPermission.objects.filter(portfolio=portfolio, roles__contains=[admin_permission])
admin_count = admins.count()
# Check if there is more than one admin
if admins.count() == 1 and admins.first().user == self:
# The user is the only admin
return True
# There are other admins OR the user is not the only one
# Check if the current user is in the list of admins
if admin_count == 1 and admins.first().user == self:
return True # The user is the only admin
# If there are other admins or the user is not the only one
print(f"{self} is NOT the only admin for portfolio {portfolio}.")
return False