Question

我正在尝试使用自己的管理员（add和change表单）创建自定义用户模型。我希望与原始用户管理员完全相同，只包含add和change表单中的其他字段。

有人可以帮忙吗？

models.py

from django.contrib.auth.models import User

class MyUser1(User):
    #add more fields specific to MyUser1

class MyUser2(User):
    #add more fields specific to MyUser2

admin.py

from django.contrib.auth.admin import UserAdmin

class MyUser1Admin(UserAdmin):
    # how can I add the "fields specific to MyUser1" both in the add
    # and the change forms?

admin.site.register(MyUser1, MyUser1Admin)


# same for MyUser2

Answer 1

我不得不扩展UserCreationForm和UserChangeForm：

admin.py

from django.contrib.auth.models import User
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
from django.contrib.auth.admin import UserAdmin
from django.contrib import admin
from django import forms

class MyUser1CreationForm(UserCreationForm):
    myuser1_field = forms.CharField()

    class Meta:
        model = User
        fields = ("username", "myuser1_field", "password1", "password2")

class MyUser1ChangeForm(UserChangeForm):
    myuser1_field = forms.CharField()

    class Meta:
        model = User

class MyUser1Admin(UserAdmin):

    form = MyUser1ChangeForm
    add_form = MyUser1CreationForm

    fieldsets = (
        (None, {'fields': ('username', 'password')}),
        (_('Personal info'), {'fields': ('myuser1_field', 'first_name', 'last_name', 'email')}),
        (_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
                                       'groups', 'user_permissions')}),
        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
    )


    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('username', 'myuser1_field', 'password1', 'password2')}
        ),
    )


admin.site.register(MyUser1, MyUser1Admin)

Answer 2

from django.contrib.auth.forms import UserChangeForm

class MyUser1AdminForm(UserChangeForm):

    class Meta:
        model = MyUser1

如何在扩展UserAdmin中添加字段

2 个答案: