Ticket #17464: Added-a-get_default_instance-method-to-BaseCreateVie.diff

django/views/generic/edit.py

@@ -163 +163 @@
 
     Using this base class requires subclassing to provide a response mixin.
     """
+    def get_default_instance(self):
+        """
+        Returns the default instance for the object, in case you need to
+        set some fields that not in the form.
+        """
+        return None
+
     def get(self, request, *args, **kwargs):
         self.object = None
         return super(BaseCreateView, self).get(request, *args, **kwargs)
 
     def post(self, request, *args, **kwargs):
-        self.object = None
+        self.object = self.get_default_instance()
         return super(BaseCreateView, self).post(request, *args, **kwargs)
 
 

tests/regressiontests/generic_views/edit.py

@@ -92 +92 @@
         self.assertEqual(res.status_code, 302)
         self.assertRedirects(res, 'http://testserver/accounts/login/?next=/edit/authors/create/restricted/')
 
+    def test_create_with_default_values(self):
+        res = self.client.post('/edit/authors/create/defaultinstance/',
+                        {'name': 'Randall Munroe'})
+        self.assertEqual(res.status_code, 302)
+        self.assertRedirects(res, 'http://testserver/list/authors/')
+        self.assertQuerysetEqual(Author.objects.all(), ['<Author: Randall Munroe>'])
+        author = Author.objects.get(name='Randall Munroe')
+        self.assertEqual(author.slug, 'default-slug')
+
 class UpdateViewTests(TestCase):
     urls = 'regressiontests.generic_views.urls'
 

tests/regressiontests/generic_views/urls.py

@@ -71 +71 @@
         views.AuthorCreate.as_view()),
     (r'^edit/authors/create/special/$',
         views.SpecializedAuthorCreate.as_view()),
+    (r'^edit/authors/create/defaultinstance/$',
+        views.DefaultInstanceAuthorCreate.as_view()),
 
     (r'^edit/author/(?P<pk>\d+)/update/naive/$',
         views.NaiveAuthorUpdate.as_view()),

tests/regressiontests/generic_views/views.py

@@ -3 +3 @@
 from django.contrib.auth.decorators import login_required
 from django.core.paginator import Paginator
 from django.core.urlresolvers import reverse
+from django.forms.models import modelform_factory
 from django.utils.decorators import method_decorator
 from django.views import generic
 
@@ -103 +104 @@
     post = method_decorator(login_required)(AuthorCreate.post)
 
 
+class DefaultInstanceAuthorCreate(AuthorCreate):
+    form_class = modelform_factory(Author, fields=('name',))
+    def get_default_instance(self):
+        return Author(slug='default-slug')
+
+
 class ArtistUpdate(generic.UpdateView):
     model = Artist