Використання Django auth UserAdmin для користувацької моделі користувача

З документів Django.Contrib.Auth :

Розширення користувача за замовчуванням Django Якщо вас цілком влаштовує модель користувача Django, і ви просто хочете додати додаткову інформацію про профіль, ви можете просто зробити підклас django.contrib.auth.models.AbstractUserта додати власні поля профілю. Цей клас забезпечує повну реалізацію користувача за замовчуванням як абстрактну модель.

Сказано і зроблено. Я створив нову модель, як показано нижче:

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

Це відображається в адміністраторі майже як стандарт Django User. Однак найважливіша відмінність адміністратора полягає в тому, що поля для (пере) встановлення пароля немає, а замість цього відображається звичайний CharField. Чи справді мені доводиться перевизначати матеріали в адмін-конфігурі, щоб це працювало? Якщо так, то як я можу зробити це дещо СУХО (тобто без копіювання матеріалів із джерела Django ... eww ...)?

Покопавшись якийсь час навколо вихідного коду Django, я знайшов дієвий дух. Я не повністю задоволений цим рішенням, але, схоже, це працює. Не соромтеся пропонувати кращі рішення!

Django використовує, UserAdminщоб зробити приємний зовнішній вигляд Userмоделі. Просто використовуючи це у нашому admin.py-файлі, ми можемо отримати такий же вигляд для нашої моделі.

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

Однак одне лише це, мабуть, не є гарним рішенням, оскільки адміністратор Django не відображатиме жодного з ваших спеціальних полів. Цьому є дві причини:

• UserAdminвикористовує UserChangeFormяк форму, яка буде використана при зміні об’єкта, який, у свою чергу, використовує Userяк свою модель.
• UserAdminвизначає formsets-property, який пізніше використовується UserChangeForm, який не включає ваші спеціальні поля.

Отже, я створив спеціальну форму зміни, яка перевантажує внутрішній клас Meta, щоб форма зміни використовувала правильну модель. Мені також довелося перевантажити, UserAdminщоб додати свої спеціальні поля до набору полів, що є частиною цього рішення, яке мені трохи не подобається, оскільки це виглядає трохи потворно. Не соромтеся пропонувати вдосконалення!

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)

Відповідь nico була надзвичайно корисною, але я виявив, що Django все ще посилається на модель користувача при створенні нового користувача.

Квиток № 19353 стосується цієї проблеми.

Для того, щоб це виправити, мені довелося зробити ще кілька доповнень 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)

Більш просте рішення, 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 буде правильно посилатися на модель MyUser для створення та модифікації. Я використовую Django 1.6.2.

Відповідь cesc не спрацювала для мене, коли я спробував додати власне поле до форми створення. Можливо, це змінилося з 1.6.2? У будь-якому випадку, я виявив, що додавання поля до обох наборів полів і add_fieldsets зробили свою справу.

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)

Ще одне подібне рішення ( взяв звідси ):

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)

Просто зробіть наступне

from django.contrib import admin
from .models import MyUser

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

Це воно. Будуть показані всі поля моделі MyUser. Список полів UserAdmin для показу має жорсткий код (див. Його вихідний код), але ModelAdmin - ні. Це гнучко.