47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from flask import Blueprint, redirect, url_for, flash, render_template, current_app
|
|
from flask_login import current_user
|
|
from app.models import Contributor
|
|
from app.forms import ResetPasswordForm, ResetPasswordRequestForm
|
|
from .email import send_password_reset_email
|
|
from .. import db
|
|
|
|
pwd = Blueprint(
|
|
"pwd", __name__, template_folder="templates"
|
|
)
|
|
|
|
|
|
@pwd.route('/reset-password/<token>', methods=['GET', 'POST'])
|
|
def reset_password(token):
|
|
if current_user.is_authenticated:
|
|
return redirect(url_for('proute.index'))
|
|
contributor = Contributor.verify_reset_password_token(token)
|
|
if not contributor:
|
|
return redirect(url_for('proute.index'))
|
|
form = ResetPasswordForm()
|
|
if form.validate_on_submit():
|
|
contributor.set_password(form.password.data)
|
|
db.session.commit()
|
|
flash('Your password has been reset.')
|
|
return redirect(url_for('auths.login'))
|
|
return render_template('reset_password.html', title="New Password?", form=form)
|
|
|
|
|
|
@pwd.route('/reset-password-request', methods=['GET', 'POST'])
|
|
def reset_password_request():
|
|
if current_user.is_authenticated:
|
|
return(redirect(url_for('proute.index')))
|
|
else:
|
|
form = ResetPasswordRequestForm()
|
|
if form.validate_on_submit():
|
|
contributor = Contributor.query.filter_by(email=form.email.data).first()
|
|
if contributor:
|
|
send_password_reset_email(contributor, current_app.config['EXTERNAL_URL'])
|
|
flash('Check your email for the instructions to reset your password')
|
|
return redirect(url_for('auths.login'))
|
|
else:
|
|
flash('Sorry, invalid email')
|
|
return redirect(url_for('auths.login'))
|
|
return render_template('reset_password_request.html', title='Reset Password', form=form)
|