Initial commit

app/server/routes/verify_code.py

@@ -0,0 +1,32 @@
+# ------- 3rd party imports -------
+from flask import Blueprint, render_template, request, redirect, url_for, session
+
+# ------- local imports -------
+from app.server.db.extensions import db
+from app.server.db.models import VerificationCode, AuthToken, Email
+
+verify_code_blueprint = Blueprint('verify_code_blueprint', __name__, template_folder='templates')
+
+
+@verify_code_blueprint.route('/verify', methods=['GET', 'POST'])
+def enter_verification_code():
+    is_verified = request.args.get('is_verified')
+    is_code_valid = request.args.get('is_code_valid')
+    return render_template('verify.html', is_code_valid=is_code_valid, is_verified=is_verified)
+
+
+@verify_code_blueprint.route('/validate_code', methods=['POST'])
+def validate_code():
+    input_code = request.form['verification']
+    code = VerificationCode.query.filter_by(verification_code=input_code).first()
+
+    if code:
+        email = Email.query.filter_by(email=session['user_email']).first()
+        email.is_verified = True
+        auth_token = AuthToken(email=email)
+        db.session.add(auth_token)
+        db.session.commit()
+        return redirect(url_for('your_api_token_blueprint.your_api_token', auth_token=auth_token))
+
+    else:
+        return redirect(url_for('verify_code_blueprint.enter_verification_code', is_code_valid=False))