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
