检查用户帐户模型类型

Question

检查已扩展 Django 用户模型的帐户的模型类型的最佳方法是什么？ 我有两种扩展 django 用户模型的用户帐户（学生和教师，比如说）：

class Student(models.Model):
     student_bd = models.CharField(max_length=50)
     student_user = models.ForeignKey(User)
     ...

class Teacher(models.Model):
     teacher_bd = models.CharField(max_length=50)
     teacher_user = models.ForeignKey(User)
     ...

那么，我怎么知道 request.user 是 Student 还是 Teacher？

Answer 1

User 表中没有保存关于Student 和Teacher 模型的数据，因此您必须查询相关对象。

如果您不需要实际的配置文件（Student 或Teacher 对象），最快的方法是使用相关集上的exists() 函数：

if request.user.student_set.exists():
    # student
elif request.user.teacher_set.exists():
    # teacher

如果您知道稍后将需要配置文件对象，那么在 Python 中获取结果并检查现有对象会更快：

if request.user.student_set.all():
    # do something with student_set
elif request.user.teacher_set.all():
    # do something with teacher_set

旁注： 如果您的用户不应该拥有多个Student 或Teacher 相关对象，那么使用OneToOneField 而不是ForeignKey 可能是一个更好的主意。这样，您就不必为all() 等而烦恼，但您可以使用user.student 或user.teacher 访问相关的模型实例。这也将允许您使用select_related，以便在单个查询中获取用户对象和学生或教师对象。

Answer 2

如果您有一个 User 模型并想检查该模型是否对应于 Teacher 或 Student 模型，则需要从每个表中获取相应的行：

try:
    Student.objects.get(student_user=request.user)
    is_student = True
except Student.DoesNotExist:
    is_student = False

老师也一样。 is_student 和 is_teacher 将分别告诉您 User 是 Student 还是 Teacher。如果用户同时是学生和教师，这也适用于奇怪的情况:) 在这种情况下没有用，但认为它有助于学校中的其他角色。