refactor group custom list_display

This commit is contained in:
Rachid Mrad 2023-09-29 18:57:24 -04:00
parent b4506d0157
commit 0c05518d61
No known key found for this signature in database
GPG key ID: EF38E4CEC4A8F3CF

View file

@ -142,9 +142,9 @@ class MyUserAdmin(BaseUserAdmin):
# 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":
if obj.groups.filter(name="full_access_group").exists():
return "Super User"
elif f"{obj.groups.first()}" == "cisa_analysts_group":
elif obj.groups.filter(name="cisa_analysts_group").exists():
return "Analyst"
return ""