56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from flask import Blueprint, redirect, url_for, request, flash, render_template
|
|
from flask_login import current_user
|
|
from .. import db
|
|
from app.models import Contributor
|
|
from app.forms import EditProfile, ChangePassword
|
|
|
|
prof = Blueprint(
|
|
"prof", __name__, template_folder="templates"
|
|
)
|
|
|
|
|
|
@prof.route("/change-password", methods=["GET", "POST"])
|
|
def change_password():
|
|
if not current_user.is_authenticated:
|
|
return(redirect(url_for('proute.index')))
|
|
contributor = Contributor.query.get(current_user.id)
|
|
form = ChangePassword()
|
|
if form.validate_on_submit():
|
|
if contributor.check_password(form.password.data):
|
|
contributor.set_password(form.new_password.data)
|
|
db.session.commit()
|
|
flash("Thanks for the update!")
|
|
return(redirect(url_for('proute.index')))
|
|
else:
|
|
flash("Error Invalid Password")
|
|
return(redirect(url_for('prof.change_password')))
|
|
return render_template('change_password.html', title='Change Password', form=form)
|
|
|
|
|
|
@prof.route("/edit-profile", methods=["GET", "POST"])
|
|
def edit_profile():
|
|
if current_user.is_anonymous:
|
|
return(redirect(url_for('proute.index')))
|
|
contributor = Contributor.query.get(current_user.id)
|
|
form = EditProfile()
|
|
if request.method == 'GET':
|
|
form.username.data = contributor.name
|
|
form.email.data = contributor.email
|
|
if form.validate_on_submit():
|
|
if contributor.check_password(form.password.data):
|
|
contributor.name = form.username.data
|
|
contributor.email = form.email.data
|
|
db.session.commit()
|
|
flash("Thanks for the update!")
|
|
return(redirect(url_for('proute.index')))
|
|
else:
|
|
flash("Error Invalid Password")
|
|
return(redirect(url_for('prof.edit_profile')))
|
|
return render_template(
|
|
'edit_profile.html',
|
|
title='Edit Profile', form=form,
|
|
contributor_use_totp=contributor.use_totp
|
|
)
|