Django 自定义 User model

Django 自带的 User 及 auth 库为快速开发带来了很大便利,也增加了扩展其功能的难度。 Django 官方文档中是推荐使用一对一的关联 Model 来扩展 User model:

class UserExtend(Model):
    user = OneToOneField(User, auto_create=True, relate_name='extend')
    field1 = XxxField()

不过在使用时仍需要 user.extend.xxx_property,不是很方便。

在较新版本的 Django 提供了另一种扩展 Django 的方法——替换 AUTH_USER_MODEL。 Django 提供了 AbstractUserAbstractBaseUser model 类提供不同程度抽象的 User 抽象 model,继承它之后再在配置文件中指定 AUTH_USER_MODEL 即可:

# settings.py
AUTH_USER_MODEL = 'app.User'

需要在 Django app 的 init.py 文件中引用 User model,这里似乎是 Django 解析模块名和类名的函数 bug 的影响。

# app/__init__.py
from app.models import User
# app/models.py
class User(AbstractUser):
    name = CharField(max_length=100)
    level = IntegerField(null=True)     # 注意,如果存在不可为空的非字符字段,使用创建用户命令时会报错
    birthday = DatetimeField(null=True)
    gender = NullableBooleanField()

最后测试一下是否成功:

from django.contrib.auth import get_user_model
print(get_user_model())        # app.models.User

以上!

参考连接:

扩展阅读:Migrating to a Custom User Model in Django

Kane Blueriver all right reserved,powered by Gitbook最近修改时间: 2018-05-21 17:28:09

results matching ""

    No results matching ""