todo_app_flask/app/auth/profile.py

68 lines
2.2 KiB
Python
Raw Normal View History

2021-01-20 01:36:22 -08:00
#!/usr/bin/env python3
from flask import Blueprint, redirect, url_for, request, flash, render_template
from flask_login import current_user
from app.models import Contributor
from app.forms import EditProfile, ChangePassword
from .. import db
prof = Blueprint(
"prof", __name__, template_folder="templates"
)
@prof.route("/edit-profile", methods=["GET", "POST"])
def edit_profile():
if current_user.is_anonymous:
return(redirect(url_for('cats.index')))
contributor = Contributor.query.get(current_user.id)
form = EditProfile()
navbar_links = (('cancel', url_for('cats.index')), )
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('cats.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,
navbar_links=navbar_links
)
@prof.route("/change-password", methods=["GET", "POST"])
def change_password():
if not current_user.is_authenticated:
return(redirect(url_for('cats.index')))
contributor = Contributor.query.get(current_user.id)
form = ChangePassword()
nl = (('cancel', url_for('prof.edit_profile')), )
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('cats.index')))
else:
flash("Error Invalid Password")
return(redirect(url_for('prof.change_password')))
return render_template(
'change_password.html',
title='Change Password',
form=form, navbar_links=nl
)