Example app code deduplication

Author: insspb

example_app/boolean_demo.py

@@ -1,7 +1,5 @@
 """Booleans fields demo model."""
 
-from flask import render_template, request
-
 from example_app.models import db
 
 
@@ -24,25 +22,8 @@ class BooleanDemoModel(db.Document):
 
 def boolean_demo_view(pk=None):
     """Return all fields demonstration."""
-    form = BooleanDemoForm()
-    obj = None
-    if pk:
-        obj = BooleanDemoModel.objects.get(pk=pk)
-        form = BooleanDemoForm(obj=obj)
-
-    if request.method == "POST" and form.validate_on_submit():
-        if pk:
-            form.populate_obj(obj)
-            obj.save()
-        else:
-            form.save()
-    page_num = int(request.args.get("page") or 1)
-    page = BooleanDemoModel.objects.paginate(page=page_num, per_page=100)
-
-    return render_template(
-        "form_demo.html",
-        view=boolean_demo_view.__name__,
-        page=page,
-        form=form,
-        model=BooleanDemoModel,
+    from example_app.views import demo_view
+
+    return demo_view(
+        model=BooleanDemoModel, view_name=boolean_demo_view.__name__, pk=pk
     )

example_app/dates_demo.py

@@ -1,6 +1,5 @@
 """Date and DateTime fields demo model."""
 
-from flask import render_template, request
 from wtforms.fields import DateTimeField
 
 from example_app.models import db
@@ -28,25 +27,6 @@ class DateTimeModel(db.Document):
 
 def dates_demo_view(pk=None):
     """Return all fields demonstration."""
-    form = DateTimeDemoForm()
-    obj = None
-    if pk:
-        obj = DateTimeModel.objects.get(pk=pk)
-        form = DateTimeDemoForm(obj=obj)
-
-    if request.method == "POST" and form.validate_on_submit():
-        if pk:
-            form.populate_obj(obj)
-            obj.save()
-        else:
-            form.save()
-    page_num = int(request.args.get("page") or 1)
-    page = DateTimeModel.objects.paginate(page=page_num, per_page=100)
-
-    return render_template(
-        "form_demo.html",
-        view=dates_demo_view.__name__,
-        page=page,
-        form=form,
-        model=DateTimeModel,
-    )
+    from example_app.views import demo_view
+
+    return demo_view(model=DateTimeModel, view_name=dates_demo_view.__name__, pk=pk)

example_app/numbers_demo.py

@@ -2,8 +2,6 @@
 
 from decimal import Decimal
 
-from flask import render_template, request
-
 from example_app.models import db
 
 
@@ -26,25 +24,8 @@ class NumbersDemoModel(db.Document):
 
 def numbers_demo_view(pk=None):
     """Return all fields demonstration."""
-    form = NumbersDemoForm()
-    obj = None
-    if pk:
-        obj = NumbersDemoModel.objects.get(pk=pk)
-        form = NumbersDemoForm(obj=obj)
-
-    if request.method == "POST" and form.validate_on_submit():
-        if pk:
-            form.populate_obj(obj)
-            obj.save()
-        else:
-            form.save()
-    page_num = int(request.args.get("page") or 1)
-    page = NumbersDemoModel.objects.paginate(page=page_num, per_page=100)
-
-    return render_template(
-        "form_demo.html",
-        view=numbers_demo_view.__name__,
-        page=page,
-        form=form,
-        model=NumbersDemoModel,
+    from example_app.views import demo_view
+
+    return demo_view(
+        model=NumbersDemoModel, view_name=numbers_demo_view.__name__, pk=pk
     )

example_app/strings_demo.py

@@ -1,8 +1,6 @@
 """Strings and strings related fields demo model."""
 import re
 
-from flask import render_template, request
-
 from example_app.models import db
 from flask_mongoengine.wtf import fields as mongo_fields
 
@@ -32,25 +30,8 @@ class StringsDemoModel(db.Document):
 
 def strings_demo_view(pk=None):
     """Return all fields demonstration."""
-    form = StringsDemoForm()
-    obj = None
-    if pk:
-        obj = StringsDemoModel.objects.get(pk=pk)
-        form = StringsDemoForm(obj=obj)
-
-    if request.method == "POST" and form.validate_on_submit():
-        if pk:
-            form.populate_obj(obj)
-            obj.save()
-        else:
-            form.save()
-    page_num = int(request.args.get("page") or 1)
-    page = StringsDemoModel.objects.paginate(page=page_num, per_page=100)
-
-    return render_template(
-        "form_demo.html",
-        view=strings_demo_view.__name__,
-        page=page,
-        form=form,
-        model=StringsDemoModel,
+    from example_app.views import demo_view
+
+    return demo_view(
+        model=StringsDemoModel, view_name=strings_demo_view.__name__, pk=pk
     )

example_app/views.py

@@ -84,3 +84,21 @@ def pagination():
         todos_page = models.Todo.objects.paginate(page=page_num, per_page=3)
 
     return render_template("pagination.html", todos_page=todos_page, form=form)
+
+
+def demo_view(model, view_name, pk=None):
+    """Return all fields demonstration."""
+    FormClass = model.to_wtf_form()
+    form = FormClass()
+    if pk:
+        obj = model.objects.get(pk=pk)
+        form = FormClass(obj=obj)
+
+    if request.method == "POST" and form.validate_on_submit():
+        form.save()
+    page_num = int(request.args.get("page") or 1)
+    page = model.objects.paginate(page=page_num, per_page=100)
+
+    return render_template(
+        "form_demo.html", view=view_name, page=page, form=form, model=model
+    )