Korzystanie z Django auth UserAdmin dla niestandardowego modelu użytkownika

Z dokumentacji Django.Contrib.Auth :

Rozszerzanie domyślnego użytkownika Django Jeśli jesteś całkowicie zadowolony z modelu użytkownika Django i chcesz tylko dodać dodatkowe informacje o profilu, możesz po prostu utworzyć podklasę django.contrib.auth.models.AbstractUseri dodać własne pola profilu. Ta klasa zapewnia pełną implementację domyślnego User jako modelu abstrakcyjnego.

Powiedziane i zrobione. Stworzyłem nowy model jak poniżej:

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

To pojawia się w panelu administracyjnym prawie tak, jak w standardzie Django User. Jednak najważniejszą różnicą w ustawieniach admin jest to, że nie ma pola ustawiania (ponownego) hasła, ale zamiast tego wyświetlany jest normalny CharField. Czy naprawdę muszę przesłonić rzeczy w admin-config, aby to zadziałało? Jeśli tak, jak mogę to zrobić w nieco SUCHY sposób (tj. Bez kopiowania rzeczy ze źródła Django… eww…)?

Po pewnym czasie przekopania się w kodzie źródłowym Django, znalazłem działającą duszę. Nie jestem do końca zadowolony z tego rozwiązania, ale wydaje się, że działa. Zapraszam do proponowania lepszych rozwiązań!

Django używa UserAdmindo renderowania ładnego administracyjnego wyglądu Usermodelu. Używając tego w naszym admin.py-file, możemy uzyskać ten sam wygląd naszego modelu.

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

Jednak samo to prawdopodobnie nie jest dobrym rozwiązaniem, ponieważ Django Admin nie wyświetli żadnego z twoich specjalnych pól. Są ku temu dwa powody:

• UserAdminużywa UserChangeFormjako formy używanej podczas modyfikowania obiektu, który z kolei używa Userjako swojego modelu.
• UserAdmindefiniuje formsets-właściwość, używaną później przez UserChangeForm, która nie obejmuje twoich specjalnych pól.

Dlatego stworzyłem specjalny formularz zmiany, który przeciąża wewnętrzną klasę Meta, tak że formularz zmiany używa poprawnego modelu. Musiałem też przeładować, UserAdminaby dodać moje pola specjalne do zestawu pól, co jest częścią tego rozwiązania, które trochę nie lubię, ponieważ wygląda trochę brzydko. Zapraszam do sugerowania ulepszeń!

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)

Odpowiedź nico była niezwykle pomocna, ale zauważyłem, że Django nadal odwołuje się do modelu użytkownika podczas tworzenia nowego użytkownika.

Bilet nr 19353 odnosi się do tego problemu.

Aby to naprawić, musiałem zrobić kilka dodatkowych dodatków 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)

Prostsze rozwiązanie, 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 będzie poprawnie odwoływać się do modelu MyUser podczas tworzenia i modyfikacji. Używam Django 1.6.2.

Odpowiedź cesc nie działała dla mnie, gdy próbowałem dodać niestandardowe pole do formularza tworzenia. Może to się zmieniło od wersji 1.6.2? Tak czy inaczej, stwierdziłem, że dodanie pola do obu zestawów pól i add_fieldsets załatwiło sprawę.

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)

Inne podobne rozwiązanie ( wzięte stąd ):

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)

Po prostu wykonaj następujące czynności

from django.contrib import admin
from .models import MyUser

admin.site.register(MyUser, admin.ModelAdmin)

Otóż ​​to. Zostaną wyświetlone wszystkie pola modelu MyUser. Lista pól UserAdmin do wyświetlenia jest zakodowana na stałe (zobacz kod źródłowy), ale ModelAdmin nie. Jest elastyczny.