django模型:如何克服'通过'ManyToMany选项限制

Question

我正在开发一个应用程序,允许用户自己创建和管理用户组.

问题是我想存储哪个用户向任何组添加了新成员.

这些是我目前的模特:

class UserManagedGroup(Group):

    leader = models.ForeignKey(User, verbose_name=_('group leader'), related_name='leaded_groups')
    members = models.ManyToManyField(User, verbose_name=_('members'), through='Membership',
                                     related_name='managed_groups')

class Membership(models.Model):

    user = models.ForeignKey(User, related_name='memberships')
    group = models.ForeignKey(UserManagedGroup, related_name='memberships')
    info = models.OneToOneField('MembershipInfo', verbose_name=_('membership information'))

    class Meta:
        unique_together = ['user', 'group']

class MembershipInfo(models.Model):

    date_added = models.DateField(_('date added'), auto_now_add=True)
    made_member_by = models.ForeignKey(User, verbose_name=_('user who made him a member'))
    membership_justification = models.TextField(_('membership justification'), blank=True, default='')


@receiver(signals.post_delete, sender=Membership)
def delete_membership_info(sender, instance, *args, **kwargs):
    if instance.info.pk:
        instance.info.delete()

正如你所看到的,我有一个愚蠢的MembershipInfo模型,Membership由于其领域的性质,它更适合合并.此外,MembershipInfos生命受其约束Membership(这就是为什么我必须创建这个post_delete信号连接).

由于这个原因,我无法合并它们:

您的中间模型必须包含一个 - 且只有一个 - 目标模型的外键(在我们的示例中,这将是Person).如果您有多个外键,则会引发验证错误.

(在我的情况下,我不能使用2个外键给用户)

现在,这实际上有效,但我不喜欢它.它使Membership实例创建繁琐,因为我必须首先创建一个MembershipInfo实例.此外,2个查询而不是1个.

问题将2个外键存储到与User我的成员关系绑定的同一模型()的最佳方法.

Answer 1

我刚刚解决了类似的问题,其中包括一个带有两个外键的中间模型到同一个目标.这就是我的系统的样子:

class Node(models.Model):
    receivers = models.ManyToManyField('self', through='Connection',  related_name='senders',  symmetrical=False)

class Connection(models.Model):
    sender = models.ForeignKey(Node, related_name='outgoing')
    receiver = models.ForeignKey(Node, related_name='incoming')

我认为这说明了在中间模型中对同一目标使用两个外键的主要要求.也就是说,模型应该具有ManyToManyField目标'self'(递归ManyToMany)和through指向中间模型的属性.我认为每个外键都必须分配一个唯一的related_name.symmetrical=False如果您希望它们是单向的,则该参数适用于递归关系,例如,Node1向Node2发送信号,但Node2不一定向Node1发送信号.symmetrical=False为了使递归的ManyToMany使用自定义的"通过"模型,必须定义关系.如果您想使用自定义"通过"模型创建对称递归ManyToMany,可以在此处找到建议.

我发现所有这些相互关系相当混乱,所以我花了一些时间来选择合理的模型属性和相关的名称来实际捕获代码正在做的事情.为了阐明这是如何工作的,如果我有节点对象N,则调用N.receivers.all()或N.senders.all()返回从N接收数据或分别向N发送数据的其他节点集.通过related_names 调用N.outgoing.all()或N.incoming.all()访问Connection对象本身.请注意,仍然存在一些不确定性senders,并receivers可能在ManyToManyField被交换和代码来完成工作,但方向被逆转.我通过检查一个测试用例来确定"发送者"是否实际发送给"接收者",反之亦然.

在您的情况下,将两个外键定位到User会增加一个复杂性,因为如何直接向User添加递归ManyToManyField并不明显.我认为自定义User模型的首选方法是通过OneToOneField连接到User的代理扩展它.这可能不会令人满意,因为使用MembershipInfo扩展成员资格并不令人满意,但它至少可以让您轻松地为User模型添加进一步的自定义.

所以对于你的系统,我会尝试这样的事情(未经测试):

class Member(models.Model):
    user = models.OneToOneField(User, related_name='member')
    recruiters = models.ManyToManyField('self', through = 'Membership',  related_name = 'recruits',  symmetrical=False)
    other_custom_info = ... 

class UserManagedGroup(Group):
    leader = models.ForeignKey(Member, related_name='leaded_groups')
    members = models.ManyToManyField(Member, through='Membership', related_name='managed_groups')

class Membership(models.Model):
    member = models.ForeignKey(Member, related_name='memberships')
    made_member_by = models.ForeignKey(Member, related_name='recruitments')
    group = models.ForeignKey(UserManagedGroup, related_name='memberships')

    date_added = ...
    membership_justification = ...

递归字段应该是不对称的,因为Member1招募Member2不应该也意味着Member2招募了Member1.我改变了一些属性,以更清楚地传达关系.您可以使用代理成员,否则您将使用User,因为如果您需要访问用户对象,您始终可以访问Member.user.如果这按预期工作,您应该能够对给定的成员M执行以下操作:

M.recruiters.all() -> set of other members that have recruited M to groups
M.recruits.all() -> set of other members that M has recruited to groups
M.leaded_groups.all() -> set of groups M leads
M.managed_groups.all() -> set of groups of which M is a member
M.memberships.all() -> set of Membership objects in which M has been recruited
M.recruitments.all() -> set of Membership objects in which M has recruited someone

对于G组,

G.memberships.all() -> set of Memberships associated with the group

我认为这应该工作并提供比单独的MembershipInfo模型更"清洁"的解决方案,但它可能需要一些调整,例如检查递归字段的方向以确保招聘人员正在招募新员工而不是反之亦然.

编辑:我忘了将会员模型链接到用户模型.这将是这样做的:

def create_member(member, instance, created, **kwargs):
    if created:
        member, created = Member.objects.get_or_create(user=instance)

post_save.connect(create_member, member=User)

请注意,create_member不是Member的方法,但在定义Member之后调用.通过这样做,每当创建用户时,都应自动创建一个Member对象(如果要在不初始化Member字段的情况下添加用户,则可能需要将成员字段设置为null = True和/或blank = True).