像Pro一样扩展Django用户模型(UserProfile)

我发现这个非常有用:link

摘录:

from django.contrib.auth.models import User

class Employee(models.Model):
    user = models.OneToOneField(User)
    department = models.CharField(max_length=100)

>>> u = User.objects.get(username='fsmith')
>>> freds_department = u.employee.department

在这里，我试图解释如何用额外的字段来扩展Django的Default用户模型 很简单，就这么做。

Django允许使用AbstractUser扩展默认的用户模型

注意:-首先创建一个额外的字段模型，你想添加到用户模型，然后运行命令python manage.py makemigrations和python manage.py migrate

首先运行——> python manage.py makemigrationthen

第二步运行python manage.py migrate

步骤:-创建一个带有额外字段的模型，你想在Django默认用户模型中添加这些字段(在我的例子中，我创建了CustomUser

model.py

from django.db import models
from django.contrib.auth.models import AbstractUser
# Create your models here.


class CustomUser(AbstractUser):
    mobile_no = models.IntegerField(blank=True,null=True)
    date_of_birth = models.DateField(blank=True,null=True)

在settings.py中添加你创建的模型名称，在我的例子中CustomUser是用户模型。在settings .py中注册，使其成为默认用户模型，

#settings.py

AUTH_USER_MODEL = 'myapp.CustomUser'

最后在admin.py中注册CustomUser模型 # admin.py

@admin.register(CustomUser)
class CustomUserAdmin(admin.ModelAdmin):
    list_display = ("username","first_name","last_name","email","date_of_birth", "mobile_no")

然后执行命令python manage.py makemigrations

然后python manage.py migrate

然后python manage.py createsuperuser

现在你可以看到你的模型默认用户模型扩展了(mobile_no,date_of_birth)

关于存储用户的附加信息，有一个官方建议。 Django手册也在Profiles部分讨论了这个问题。

注意:这个答案是不赞成的。如果你使用的是Django 1.7或更高版本，请参阅其他答案。

我就是这么做的。

#in models.py
from django.contrib.auth.models import User
from django.db.models.signals import post_save

class UserProfile(models.Model):  
    user = models.OneToOneField(User)  
    #other fields here

    def __str__(self):  
          return "%s's profile" % self.user  

def create_user_profile(sender, instance, created, **kwargs):  
    if created:  
       profile, created = UserProfile.objects.get_or_create(user=instance)  

post_save.connect(create_user_profile, sender=User) 

#in settings.py
AUTH_PROFILE_MODULE = 'YOURAPP.UserProfile'

这将在每次保存用户时创建一个用户配置文件。 然后你可以使用

  user.get_profile().whatever

以下是文档中的更多信息

http://docs.djangoproject.com/en/dev/topics/auth/#storing-additional-information-about-users

更新:请注意AUTH_PROFILE_MODULE自v1.5起已弃用:https://docs.djangoproject.com/en/1.5/ref/settings/#auth-profile-module

简单而有效的方法是 models.py

from django.contrib.auth.models import User
class CustomUser(User):
     profile_pic = models.ImageField(upload_to='...')
     other_field = models.CharField()

这就是我所做的，在我看来这是最简单的方法。为您的新定制模型定义对象管理器，然后定义您的模型。

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

class User_manager(BaseUserManager):
    def create_user(self, username, email, gender, nickname, password):
        email = self.normalize_email(email)
        user = self.model(username=username, email=email, gender=gender, nickname=nickname)
        user.set_password(password)
        user.save(using=self.db)
        return user

    def create_superuser(self, username, email, gender, password, nickname=None):
        user = self.create_user(username=username, email=email, gender=gender, nickname=nickname, password=password)
        user.is_superuser = True
        user.is_staff = True
        user.save()
        return user



  class User(PermissionsMixin, AbstractBaseUser):
    username = models.CharField(max_length=32, unique=True, )
    email = models.EmailField(max_length=32)
    gender_choices = [("M", "Male"), ("F", "Female"), ("O", "Others")]
    gender = models.CharField(choices=gender_choices, default="M", max_length=1)
    nickname = models.CharField(max_length=32, blank=True, null=True)

    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    REQUIRED_FIELDS = ["email", "gender"]
    USERNAME_FIELD = "username"
    objects = User_manager()

    def __str__(self):
        return self.username

不要忘记在你的settings.py中添加这行代码:

AUTH_USER_MODEL = 'YourApp.User'

我就是这么做的，而且总是管用。