def loginButton():
email = request.form["email"]
password = request.form["password"]
user = query_user(email)
if user != None:
if check_password_hash(user.password, password):
session['email'] = email
returnUrl = session.pop('return_url', None)
if returnUrl:
if 'diagram' in request.args:
return redirect(returnUrl + '?diagram=true')
return redirect(returnUrl)
else:
return redirect('/')
flash('Incorrect Email/Password')
return redirect('/login')
def signUpButton():
email = request.form["email"]
user = query_user(email)
if user == None:
password = request.form["password"]
password_hash = generate_password_hash(password)
insert_user(email, password_hash)
session['email'] = email
returnUrl = session.pop('return_url', None)
if returnUrl:
return redirect(returnUrl)
else:
return redirect('/')
# email has been used
flash('Email already in use')
return redirect('/signup')
