Переопределение метода сохранения в Django ModelForm - PullRequest
Новая
       78

Переопределение метода сохранения в Django ModelForm

57 голосов
/ 03 мая 2009

У меня проблемы с переопределением ModelForm метода сохранения. Это ошибка, которую я получаю: 

Exception Type:     TypeError  
Exception Value:    save() got an unexpected keyword argument 'commit'

Мое намерение состоит в том, чтобы форма представила множество значений для 3 полей, чтобы затем создать объект для каждой комбинации этих полей и сохранить каждый из этих объектов. Полезным толчком в правильном направлении будет туз.

Файл models.py

class CallResultType(models.Model):
    id = models.AutoField(db_column='icontact_result_code_type_id', primary_key=True)
    callResult = models.ForeignKey('CallResult', db_column='icontact_result_code_id')
    campaign = models.ForeignKey('Campaign', db_column='icampaign_id')
    callType = models.ForeignKey('CallType', db_column='icall_type_id')
    agent = models.BooleanField(db_column='bagent', default=True)
    teamLeader = models.BooleanField(db_column='bTeamLeader', default=True)
    active = models.BooleanField(db_column='bactive', default=True)

Файл forms.py

from django.forms import ModelForm, ModelMultipleChoiceField
from callresults.models import *

class CallResultTypeForm(ModelForm):
    callResult = ModelMultipleChoiceField(queryset=CallResult.objects.all())
    campaign = ModelMultipleChoiceField(queryset=Campaign.objects.all())
    callType = ModelMultipleChoiceField(queryset=CallType.objects.all())

    def save(self, force_insert=False, force_update=False):
        for cr in self.callResult:
            for c in self.campain:
                for ct in self.callType:
                    m = CallResultType(self) # this line is probably wrong
                    m.callResult = cr
                    m.campaign = c
                    m.calltype = ct
                    m.save()

    class Meta:
        model = CallResultType

Файл admin.py

class CallResultTypeAdmin(admin.ModelAdmin):
    form = CallResultTypeForm

1 Ответ

134 голосов
/ 03 мая 2009

В вашем save вы должны иметь аргумент commit. Если что-то переопределяет вашу форму или хочет изменить то, что она сохраняет, оно выполнит save(commit=False), изменит вывод, а затем сохранит само.

Кроме того, ваша ModelForm должна возвращать модель, которую она сохраняет. Обычно ModelForm save будет выглядеть примерно так: 

def save(self, commit=True):
    m = super(CallResultTypeForm, self).save(commit=False)
    # do custom stuff
    if commit:
        m.save()
    return m

Читайте метод save .

Наконец, большая часть этой ModelForm не будет работать только из-за вашего доступа к вещам. Вместо self.callResult вам нужно использовать self.fields['callResult'].

ОБНОВЛЕНИЕ : В ответ на ваш ответ:

В сторону: Почему бы просто не использовать ManyToManyField s в модели, чтобы вам не пришлось это делать? Похоже, вы храните избыточные данные и делаете больше для себя (и я :P). 

from django.db.models import AutoField  
def copy_model_instance(obj):  
    """
    Create a copy of a model instance. 
    M2M relationships are currently not handled, i.e. they are not copied. (Fortunately, you don't have any in this case)
    See also Django #4027. From http://blog.elsdoerfer.name/2008/09/09/making-a-copy-of-a-model-instance/
    """  
    initial = dict([(f.name, getattr(obj, f.name)) for f in obj._meta.fields if not isinstance(f, AutoField) and not f in obj._meta.parents.values()])  
    return obj.__class__(**initial)  

class CallResultTypeForm(ModelForm):
    callResult = ModelMultipleChoiceField(queryset=CallResult.objects.all())
    campaign = ModelMultipleChoiceField(queryset=Campaign.objects.all())
    callType = ModelMultipleChoiceField(queryset=CallType.objects.all())

    def save(self, commit=True, *args, **kwargs):
        m = super(CallResultTypeForm, self).save(commit=False, *args, **kwargs)
        results = []
        for cr in self.callResult:
            for c in self.campain:
                for ct in self.callType:
                    m_new = copy_model_instance(m)
                    m_new.callResult = cr
                    m_new.campaign = c
                    m_new.calltype = ct
                    if commit:
                        m_new.save()
                    results.append(m_new)
         return results

Это позволяет наследовать CallResultTypeForm, на всякий случай, когда это необходимо.

...