AbstractUser重写用户模型

先贴个官方文档:AbstractUser

这个AbstractUser前期用起来有点麻烦,我们都知道django是自带了User的,但是他不能满足所有的业务,所以需要我们重写,接下来走一下流程:
一定要注意,AbstractUser一定要在第一次数据库迁移的时候用,即应用的0001_initial.py一定是为它创建的,不然底下的所有步骤都走不通

一 对应的app里注册模型:一般是在登录、注册那个app里写

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


class UserInfo(AbstractBaseUser):
    identifier = models.CharField(max_length=40, unique=True)
    USERNAME_FIELD = 'identifier'

    phone = models.CharField(max_length=11, default='', verbose_name='电话号码')

    class Meta:
        db_table = 'UserInfo'
        verbose_name = '用户'
        verbose_name_plural = verbose_name

    def __str__(self):
        return self.phone

如果不放identifier = models.CharField(max_length=40, unique=True) USERNAME_FIELD = 'identifier'这两句,会报错说是USERNAME_FIELD不存在

二、如果是整个项目只有一个应用或者是从头到尾都没有用到过 有关于用户的数据关联,那么走完这一步直接数据库迁移就ok了

在settings里面,AUTH_USER_MODEL = 'test_auth.UserInfo'写这一句,test_auth是我应用的名字,UserInfo是类名

三、关于用户的数据关联:

比如,我写了一个放友情链接的模型,里面需要关联整个的作者是谁,正常情况下需要这么写:

from django.contrib.auth.models import User
from django.db import models
from test_auth.models import UserInfo

# Create your models here.
class Link(models.Model):
    author = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)

但是!!!!!
只要执行了第二步,再这么写就会报错,因为现在你已经告诉django,我以后都要用UserInfo替代User,所以django会把以前所有用到User的地方标错,并告诉你:在这里插入图片描述

所以只要改成这个就行:

from django.contrib.auth.models import User
from django.db import models
from test_auth.models import UserInfo

# Create your models here.
class Link(models.Model):
    author = models.ForeignKey(UserInfo, on_delete=models.SET_NULL, null=True)

然后就能迁移了,以后直接用Userinfo就行

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值