Django - 多个用户配置文件

Question

最初，我的 UserProfile 是这样开始的：

from django.db import models
from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.OneToOneField(User)
    verified = models.BooleanField()
    mobile = models.CharField(max_length=32)

    def __unicode__(self):
        return self.user.email

与settings.py 中设置的AUTH_PROFILE_MODULE = 'accounts.UserProfile' 配合得很好。

但是，我的网站中有两种不同类型的用户，个人和公司，每个用户都有自己独特的属性。例如，我希望我的个人用户只有一个用户，因此拥有user = models.OneToOneField(User)，而对于企业，我希望他们拥有多个与同一个人资料相关的用户，所以我将拥有user = models.ForeignKey(User)。

所以我考虑将模型分成两个不同的模型，IndivProfile 和 CorpProfile，它们都继承自 UserProfile，同时将特定于模型的属性移动到相关的子模型中。对我来说似乎是个好主意，并且可能会起作用，但是我无法以这种方式指定 AUTH_PROFILE_MODULE，因为我有两个不同用户的用户配置文件。

我也想过反过来做，让UserProfile从多个类（模型）继承，像这样：

class UserProfile(IndivProfile, CorpProfile):
    # some field

    def __unicode__(self):
        return self.user.email

这样我会设置AUTH_PROFILE_MODULE = 'accounts.UserProfile' 并解决它的问题。但这看起来行不通，因为 python 中的继承是从左到右工作的，IndivProfile 中的所有变量都将占主导地位。

当然，我总是可以拥有一个模型，其中 IndivProfile 和 CorpProfile 变量都混合在一起，然后我会在必要时使用所需的模型。但这对我来说看起来并不干净，我宁愿将它们隔离并在适当的地方使用适当的模型。

对干净的方法有什么建议吗？

Answer 1

值得尝试使用直通字段。它背后的想法是将 UserProfile 模型用作 CorpProfile 或 IndivProfile 模型的直通模型。这样，只要将公司或个人资料链接到用户，就会创建它：

from django.db import models
from django.contrib.auth.models import User

class UserProfile(models.Model):
    user = models.ForeignKey(User)
    profile = models.ForeignKey(Profile, related_name='special_profile')

class Profile(models.Model):
    common_property=something

class CorpProfile(Profile):
    user=models.ForeignKey(User, through=UserProfile)
    corp_property1=someproperty1
    corp_property2=someproperty2

class IndivProfile(Profile):
    user=models.ForeignKey(User, through=UserProfile, unique=true)
    indiv_property1=something
    indiv_property2=something

我认为这样应该可以设置AUTH_PROFILE_MODULE = 'accounts.UserProfile'，并且每次创建链接到真实用户的 CorpProfile 或 IndivProfile 时，都会创建一个唯一的 UserProfile 模型。然后，您可以使用 db 查询或任何您想要的方式访问它。

我没有测试过这个，所以不能保证。这可能有点hacky，但另一方面我觉得这个想法很吸引人。 :)

Answer 2

您可以通过以下方式执行此操作。有一个配置文件，其中包含两个配置文件中必需的公共字段。您已经通过创建类 UserProfile 来完成此操作。

class UserProfile(models.Model):
    user = models.ForeignKey(User)
    # Some common fields here, which are shared among both corporate and individual profiles

class CorporateUser(models.Model):
    profile = models.ForeignKey(UserProfile)
    # Corporate fields here

    class Meta:
        db_table = 'corporate_user'

class IndividualUser(models.Model):
    profile = models.ForeignKey(UserProfile)
    # Individual user fields here

   class Meta:
        db_table = 'individual_user'

这里不涉及火箭科学。只需有一个关键字来区分公司简介或个人简介。例如。考虑用户正在注册。然后在表单上有一个字段，用于区分用户是否注册公司。并使用该关键字（请求参数）将用户保存在相应的模型中。

然后当你想检查用户的个人资料是公司还是个人时，你可以通过编写一个小函数来检查它。

def is_corporate_profile(profile):
    try:
        profile.corporate_user
        return True
    except CorporateUser.DoesNotExist:
        return False

# If there is no corporate profile is associated with main profile then it will raise `DoesNotExist` exception and it means its individual profile
# You can use this function as a template function also to use in template
{% if profile|is_corporate_profile %}

希望这会带你到某个地方。谢谢！

Answer 3

我是这样做的。

PROFILE_TYPES = (
    (u'INDV', 'Individual'),
    (u'CORP', 'Corporate'),
)

# used just to define the relation between User and Profile
class UserProfile(models.Model):
    user = models.ForeignKey(User)
    profile = models.ForeignKey('Profile')
    type = models.CharField(choices=PROFILE_TYPES, max_length=16)

# common fields reside here
class Profile(models.Model):
    verified = models.BooleanField(default=False)

我最终使用了一个中间表来反映两个抽象模型之间的关系，User 已经在 Django 中定义，我的Profile 模型。如果有不常见的属性，我将创建一个新模型并将其关联到Profile。