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)