Verwenden von Django auth UserAdmin für ein benutzerdefiniertes Benutzermodell

Aus den Django.Contrib.Auth-Dokumenten :

Erweitern des Standardbenutzers von Django Wenn Sie mit dem Benutzermodell von Django vollkommen zufrieden sind und nur einige zusätzliche Profilinformationen hinzufügen möchten, können Sie einfach django.contrib.auth.models.AbstractUserIhre benutzerdefinierten Profilfelder unterordnen und hinzufügen. Diese Klasse bietet die vollständige Implementierung des Standardbenutzers als abstraktes Modell.

Gesagt, getan. Ich habe ein neues Modell wie folgt erstellt:

class MyUser(AbstractUser):
  some_extra_data = models.CharField(max_length=100, blank=True)

Dies zeigt sich in admin fast wie Djangos Standard User. Der wichtigste Unterschied in admin besteht jedoch darin, dass das Feld zum (erneuten) Festlegen des Kennworts nicht vorhanden ist, sondern stattdessen ein normales CharField angezeigt wird. Muss ich wirklich Dinge in der Admin-Konfiguration überschreiben, damit dies funktioniert? Wenn ja, wie kann ich das auf etwas TROCKENE Weise tun (dh ohne Sachen aus der Django-Quelle zu kopieren ... eww ...)?

Nachdem ich mich eine Weile im Django-Quellcode umgesehen hatte, fand ich eine funktionierende Lösung. Ich bin mit dieser Lösung nicht ganz zufrieden, aber sie scheint zu funktionieren. Fühlen Sie sich frei, bessere Lösungen vorzuschlagen!

Django verwendet UserAdmin, um den netten Admin-Look für das UserModell zu rendern . admin.pyWenn wir dies nur in unserer Datei verwenden, können wir das gleiche Aussehen für unser Modell erhalten.

from django.contrib.auth.admin import UserAdmin
admin.site.register(MyUser, UserAdmin)

Dies allein ist jedoch wahrscheinlich keine gute Lösung, da Django Admin keines Ihrer speziellen Felder anzeigt. Dafür gibt es zwei Gründe:

• UserAdminwird UserChangeFormals Formular verwendet, das beim Ändern des Objekts verwendet werden soll, das wiederum Userals Modell verwendet wird.
• UserAdminformsetsDefiniert eine Eigenschaft, die später von verwendet wird UserChangeFormund Ihre speziellen Felder nicht enthält.

Also habe ich ein spezielles Änderungsformular erstellt, das die Meta-Innenklasse überlastet, sodass das Änderungsformular das richtige Modell verwendet. Ich musste auch überladen UserAdmin, um meine speziellen Felder zum Feldsatz hinzuzufügen. Dies ist der Teil dieser Lösung, den ich ein bisschen nicht mag, da sie ein bisschen hässlich aussieht. Fühlen Sie sich frei, Verbesserungen vorzuschlagen!

from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.forms import UserChangeForm

class MyUserChangeForm(UserChangeForm):
    class Meta(UserChangeForm.Meta):
        model = MyUser

class MyUserAdmin(UserAdmin):
    form = MyUserChangeForm

    fieldsets = UserAdmin.fieldsets + (
            (None, {'fields': ('some_extra_data',)}),
    )


admin.site.register(MyUser, MyUserAdmin)

Die Antwort von nico war äußerst hilfreich, aber ich fand, dass Django beim Erstellen eines neuen Benutzers immer noch auf das Benutzermodell verweist.

Ticket Nr. 19353 verweist auf dieses Problem.

Um das Problem zu beheben, musste ich noch einige Ergänzungen vornehmen admin.py

admin.py:

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.forms import UserChangeForm, UserCreationForm
from main.models import MyUser
from django import forms


class MyUserChangeForm(UserChangeForm):
    class Meta(UserChangeForm.Meta):
        model = MyUser


class MyUserCreationForm(UserCreationForm):
    class Meta(UserCreationForm.Meta):
        model = MyUser

    def clean_username(self):
        username = self.cleaned_data['username']
        try:
            MyUser.objects.get(username=username)
        except MyUser.DoesNotExist:
            return username
        raise forms.ValidationError(self.error_messages['duplicate_username'])


class MyUserAdmin(UserAdmin):
    form = MyUserChangeForm
    add_form = MyUserCreationForm
    fieldsets = UserAdmin.fieldsets + (
        (None, {'fields': ('extra_field1', 'extra_field2',)}),
    )

admin.site.register(MyUser, MyUserAdmin)

Eine einfachere Lösung, admin.py:

from django.contrib.auth.admin import UserAdmin
from main.models import MyUser

class MyUserAdmin(UserAdmin):
    model = MyUser

    fieldsets = UserAdmin.fieldsets + (
            (None, {'fields': ('some_extra_data',)}),
    )

admin.site.register(MyUser, MyUserAdmin)

Django verweist beim Erstellen und Ändern korrekt auf das MyUser-Modell. Ich benutze Django 1.6.2.

Die Antwort von cesc funktionierte nicht für mich, als ich versuchte, dem Erstellungsformular ein benutzerdefiniertes Feld hinzuzufügen. Vielleicht hat es sich seit 1.6.2 geändert? In beiden Fällen habe ich festgestellt, dass das Hinzufügen des Felds zu beiden Feldsätzen und add_fieldsets den Trick ausgeführt haben.

ADDITIONAL_USER_FIELDS = (
    (None, {'fields': ('some_additional_field',)}),
)

class MyUserAdmin(UserAdmin):
    model = MyUser

    add_fieldsets = UserAdmin.add_fieldsets + ADDITIONAL_USER_FIELDS
    fieldsets = UserAdmin.fieldsets + ADDITIONAL_USER_FIELDS

admin.site.register(MyUser, MyUserAdmin)

Eine andere ähnliche Lösung ( von hier übernommen ):

from __future__ import unicode_literals

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import AbstractUser
from django.utils.translation import ugettext_lazy as _

from .models import User


class UserAdminWithExtraFields(UserAdmin):

    def __init__(self, *args, **kwargs):
        super(UserAdminWithExtraFields, self).__init__(*args, **kwargs)

        abstract_fields = [field.name for field in AbstractUser._meta.fields]
        user_fields = [field.name for field in self.model._meta.fields]

        self.fieldsets += (
            (_('Extra fields'), {
                'fields': [
                    f for f in user_fields if (
                        f not in abstract_fields and
                        f != self.model._meta.pk.name
                    )
                ],
            }),
        )


admin.site.register(User, UserAdminWithExtraFields)