def add_user_to_db(user_data): # First salt and hash the password pw_hash = User.get_pw_hash(user_data['password']) conn = sqlite3.connect(app.config['DATABASE']) c = conn.cursor() data = (user_data['username'], pw_hash, user_data['email'], user_data['display'], ) c.execute(INSERT_QUERY, data) conn.commit() conn.close()
def do_account_update(): messages = {} new_display = request.form.get('display') new_email = request.form.get('email') new_password = request.form.get('password') confirm = request.form.get('confirm') if display_taken(new_display): messages['error'] = errors.get("display_in_use") else: conn = sqlite3.connect(app.config['DATABASE']) c = conn.cursor() if new_password: if new_password != confirm: messages['error'] = errors.get("pass_match") else: pw_hash = User.get_pw_hash(password) data = (pw_hash, current_user.get_user_id()) c.execute(UPDATE_PW_QUERY, data) data = (new_display, new_email, current_user.get_user_id()) c.execute(UPDATE_QUERY, data) conn.commit() conn.close() return messages