Django的暂时禁用信号(django temporarily disable signals)

2019-07-30 10:18发布

我有在Django的信号回调:

@receiver(post_save, sender=MediumCategory)
def update_category_descendants(sender, **kwargs):

    def children_for(category):
        return MediumCategory.objects.filter(parent=category)

    def do_update_descendants(category):
        children = children_for(category)
        descendants = list() + list(children)

        for descendants_part in [do_update_descendants(child) for child in children]:
            descendants += descendants_part

        category.descendants.clear()
        for descendant in descendants:
            if category and not (descendant in category.descendants.all()):
                category.descendants.add(descendant)
                category.save()
        return list(descendants)

    # call it for update
    do_update_descendants(None)

但在函数体中我使用.save()的模型MediumCategory是couses该信号再次出动。 我如何可以禁用它; 完美的解决方案将是一个with声明的一些“神奇”内。

UPDATE:这是最终的解决方案,如果有人感兴趣。

class MediumCategory(models.Model):
    name = models.CharField(max_length=100)
    slug = models.SlugField(blank=True)
    parent = models.ForeignKey('self', blank=True, null=True)
    parameters = models.ManyToManyField(AdvertisementDescriptonParameter, blank=True)
    count_mediums = models.PositiveIntegerField(default=0)
    count_ads = models.PositiveIntegerField(default=0)

    descendants = models.ManyToManyField('self', blank=True, null=True)

    def save(self, *args, **kwargs):
        self.slug = slugify(self.name)
        super(MediumCategory, self).save(*args, **kwargs)

    def __unicode__(self):
        return unicode(self.name)
(...)
@receiver(post_save, sender=MediumCategory)
def update_category_descendants(sender=None, **kwargs):
    def children_for(category):
        return MediumCategory.objects.filter(parent=category)

    def do_update_descendants(category):
        children = children_for(category)
        descendants = list() + list(children)

        for descendants_part in [do_update_descendants(child) for child in children]:
            descendants += descendants_part

        if category:
            category.descendants.clear()
            for descendant in descendants:
                category.descendants.add(descendant)
        return list(descendants)

    # call it for update
    do_update_descendants(None)

Answer 1:

也许我错了,但我认为category.save()因为变化在后代但在产品制造,不需要在你的代码中,添加()是不够的。

此外,为避免您能信号:

  • 断开信号并重新连接。
  • 使用更新 : Descendant.objects.filter( pk = descendant.pk ).update( category = category )


Answer 2:

@danihp断开的信号是不是一个DRY和一致的解决方案,例如,使用更新()而不是保存()。

要在模型中禁用的信号,一个简单的方法去是对当前实例的属性集以防止即将到来的信号发射。

这可以通过使用一个简单的装饰,检查如果给定实例具有“skip_signal”属性来完成,并且如果是这样防止被调用的方法:

from functools import wraps

def skip_signal():
    def _skip_signal(signal_func):
        @wraps(signal_func)
        def _decorator(sender, instance, **kwargs):
            if hasattr(instance, 'skip_signal'):
                return None
            return signal_func(sender, instance, **kwargs)  
        return _decorator
    return _skip_signal

现在,您可以使用这种方式:

from django.db.models.signals import post_save
from django.dispatch import receiver

@receiver(post_save, sender=MyModel)
@skip_signal()
def my_model_post_save(sender, instance, **kwargs):
    instance.some_field = my_value
    # Here we flag the instance with 'skip_signal'
    # and my_model_post_save won't be called again
    # thanks to our decorator, avoiding any signal recursion
    instance.skip_signal  = True
    instance.save()

希望这可以帮助。



文章来源: django temporarily disable signals