/ admin / login / Manager中的AttributeError不可用; “ auth.User”已交换为“ users.CustomUser”

时间:2019-05-11 11:16:27

标签: python django

我编辑默认用户模型以进行自定义,以便在Django的admin用户模块中添加自定义字段

我遵循以下步骤:

第1步:创建名为用户的新应用

第2步:在用户应用程序model.py文件中,添加以下代码

from django.contrib.auth.models import BaseUserManager,AbstractBaseUser
from django.db import models


class MyUserManager(BaseUserManager):
    def create_user(self,email,dob,password=None):
        if not email:
            raise ValueError('Users must have an email address')
        user = self.model(
            email=self.normalize_email(email),
            dob = dob,
        )
        user.set_password(password)
        user.save(using=self.db)
        return user

    def create_superuser(self, email, dob, password):
        user = self.create_user(
            email,
            password=password,
            dob=dob,
        )
        user.is_admin   =   True
        user.save(using=self._db)
        return user

class CustomUser(AbstractBaseUser):
    # add additional fields in here
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )
    phone = models.CharField(max_length=555)
    age = models.CharField(max_length=555)
    dob = models.DateField()
    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)

    objects = MyUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['dob']

    def get_full_name(self):
        # The user is identified by their email address
        return self.email

    def get_short_name(self):
        # The user is identified by their email address
        return self.email

    def __str__(self):              # __unicode__ on Python 2
        return self.email

    def has_perm(self, perm, obj=None):
        "Does the user have a specific permission?"
        # Simplest possible answer: Yes, always
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app `app_label`?"
        # Simplest possible answer: Yes, always
        return True

    @property
    def is_staff(self):
        "Is the user a member of staff?"
        # Simplest possible answer: All admins are staff
        return self.is_admin

步骤3:在我的admin.py文件代码中

from django import forms
from django.contrib import admin
from django.contrib.auth.models import Group
from django.contrib.auth import get_user_model
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth.forms import ReadOnlyPasswordHashField
from .models import CustomUser


class UserCreationForm(forms.ModelForm):
    """A form for creating new users. Includes all the required
    fields, plus a repeated password."""
    password1 = forms.CharField(label='Password', widget=forms.PasswordInput)
    password2 = forms.CharField(label='Password confirmation', widget=forms.PasswordInput)

    class Meta:
        model = CustomUser
        fields = ('email', 'dob')

    def clean_password2(self):
        password1 = self.cleaned_data.get("password1")
        password2 = self.cleaned_data.get("password2")
        if password1 and password2 and password1 != password2:
            raise forms.ValidationError("Passwords don't match")
        return password2

    def save(self, commit=True):
        user = super(UserCreationForm, self).save(commit=False)
        user.set_password(self.cleaned_data["password1"])
        if commit:
            user.save()
        return user


class UserChangeForm(forms.ModelForm):
    """A form for updating users. Includes all the fields on
    the user, but replaces the password field with admin's
    password hash display field.
    """
    password = ReadOnlyPasswordHashField()

    class Meta:
        model = CustomUser
        fields = ('email', 'password', 'dob', 'is_active', 'is_admin')

    def clean_password(self):
        return self.initial["password"]


class UserAdmin(BaseUserAdmin):

    form = UserChangeForm
    add_form = UserCreationForm


    list_display = ('email', 'dob', 'is_admin')
    list_filter = ('is_admin',)
    fieldsets = (
        ('Data', {'fields': ('email', 'password')}),
        ('Personal info', {'fields': ('dob',)}),
        ('Permissions', {'fields': ('is_admin',)}),
    )

    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('email', 'dob', 'password1', 'password2')}
        ),
    )
    search_fields = ('email',)
    ordering = ('email',)
    filter_horizontal = ()

admin.site.register(CustomUser, UserAdmin)
admin.site.unregister(Group)

步骤5:我被添加到setting.py文件中

AUTH_USER_MODEL = 'users.CustomUser'

比起我登录管理员并添加无效凭据,我得到的错误是这样的:

AttributeError at /admin/login/
Manager isn't available; 'auth.User' has been swapped for 'users.CustomUser'

0 个答案:

没有答案