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

这里的MyUserManagerdjangoUserManager大同小异,也可以直接继承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.csdn.net/watsy/article/details/15506351

 

posted @ 2018-04-19 19:30  aaron_agu  阅读(859)  评论(0编辑  收藏  举报