from app import db, logger from flask import abort, current_app, flash, redirect, render_template, url_for from flask_login import current_user, login_required, logout_user from . import profile from .background_functions import delete_user_ from .forms import ChangePasswordForm, EditProfileForm, EditUserSettingsForm import threading @profile.route('/', methods=['GET', 'POST']) @login_required def index(): """ View where loged in User can change own User information like Password etc. """ edit_user_info_form = EditProfileForm(user=current_user) edit_user_info_form.email.data = current_user.email return render_template('profile/index.html.j2', change_password_form=ChangePasswordForm(), edit_user_info_form=edit_user_info_form, edit_user_settings_form=EditUserSettingsForm(), title='Profile') @profile.route('/change_password', methods=['POST']) @login_required def profile_change_password(): edit_user_info_form = EditProfileForm(user=current_user) change_password_form = ChangePasswordForm() if change_password_form.validate_on_submit(): if current_user.verify_password(change_password_form.old_password.data): current_user.password = change_password_form.new_password.data db.session.add(current_user) db.session.commit() flash('Your password has been updated.') return render_template('profile/index.html.j2', change_password_form=change_password_form, edit_user_info_form=edit_user_info_form, edit_user_settings_form=EditUserSettingsForm(), title='Profile') else: flash('Invalid password.') return render_template('profile/index.html.j2', change_password_form=change_password_form, edit_user_info_form=edit_user_info_form, edit_user_settings_form=EditUserSettingsForm(), title='Profile') @profile.route('/edit_user_info', methods=['POST']) @login_required def profile_edit_user_info(): edit_user_info_form = EditProfileForm(user=current_user) if edit_user_info_form.validate_on_submit(): current_user.email = edit_user_info_form.email.data db.session.add(current_user._get_current_object()) db.session.commit() flash('Your email has been updated.') else: logger.warning('Form: {}'.format(edit_user_info_form.errors)) return render_template('profile/index.html.j2', change_password_form=ChangePasswordForm(), edit_user_info_form=edit_user_info_form, edit_user_settings_form=EditUserSettingsForm(), title='Profile') edit_user_info_form.email.data = current_user.email return render_template('profile/index.html.j2', change_password_form=ChangePasswordForm(), edit_user_info_form=EditProfileForm(user=current_user), edit_user_settings_form=EditUserSettingsForm(), title='Profile') @profile.route('/edit_user_settings', methods=['POST']) @login_required def profile_edit_user_settings(): edit_user_settings_form = EditUserSettingsForm() if not edit_user_settings_form.validate_on_submit(): abort(400) current_user.is_dark = edit_user_settings_form.is_dark.data logger.warning('Form data: {}'.format(current_user.is_dark)) db.session.add(current_user) db.session.commit() if current_user.is_dark is True: flash('Dark mode has been activated!') else: flash('Dark mode has been deactivated!') return redirect(url_for('profile.index')) @profile.route('/delete_self', methods=['GET', 'POST']) @login_required def delete_self(): """ View to delete yourslef and all associated data. """ delete_thread = threading.Thread(target=delete_user_, args=(current_app._get_current_object(), current_user.id)) delete_thread.start() logout_user() flash('Your account has been deleted!') return redirect(url_for('main.index'))