feat: Support UserAdmin add_fieldsets

Refs #654
deschler · Oct 12, 2022 · d414cd3 · d414cd3
1 parent 6aa2ff0
commit d414cd3
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/modeltranslation/admin.py b/modeltranslation/admin.py
@@ -34,8 +34,14 @@ def _get_declared_fieldsets(self, request, obj=None):
         # Take custom modelform fields option into account
         if not self.fields and hasattr(self.form, '_meta') and self.form._meta.fields:
             self.fields = self.form._meta.fields
-        if self.fieldsets:
-            return self._patch_fieldsets(self.fieldsets)
+        # takes into account non-standard add_fieldsets attribute used by UserAdmin
+        fieldsets = (
+            self.add_fieldsets
+            if getattr(self, 'add_fieldsets', None) and obj is None
+            else self.fieldsets
+        )
+        if fieldsets:
+            return self._patch_fieldsets(fieldsets)
         elif self.fields:
             return [(None, {'fields': self.replace_orig_field(self.get_fields(request, obj))})]
         return None