zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-16 09:37:03 +02:00
Code Issues Projects Releases Packages Wiki Activity

Clean up group display string

Browse source
This commit is contained in:
Rachid Mrad 2023-09-29 18:52:59 -04:00
parent ee11c100a2
commit b4506d0157
No known key found for this signature in database
GPG key ID: EF38E4CEC4A8F3CF
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/registrar/admin.py
View file

@ -135,13 +135,18 @@ class MyUserAdmin(BaseUserAdmin):
"email",
"first_name",
"last_name",
"first_group",
"group",
"status",
)
# First group (which should in theory be the ONLY group)
def first_group(self, obj):
return f"{obj.groups.first()}"
# Let's define First group
# (which should in theory be the ONLY group)
def group(self, obj):
if f"{obj.groups.first()}" == "full_access_group":
return "Super User"
elif f"{obj.groups.first()}" == "cisa_analysts_group":
return "Analyst"
return ""
fieldsets = (
(
@ -215,7 +220,7 @@ class MyUserAdmin(BaseUserAdmin):
"email",
"first_name",
"last_name",
"first_group",
"group",
"status",
)