Django处理多用户类型的方法介绍
起步
这是许多开发者在项目初期要面临的一个普遍问题。要怎样来处理多用户类型。
本文讲介绍对于不同场景和业务需求如何设计用户模型。为项目提供指导设计。
设计之前
在梳理用户设计之前，有几个前提需要遵守。
1. 不要使用 Django 内置的 User 模型，尽管它能满足应用程序的所有要求。
正如官方Django文档强烈建议为新项目使用自定义用户模型。需求总是在变，只用固定的模型不太现实，并且一旦设置了 AUTH_USER_MODEL 后续再修改就很麻烦了。
2.无论最后选择什么方案，无论有什么业务，都始终只使用一个 Django 模型来处理身份验证。
永远都使用一个用户模型，这就是标题为什么是处理多用户类型而不是多用户模型。这也就能使用统一的身份认证机制。它仍然可以拥有多种用户类型。
方案
不同的项目要求设计出来的方案也是不同的，不妨先问问自己这几个问题：
- 需要维护多少种用户类型？
- 用户可以同时拥有多种角色吗？比如用户可以同时成为学生或老师吗？
- 不同用户类型是否需要存储不同的信息？
一种非常常见的情况就是拥有普通用户和管理员。在这种场景下，可以使用内置的 is_staff 来区分普通用户和管理员。实际上，内置模型中有两个字段来处理这种情况：is_staff 和 is_superuser 。is_staff 标志着用户是否允许登录 Django Admin 页面，至于该用户能做什么或不能做什么，就由权限框架来授权。而 is_superuser 是一个额外的标志，意味着拥有所有权限。所以这里可以看出来，权限有两个级别的管理。
需要维护多少种用户类型？ 如果类型数量比较少，那就可以效仿Django的方式，用几个字段来简单表示：
class User(AbstractUser): is_student = models.BooleanField(default=False) is_teacher = models.BooleanField(default=False)
这可能是处理多种用户类型最简单的方法了。
另一种选择是，只用一个字段就来表示用户:
class User(AbstractUser): USER_TYPE_CHOICES = ( (1, 'student'), (2, 'teacher'), (3, 'secretary'), (4, 'supervisor'), (5, 'admin'), ) user_type = models.PositiveSmallIntegerField(choices=USER_TYPE_CHOICES)
超过 3 种类型可以选择这种方式。
用户可以同时拥有多种角色吗？
如果用户是可以同时承担多种角色的话，那么就需要一个额外的表表示它们多对多的关系，显然可以用到 ManyToManyField ：
class Role(models.Model): ''' The Role entries are managed by the system, automatically created via a Django data migration. ''' STUDENT = 1 TEACHER = 2 SECRETARY = 3 SUPERVISOR = 4 ADMIN = 5 ROLE_CHOICES = ( (STUDENT, 'student'), (TEACHER, 'teacher'), (SECRETARY, 'secretary'), (SUPERVISOR, 'supervisor'), (ADMIN, 'admin'), ) id = models.PositiveSmallIntegerField(choices=ROLE_CHOICES, primary_key=True) def __str__(self): return self.get_id_display() class User(AbstractUser): roles = models.ManyToManyField(Role)
这种方案其实并不常见，因为Django已经提供权限组的功能并拥有灵活的权限管理。你得评估下创建自定义权限组是否更好。
不同用户类型是否需要存储不同的信息？ 如果存储的信息(如头像)与所有用户相关，那么最好的做法就是直接向用户模型添加额外的字段。
如果储存的信息只与特定的类型用户有关，那么可以用 OneToOneField 来进行拓展：
from django.contrib.auth.models import AbstractUser from django.db import models class User(AbstractUser): is_student = models.BooleanField(default=False) is_teacher = models.BooleanField(default=False) class Student(models.Model): user = models.OneToOneField(User, on_delete=models.CASCADE, primary_key=True) ... class Teacher(models.Model): user = models.OneToOneField(User, on_delete=models.CASCADE, primary_key=True) ...
总结
可以参照如下的流程图来帮助设计：
总的来说，始终都用一个用户模型来处理身份认证，不要在多有模型中存储用户名和密码。通常对 User 模型添加布尔标记位就能适应大多数情况。如果需要灵活的权限管理，可以使用Django内置的权限框架或第三方库。
好了，以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对【听图阁-专注于Python设计】的支持。
