def register():
"""
Register new user.
"""
form = RegisterForm(request.form)
if form.validate_on_submit():
u = User(
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
)
u.enforce_admin()
u.save()
if u.active:
flash("Thanks for registering! You're now logged in.", "success")
login_user(u)
else:
flash(
"Thanks for registering! You will now need to wait for your admin to approve this account.",
"success",
)
return redirect(url_for("public.home"))
else:
flash_errors(form)
return render_template("public/register.html", form=form)
def login():
"""Log in."""
form = LoginForm(request.form)
# Handle logging in
if request.method == 'POST':
if form.validate_on_submit():
login_user(form.user)
flash('Welcome {}! You are now logged in.'.format(current_user.first_name), 'success')
redirect_url = request.args.get('next') or url_for('public.home')
return redirect(redirect_url)
else:
flash_errors(form)
return render_template('public/login.html', form=form)
def register():
"""Register new user."""
form = RegisterForm(request.form, csrf_enabled=False)
if form.validate_on_submit():
user_id = (db.session.query(func.max(User.user_id)).scalar() or 0) + 1
User.create(user_id=user_id,
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
active=True)
flash("Thanks for registering! You're now logged in.", 'success')
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
def login():
"""
Log in.
"""
form = LoginForm(request.form)
# Handle logging in
if request.method == "POST":
if form.validate_on_submit():
login_user(form.user)
flash(
"Welcome {}! You are now logged in.".format(current_user.first_name),
"success",
)
redirect_url = request.args.get("next") or url_for("public.home")
return redirect(redirect_url)
else:
flash_errors(form)
return render_template("public/login.html", form=form)
def register():
"""Register new user."""
form = RegisterForm(request.form)
if form.validate_on_submit():
user_cnt = db.session.query(User).count()
u = User.create(
username=form.username.data,
email=form.email.data,
password=form.password.data,
first_name=form.first_name.data,
last_name=form.last_name.data,
active=True,
is_admin=True if user_cnt == 0 else False,
)
flash("Thanks for registering! You're now logged in.", 'success')
login_user(u)
return redirect(url_for('public.home'))
else:
flash_errors(form)
return render_template('public/register.html', form=form)
