Używanie Django m2m_changed do modyfikowania tego, co jest zapisywane pre_add

Question

Nie bardzo znam sygnały Django i mogę skorzystać z pomocy.

Jak zmodyfikować zestaw pk_set przed zapisaniem instancji? Czy muszę zwrócić coś do osoby dzwoniącej (np. kwargs)? Czy mogę samodzielnie zapisać instance?

Jako uproszczony przykład: chciałem upewnić się, że kategoria zawierająca pk=1 jest dołączona do wszystkich moich filmów wideo po ich zapisaniu. Jak to zrobić z m2m_changed?

class Video(models.Model): 
    category = models.ManyToManyField('Category') 

def video_category_changed(sender, **kwargs): 
    action = kwargs.pop('action', None) 
    pk_set = kwargs.pop('pk_set', None) 
    instance = kwargs.pop('instance', None) 

    if action == "pre_add": 
     if 1 not in pk_set: 
      pk_set.update([ 1 ]) # adding this to the set 
      # do something else? 
      # profit? 

m2m_changed.connect(video_category_changed, sender=Video.category.through) 

Answer 1

Wystarczy aktualizacja pk_set. Nie musisz wykonywać żadnej dodatkowej pracy. Po zapisaniu instancji wideo będzie miała kategorię z pk = 1.

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

class Category(models.Model): 
    pass 

class Video(models.Model): 
    category = models.ManyToManyField('Category') 

@receiver(m2m_changed, sender=Video.category.through) 
def video_category_changed(sender, **kwargs): 
    action = kwargs.pop('action', None) 
    pk_set = kwargs.pop('pk_set', None)  
    if action == "pre_add": 
     if 1 not in pk_set: 
      pk_set.update([1]) 

W powyższej metodzie kategorie zostaną zapisane dopiero po zapisaniu instancji wideo. Jeśli chcesz WYGLĄDNIE zapisać je w instancji m2m_changed, możesz to zrobić w następujący sposób.

@receiver(m2m_changed, sender=Video.category.through) 
def video_category_changed(sender, **kwargs): 
    instance = kwargs.pop('instance', None) 
    pk_set = kwargs.pop('pk_set', None) 
    action = kwargs.pop('action', None) 
    if action == "pre_add": 
     if 1 not in pk_set: 
      c = Category.objects.get(pk=1) 
      instance.category.add(c) 
      instance.save()