扩展Django用户模型并在管理员上显示

Question

我使用最新的django进行内部网项目。好吧，我按照django文档来扩展我的模型：

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)

之后，我重写了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)

正如您在admin.py的最后一行所看到的那样，如果我注册用户模型，则会出错。如果我评论我的最后一行

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

我没有我的用户管理：

Answer 1

您的问题是您的项目具有自定义模型用户定义。在这种情况下获取用户类的建议方法是使用get_user_model()中的django.contrib.auth。