nopaque/app/profile/views.py

from flask import flash, redirect, render_template, url_for
from flask_login import current_user, login_required, logout_user
from . import profile
from . import tasks
from .forms import EditEmailForm, EditGeneralSettingsForm, EditPasswordForm
from .. import db


@profile.route('/settings', methods=['GET', 'POST'])
@login_required
def settings():
    edit_email_form = EditEmailForm(prefix='edit-email-form')
    edit_general_settings_form = EditGeneralSettingsForm(
        prefix='edit-general-settings-form')
    edit_password_form = EditPasswordForm(prefix='edit-password-form',
                                          user=current_user)
    # Check if edit_email_form is submitted and valid
    if (edit_email_form.save_email.data
            and edit_email_form.validate_on_submit()):
        db.session.add(current_user)
        db.session.commit()
        flash('Your email address has been updated.')
        return redirect(url_for('profile.settings'))
    # Check if edit_settings_form is submitted and valid
    if (edit_general_settings_form.save_settings.data
            and edit_general_settings_form.validate_on_submit()):
        current_user.setting_dark_mode = \
            edit_general_settings_form.dark_mode.data
        current_user.setting_job_status_mail_notifications = \
            edit_general_settings_form.job_status_mail_notifications.data
        current_user.setting_job_status_site_notifications = \
            edit_general_settings_form.job_status_site_notifications.data
        db.session.add(current_user)
        db.session.commit()
        flash('Your settings have been updated.')
        return redirect(url_for('profile.settings'))
    # Check if edit_password_form is submitted and valid
    if (edit_password_form.save_password.data
            and edit_password_form.validate_on_submit()):
        current_user.password = edit_password_form.password.data
        db.session.add(current_user)
        db.session.commit()
        flash('Your password has been updated.')
        return redirect(url_for('profile.settings'))
    # If no form is submitted or valid, fill out fields with current values
    edit_email_form.email.data = current_user.email
    edit_general_settings_form.dark_mode.data = current_user.setting_dark_mode
    edit_general_settings_form.job_status_site_notifications.data = \
        current_user.setting_job_status_site_notifications
    edit_general_settings_form.job_status_mail_notifications.data = \
        current_user.setting_job_status_mail_notifications
    return render_template(
        'profile/settings.html.j2',
        edit_email_form=edit_email_form,
        edit_password_form=edit_password_form,
        edit_general_settings_form=edit_general_settings_form,
        title='Settings')


@profile.route('/delete', methods=['GET', 'POST'])
@login_required
def delete():
    """
    View to delete yourslef and all associated data.
    """
    tasks.delete_user(current_user.id)
    logout_user()
    flash('Your account has been deleted!')
    return redirect(url_for('main.index'))