Add boolean fields to example app

Author: insspb

example_app/app.py

@@ -3,6 +3,7 @@
 from pymongo import monitoring
 
 from example_app import views
+from example_app.boolean_demo import boolean_demo_view
 from example_app.dates_demo import dates_demo_view
 from example_app.models import db
 from example_app.numbers_demo import numbers_demo_view
@@ -49,6 +50,8 @@
 app.add_url_rule("/numbers/<pk>/", view_func=numbers_demo_view, methods=["GET", "POST"])
 app.add_url_rule("/dates", view_func=dates_demo_view, methods=["GET", "POST"])
 app.add_url_rule("/dates/<pk>/", view_func=dates_demo_view, methods=["GET", "POST"])
+app.add_url_rule("/bool", view_func=boolean_demo_view, methods=["GET", "POST"])
+app.add_url_rule("/bool/<pk>/", view_func=boolean_demo_view, methods=["GET", "POST"])
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=8000)

example_app/boolean_demo.py

@@ -0,0 +1,48 @@
+"""Booleans fields demo model."""
+
+from flask import render_template, request
+
+from example_app.models import db
+
+
+class BooleanDemoModel(db.Document):
+    """Documentation example model."""
+
+    simple_sting_name = db.StringField()
+    boolean_field = db.BooleanField()
+    boolean_field_with_null = db.BooleanField(null=True)
+    true_boolean_field_with_allowed_null = db.BooleanField(null=True, default=True)
+    boolean_field_with_as_choices_replace = db.BooleanField(
+        wtf_options={
+            "choices": [("", "Not selected"), ("yes", "Positive"), ("no", "Negative")]
+        }
+    )
+
+
+BooleanDemoForm = BooleanDemoModel.to_wtf_form()
+
+
+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,
+    )

example_app/templates/layout.html

@@ -21,6 +21,7 @@
     <li><a href="{{ url_for("strings_demo_view") }}">Strings demo</a></li>
     <li><a href="{{ url_for("numbers_demo_view") }}">Numbers demo</a></li>
     <li><a href="{{ url_for("dates_demo_view") }}">DateTime demo</a></li>
+    <li><a href="{{ url_for("boolean_demo_view") }}">Booleans demo</a></li>
   </ul>
 </nav>
 <div>

example_app/views.py

@@ -4,6 +4,7 @@
 from mongoengine.context_managers import switch_db
 
 from example_app import models
+from example_app.boolean_demo import BooleanDemoModel
 from example_app.dates_demo import DateTimeModel
 from example_app.numbers_demo import NumbersDemoModel
 from example_app.strings_demo import StringsDemoModel
@@ -50,6 +51,7 @@ def delete_data():
     """Clear database."""
     with switch_db(models.Todo, "default"):
         models.Todo.objects().delete()
+        BooleanDemoModel.objects().delete()
         DateTimeModel.objects().delete()
         StringsDemoModel.objects().delete()
         NumbersDemoModel.objects().delete()