python - 使用 hasattr() 检查用户类型

Question

我的一些用户是学生。当用户创建学生文件时，StudentProfile 类被实例化:

class StudentProfile(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL, primary_key=True)
    …

如何检查用户是否是学生？

hasattr(request.user, 'StudentProfile')

即使登录用户有关联的 StudentProfile，也返回 False。

Answer 1

你就快完成了 - 你只需要使用小写的 studentprofile 而不是 StudentProfile:

hasattr(request.user, 'studentprofile')

来自 the docs :

If you do not specify the related_name argument for the OneToOneField, Django will use the lower-case name of the current model as default value.