扩展Django用户模型并在admin上显示

Question

I use the latest django for an intranet project. Well, I followed the django documentation to extend my models:

from django.db import models
from django.contrib.auth.models import AbstractUser

class Employee(AbstractUser):
    DEPARTMENTS = (
        ('ARE', 'Area Manager'),
        ('IT', 'IT'),
        ('CAT', 'Category manager'),
        ('CON', 'Controling')
    )
    departement = models.CharField(max_length = 3, verbose_name = "Département", choices = DEPARTMENTS)

After that, I rewrited the admin.py:

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.models import User

from .models import Employee

# Define an inline admin descriptor for Employee model
# which acts a bit like a singleton
class EmployeeInline(admin.StackedInline):
    model = Employee
    can_delete = False
    verbose_name_plural = 'Employees'

# Define a new User admin
class UserAdmin(BaseUserAdmin):
    inlines = (EmployeeInline,)

# Re-register UserAdmin
# admin.site.register(User) # return error: django.contrib.admin.sites.NotRegistered: The model User is not registered
admin.site.unregister(User)
admin.site.register(User, UserAdmin)

As you can see on the 3 last line of my admin.py, if I register User model I have an error. If I comment my last lines

# Re-register UserAdmin
#admin.site.unregister(User)
admin.site.register(User, UserAdmin)

I haven't my User administration:

Answer 1

Your problem is that your project has a custom model user definition. The advised way to get the user class in this case is using get_user_model() from django.contrib.auth .