DEV Community
·
使用get_user_model()的单元测试
💡
原文英文，约300词，阅读约需1分钟。
📝
内容提要
在Django中，可以通过继承AbstractBaseUser自定义用户模型，示例代码定义了姓名、邮箱、公司等字段，并在settings.py中设置AUTH_USER_MODEL。创建单元测试以验证用户创建成功，推荐使用get_user_model以便于后续修改。
🎯
关键要点
- 在Django中，可以通过继承AbstractBaseUser自定义用户模型。
- 示例代码定义了姓名、邮箱、公司等字段。
- 在settings.py中设置AUTH_USER_MODEL为自定义用户模型。
- 创建单元测试以验证用户创建成功。
- 推荐使用get_user_model以便于后续修改用户模型。
❓
延伸问答
如何在Django中自定义用户模型？
可以通过继承AbstractBaseUser来自定义用户模型，并在settings.py中设置AUTH_USER_MODEL。
在Django中如何创建单元测试以验证用户创建成功？
可以在测试文件中使用get_user_model()创建用户，并验证其邮箱和密码是否正确。
为什么推荐使用get_user_model()而不是直接导入用户模型？
使用get_user_model()可以在更改AUTH_USER_MODEL时避免代码修改，确保代码的灵活性。
自定义用户模型中可以定义哪些字段？
可以定义姓名、邮箱、公司、地址、城市、国家、邮政编码和电话等字段。
如何在settings.py中设置自定义用户模型？
在settings.py中添加AUTH_USER_MODEL = 'core.user'来设置自定义用户模型。
自定义用户模型的默认字段是什么？
自定义用户模型的默认字段包括is_active和is_staff，分别表示用户是否激活和是否为员工。
➡️
