Django

Code

Changeset 4439

Show
Ignore:
Timestamp:
01/27/07 22:56:54 (2 years ago)
Author:
adrian
Message:

Fixed #3263 -- newforms form_for_model() and form_for_instance() now handle foreign-key and many-to-many data properly. Thanks for the patch, Jeff Hilyard

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved
  • django/trunk/django/newforms/models.py

    r4437 r4439  
    1616    if self.errors: 
    1717        raise ValueError("The %s could not be created because the data didn't validate." % self._model._meta.object_name) 
    18     obj = self._model(**self.clean_data) 
    19     if commit: 
    20         obj.save() 
    21     return obj 
     18    return save_instance(self, self._model(), commit) 
    2219 
    2320def save_instance(form, instance, commit=True): 
     
    3431        raise ValueError("The %s could not be changed because the data didn't validate." % opts.object_name) 
    3532    clean_data = form.clean_data 
    36     for f in opts.fields + opts.many_to_many
     33    for f in opts.fields
    3734        if isinstance(f, models.AutoField): 
    3835            continue 
     
    4037    if commit: 
    4138        instance.save() 
     39        for f in opts.many_to_many: 
     40            setattr(instance, f.attname, getattr(instance, f.attname).model.objects.filter(pk__in = clean_data[f.name])) 
     41    # GOTCHA: If many-to-many data is given and commit=False, the many-to-many 
     42    # data will be lost. This happens because a many-to-many options cannot be 
     43    # set on an object until after it's saved. Maybe we should raise an 
     44    # exception in that case. 
    4245    return instance 
    4346 
  • django/trunk/tests/modeltests/model_forms/models.py

    r4305 r4439  
    225225</select></li> 
    226226 
     227>>> f = TestArticleForm({'headline': u'New headline', 'pub_date': u'1988-01-04', 
     228...     'writer': u'1', 'article': u'Hello.', 'categories': [u'1', u'2']}) 
     229>>> new_art = f.save() 
     230>>> new_art.id 
     2311 
     232>>> new_art = Article.objects.get(id=1) 
     233>>> new_art.categories.all() 
     234[<Category: Entertainment>, <Category: It's a test>] 
     235 
     236Now, submit form data with no categories. This deletes the existing categories. 
     237>>> f = TestArticleForm({'headline': u'New headline', 'pub_date': u'1988-01-04', 
     238...     'writer': u'1', 'article': u'Hello.'}) 
     239>>> new_art = f.save() 
     240>>> new_art.id 
     2411 
     242>>> new_art = Article.objects.get(id=1) 
     243>>> new_art.categories.all() 
     244[] 
     245 
     246Create a new article, with categories, via the form. 
     247>>> ArticleForm = form_for_model(Article) 
     248>>> f = ArticleForm({'headline': u'The walrus was Paul', 'pub_date': u'1967-11-01', 
     249...     'writer': u'1', 'article': u'Test.', 'categories': [u'1', u'2']}) 
     250>>> new_art = f.save() 
     251>>> new_art.id 
     2522 
     253>>> new_art = Article.objects.get(id=2) 
     254>>> new_art.categories.all() 
     255[<Category: Entertainment>, <Category: It's a test>] 
     256 
     257Create a new article, with no categories, via the form. 
     258>>> ArticleForm = form_for_model(Article) 
     259>>> f = ArticleForm({'headline': u'The walrus was Paul', 'pub_date': u'1967-11-01', 
     260...     'writer': u'1', 'article': u'Test.'}) 
     261>>> new_art = f.save() 
     262>>> new_art.id 
     2633 
     264>>> new_art = Article.objects.get(id=3) 
     265>>> new_art.categories.all() 
     266[] 
     267 
    227268Here, we define a custom Form. Because it happens to have the same fields as 
    228269the Category model, we can use save_instance() to apply its changes to an