mirror of
https://github.com/kevin-DL/ShortMe-URL-Shortener.git
synced 2026-01-11 19:14:29 +00:00
33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
# ------- 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))
|