django mysql数据库使用自己的User

时间:2023-03-09 01:48:46
django mysql数据库使用自己的User

由于我需要的User模型与django自带的User有所不同,所以需要定义自己的User Model,这里记录一下方法,适用于django 1.5+。

因为使用自己的后台,放弃django的管理后台,所以无需在admin.py中注册。

定义MyUserManagerMyUser

修改myapp下的models.py文件:

from django.db import modelsfrom django.contrib.auth.models import (

BaseUserManager, AbstractBaseUser, PermissionsMixin)

class
MyUserManager(BaseUserManager):

def
_create_user(self, username, email, password, **extra_fields):

"""

Creates
and saves a User with the given username, email and password.

"""

ifnot
username:

raise
ValueError('The given username must be set')

email
= self.normalize_email(email)

user
= self.model(username=username, email=email, **extra_fields)

user.set_password(password)

user.save(using=self._db)

return
user

def
create_user(self, username, email, password, **extra_fields):

extra_fields.setdefault('is_staff',
False)

return
self._create_user(username, email, password, **extra_fields)

def
create_superuser(self, username, email, password, **extra_fields):

extra_fields.setdefault('is_staff',
True)

if
extra_fields.get('is_staff')
isnotTrue:

raise
ValueError('Superuser must have is_staff=True')

return
self._create_user(username, email, password, **extra_fields)

class
MyUser(AbstractBaseUser, PermissionsMixin):

username
= models.CharField(max_length=254,
unique=True,
db_index=True)

email
= models.EmailField('email address',
max_length=254)

is_staff
= models.BooleanField('staff status',
default=False)

is_active
= models.BooleanField('active',
default=True)

USERNAME_FIELD
= 'username'

REQUIRED_FIELDS
= ['email']

objects
= MyUserManager()

class
Meta:

db_table
= 'myuser'

def
get_full_name(self):

return
self.username

def
get_short_name(self):

return
self.username

这里的MyUserManager和django的UserManager大同小异,也可以直接继承UserManager,然后修改_create_user函数即可。

MyUser类即为我们自定义的User模型,我们可以根据需要添加各种属性。

修改settings.py

修改settings.py文件,添加如下内容设置认证使用的model:

AUTH_USER_MODEL
= 'myapp.MyUser'

更新数据库

首先删掉之前的数据库,然后重新建立,运行如下命令生成新的数据表:

$
python manage.py makemigrations myapp

$
python manage.py migrate

经过这三步,默认的User模型已经被替换成了我们自己定义的User模型了。当然我们也可以定义自己的认证模型以及权限系统,后面涉及到相关部分再添加笔记了。

如果要使用django管理后台可以参考以下方法:

https://blog.****.net/watsy/article/details/15506351