Rework the controllers.

Goutte
1 parent 3b18543e
Showing 2 changed files with 50 additions and 13 deletions Show diff stats
flaskr/controllers/main_controller.py
flaskr/controllers/main.py -> flaskr/controllers/user_controller.py
@@ -0,0 +1,44 @@
+from flask import Blueprint, render_template, flash, request, redirect, url_for
+
+from flaskr.extensions import cache
+from flaskr.forms import LoginForm, EstimateForm
+from flaskr.models import db, User, Estimation
+
+from flaskr.core import generate_unique_id
+
+main = Blueprint('main', __name__)
+
+
+@main.route('/')
+@cache.cached(timeout=1000)
+def home():
+    return render_template('index.html')
+
+
+@main.route("/estimate", methods=["GET", "POST"])
+def estimate():
+    form = EstimateForm()
+
+    if form.validate_on_submit():
+
+        # FIXME: do things here with the form
+
+        id = generate_unique_id()
+
+        # estimation = form.data
+
+        estimation = Estimation()
+        estimation.email = form.email.data
+        estimation.first_name = form.first_name.data
+        estimation.last_name = form.last_name.data
+        estimation.status = 'pending'
+
+        db.session.add(estimation)
+        db.session.commit()
+
+        flash("Estimation request submitted successfully.", "success")
+        return redirect(url_for(".home"))
+        # return render_template("estimate-debrief.html", form=form)
+
+    return render_template("estimate.html", form=form)
+
 from flask import Blueprint, render_template, flash, request, redirect, url_for
 from flask_login import login_user, logout_user, login_required
  
-from flaskr.extensions import cache
 from flaskr.forms import LoginForm
-from flaskr.models import User
+from flaskr.models import db, User
  
-main = Blueprint('main', __name__)
-
-
-@main.route('/')
-@cache.cached(timeout=1000)
-def home():
-    return render_template('index.html')
+from flaskr.controllers.main_controller import main
  
  
 @main.route("/login", methods=["GET", "POST"])
@@ -36,7 +29,7 @@ def logout():
     return redirect(url_for(".home"))
  
  
-@main.route("/restricted")
-@login_required
-def restricted():
-    return "You can only see this if you are logged in!", 200
+# @main.route("/restricted")
+# @login_required
+# def restricted():
+#     return "You can only see this if you are logged in!", 200