由于我需要的User模型与django自带的User有所不同,所以需要定义自己的User Model,这里记录一下方法,适用于django 1.5+。
因为使用自己的后台,放弃django的管理后台,所以无需在admin.py中注册。
定义MyUserManager和MyUser
修改myapp下的models.py文件:
fromdjango.dbimportmodelsfromdjango.contrib.auth.modelsimport(
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.
"""
ifnotusername:
raiseValueError('The given username must be set')
= self.normalize_email(email)
user
= self.model(username=username, email=email, **extra_fields)
user.set_password(password)
user.save(using=self._db)
returnuser
def
create_user(self, username, email, password, **extra_fields):
extra_fields.setdefault('is_staff',False)
returnself._create_user(username, email, password, **extra_fields)
def
create_superuser(self, username, email, password, **extra_fields):
extra_fields.setdefault('is_staff',True)
ifextra_fields.get('is_staff')isnotTrue:
raiseValueError('Superuser must have is_staff=True')
returnself._create_user(username, email, password, **extra_fields)
class
MyUser(AbstractBaseUser, PermissionsMixin):
username
= models.CharField(max_length=254,
unique=True,
db_index=True)
= 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):
returnself.username
def
get_short_name(self):
returnself.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.csdn.net/watsy/article/details/15506351