Django - 如何通过post_save信号保存m2m数据?

时间:2010-12-13 18:50:27

标签: django django-admin membership

(Django 1.1)我有一个项目模型,使用m2m字段跟踪其成员。它看起来像这样:

class Project(models.Model):
    members = models.ManyToManyField(User)
    sales_rep = models.ForeignKey(User)
    sales_mgr = models.ForeignKey(User)
    project_mgr = models.ForeignKey(User)
    ... (more FK user fields) ...

创建项目后,会将选定的sales_repsales_mgrproject_mgrUser添加到成员中,以便更轻松地跟踪项目权限。到目前为止,这种方法运作良好。

我现在要处理的问题是当通过管理员更新其中一个User FK字段时,如何更新项目的成员资格。我已经尝试过各种解决这个问题的方法,但最干净的方法似乎是post_save信号,如下所示:

def update_members(instance, created, **kwargs):
    """
    Signal to update project members
    """
    if not created: #Created projects are handled differently
        instance.members.clear()

        members_list = []
        if instance.sales_rep:
            members_list.append(instance.sales_rep)
        if instance.sales_mgr:
            members_list.append(instance.sales_mgr)
        if instance.project_mgr:
            members_list.append(instance.project_mgr)

        for m in members_list:
            instance.members.add(m)
signals.post_save.connect(update_members, sender=Project)  

但是,即使我通过管理员更改其中一个字段,Project仍然具有相同的成员!我已经成功使用我自己的视图在其他项目中更新成员m2m字段,但我从来没有让它与管理员一起玩得很好。

除了post_save信号之外,还有其他方法可以更新会员资格吗?在此先感谢您的帮助!

更新

只是为了澄清一下,当我在前端保存自己的表单时,post_save信号正常工作(删除旧成员,添加新成员)。但是,当我通过管理员保存项目时,post_save信号无法正常工作(成员保持不变)。

我认为Peter Rowell在这种情况下的诊断是正确的。如果我从管理表单中删除“成员”字段,则post_save信号可以正常工作。包含该字段时,它会根据保存时表单中的值保存旧成员。无论我在保存项目时对成员m2m字段做了什么更改(无论是信号还是自定义保存方法),它都将被保存前表单中存在的成员覆盖。谢谢你指出来了!

3 个答案:

答案 0 :(得分:7)

遇到同样的问题,我的解决方案是使用m2m_changed信号。您可以在两个地方使用它,如下例所示。

保存后的管理员将继续:

  • 保存模型字段
  • 发出post_save信号
  • 每个m2m
    • 发出pre_clear
    • 清除关系
    • 发出post_clear
    • emit pre_add
    • 再次填充
    • 发出post_add

这里有一个简单的示例,可以在实际保存之前更改已保存数据的内容。

class MyModel(models.Model):

    m2mfield = ManyToManyField(OtherModel)

    @staticmethod
    def met(sender, instance, action, reverse, model, pk_set, **kwargs):
        if action == 'pre_add':
            # here you can modify things, for instance
            pk_set.intersection_update([1,2,3]) 
            # only save relations to objects 1, 2 and 3, ignoring the others
        elif action == 'post_add':
            print pk_set
            # should contain at most 1, 2 and 3

m2m_changed.connect(receiver=MyModel.met, sender=MyModel.m2mfield.through)

您还可以收听pre_removepost_removepre_clearpost_clear。在我的情况下,我使用它们来过滤另一个('启用的东西')内容中的一个列表('活动的东西'),而不依赖于保存列表的顺序:

def clean_services(sender, instance, action, reverse, model, pk_set, **kwargs):
    """ Ensures that the active services are a subset of the enabled ones.
    """
    if action == 'pre_add' and sender == Account.active_services.through:
        # remove from the selection the disabled ones
        pk_set.intersection_update(instance.enabled_services.values_list('id', flat=True))
    elif action == 'pre_clear' and sender == Account.enabled_services.through:
        # clear everything
        instance._cache_active_services = list(instance.active_services.values_list('id', flat=True))
        instance.active_services.clear()
    elif action == 'post_add' and sender == Account.enabled_services.through:
        _cache_active_services = getattr(instance, '_cache_active_services', None)
        if _cache_active_services:
            instance.active_services.add(*list(instance.enabled_services.filter(id__in=_cache_active_services)))
            delattr(instance, '_cache_active_services')
    elif action == 'pre_remove' and sender == Account.enabled_services.through:
        # de-default any service we are disabling
        instance.active_services.remove(*list(instance.active_services.filter(id__in=pk_set)))

如果“已启用”的更新(清除/删除+添加回来,就像在管理员中一样),则“活动”缓存并在第一次传递('pre_clear')中清除,然后在缓存后添加回来第二遍('post_add')。

诀窍是更新另一个的m2m_changed信号上的一个列表。

答案 1 :(得分:4)

我看不出您的代码有什么问题,但我很困惑为什么您认为管理员应该与其他任何应用程序有任何不同。

但是,我必须说我认为你的模型结构是错误的。我认为你需要摆脱所有的ForeignKey字段,并且只需要一个ManyToMany - 但是使用直通表来跟踪角色。

class Project(models.Model):
    members = models.ManyToManyField(User, through='ProjectRole')

class ProjectRole(models.Model):
    ROLES = (
       ('SR', 'Sales Rep'),
       ('SM', 'Sales Manager'),
       ('PM', 'Project Manager'),
    )
    project = models.ForeignKey(Project)
    user = models.ForeignKey(User)
    role = models.CharField(max_length=2, choices=ROLES)

答案 2 :(得分:0)

当我需要通过m2m_field连接到模型的项目集中找到最新项目时,我一直坚持下去。

根据Saverio的回答,以下代码解决了我的问题:

def update_item(sender, instance, action, **kwargs):
    if action == 'post_add':
        instance.related_field = instance.m2m_field.all().order_by('-datetime')[0]
        instance.save()

m2m_changed.connect(update_item, sender=MyCoolModel.m2m_field.through)