28 - Django Model Form for Article Model

Author: codingforentrepreneurs

articles/forms.py

@@ -1,6 +1,23 @@
 from django import forms
 
-class ArticleForm(forms.Form):
+from .models import Article
+
+class ArticleForm(forms.ModelForm):
+    class Meta:
+        model = Article
+        fields = ['title', 'content']
+
+    def clean(self):
+        data = self.cleaned_data
+        title = data.get("title")
+        qs = Article.objects.filter(title__icontains=title)
+        if qs.exists():
+            self.add_error("title", f"\"{title}\" is already in use. Please pick another title.")
+        return data
+
+
+
+class ArticleFormOld(forms.Form):
     title = forms.CharField()
     content = forms.CharField()
 

articles/views.py

@@ -24,22 +24,34 @@ def article_search_view(request):
 
 @login_required
 def article_create_view(request):
-    # print(request.POST)
-    form = ArticleForm()
+    form = ArticleForm(request.POST or None)
     context = {
         "form": form
     }
-    if request.method == "POST":
-        form = ArticleForm(request.POST)
-        context['form'] = form
-        if form.is_valid():
-            title = form.cleaned_data.get("title")
-            content = form.cleaned_data.get("content")
-            article_object = Article.objects.create(title=title, content=content)
-            context['object'] = article_object
-            context['created'] = True
+    if form.is_valid():
+        article_object = form.save()
+        context['form'] = ArticleForm()
+        # context['object'] = article_object
+        # context['created'] = True
     return render(request, "articles/create.html", context=context)
 
+# def article_create_view(request):
+#     # print(request.POST)
+#     form = ArticleForm()
+#     context = {
+#         "form": form
+#     }
+#     if request.method == "POST":
+#         form = ArticleForm(request.POST)
+#         context['form'] = form
+#         if form.is_valid():
+#             title = form.cleaned_data.get("title")
+#             content = form.cleaned_data.get("content")
+#             article_object = Article.objects.create(title=title, content=content)
+#             context['object'] = article_object
+#             context['created'] = True
+#     return render(request, "articles/create.html", context=context)
+
 def article_detail_view(request, id=None):
     article_obj = None
     if id is not None: