Usando Django auth UserAdmin para un modelo de usuario personalizado

De los documentos de Django.Contrib.Auth :

Ampliación del usuario predeterminado de Django Si está completamente satisfecho con el modelo de usuario de Django y solo desea agregar información de perfil adicional, simplemente puede django.contrib.auth.models.AbstractUsercrear una subclase y agregar sus campos de perfil personalizados. Esta clase proporciona la implementación completa del usuario predeterminado como modelo abstracto.

Dicho y hecho. Creé un nuevo modelo como el siguiente:

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

Esto se muestra en admin casi como el estándar de Django User. Sin embargo, la diferencia más importante en admin es que el campo de (re) establecer contraseña no está presente, sino que se muestra un CharField normal en su lugar. ¿Realmente tengo que anular cosas en la configuración de administración para que esto funcione? Si es así, ¿cómo puedo hacer eso de una manera algo SECA (es decir, sin copiar cosas de la fuente de Django ... eww ...)?

Después de investigar el código fuente de Django por un tiempo, encontré un alma funcional. No estoy totalmente satisfecho con esta solución, pero parece funcionar. ¡No dude en sugerir mejores soluciones!

Django usa UserAdminpara renderizar el bonito Usermodelo de apariencia de administrador . Con solo usar esto en nuestro admin.py-file, podemos obtener el mismo aspecto para nuestro modelo.

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

Sin embargo, esto por sí solo probablemente no sea una buena solución, ya que Django Admin no mostrará ninguno de sus campos especiales. Hay dos razones para esto:

• UserAdminutiliza UserChangeFormcomo formulario a utilizar al modificar el objeto, que a su vez utiliza Usercomo modelo.
• UserAdmindefine una formsetspropiedad -propiedad, utilizada posteriormente por UserChangeForm, que no incluye sus campos especiales.

Entonces, creé un formulario de cambio especial que sobrecarga la clase interna Meta para que el formulario de cambio use el modelo correcto. También tuve que sobrecargarme UserAdminpara agregar mis campos especiales al conjunto de campos, que es la parte de esta solución que no me gusta un poco, ya que se ve un poco fea. ¡No dude en sugerir mejoras!

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)

La respuesta de nico ha sido extremadamente útil, pero descubrí que Django todavía hace referencia al modelo de usuario al crear un nuevo usuario.

El ticket # 19353 hace referencia a este problema.

Para solucionarlo, tuve que hacer algunas adiciones más a 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)

Una solución más simple, 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 hará referencia correctamente al modelo MyUser para su creación y modificación. Estoy usando Django 1.6.2.

La respuesta de cesc no me funcionaba cuando intenté agregar un campo personalizado al formulario de creación. ¿Quizás ha cambiado desde 1.6.2? De cualquier manera, encontré que agregar el campo a ambos conjuntos de campos y add_fieldsets hizo el truco.

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)

Otra solución similar ( Tomó de aquí ):

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)

Solo haz lo siguiente

from django.contrib import admin
from .models import MyUser

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

Eso es. Se mostrarán todos los campos del modelo MyUser. La lista de campos de UserAdmin para mostrar está codificada (consulte su código fuente), pero ModelAdmin no. Es flexible