From ddcb213051bf3f7ff58b972bf4a6fab49526ccd2 Mon Sep 17 00:00:00 2001 From: FivePixels Date: Fri, 18 Nov 2022 09:29:26 -0600 Subject: Add config to allow registrations --- app/routes/user.py | 41 ++++++++++++++++++++++------------------- 1 file changed, 22 insertions(+), 19 deletions(-) (limited to 'app/routes/user.py') diff --git a/app/routes/user.py b/app/routes/user.py index 7044366..f56f81b 100644 --- a/app/routes/user.py +++ b/app/routes/user.py @@ -34,22 +34,25 @@ def logout(): @app.route("/register", methods=["GET", "POST"]) def register(): - form = RegisterForm() - # Load data from form - username = form.username.data - password = form.password.data - email = form.email.data - - if form.validate_on_submit(): - # Valid submission - user = User.query.filter_by(username=username).first() - if not user: - # No user with this username - user = User(username, password, email) - db.session.add(user) - db.session.commit() - return redirect(url_for("login")) - else: - flash("User already exists", "danger") - - return render_template("user/register.html", form=form) + if (app.config["ALLOW_REGISTER"]): + form = RegisterForm() + # Load data from form + username = form.username.data + password = form.password.data + email = form.email.data + + if form.validate_on_submit(): + # Valid submission + user = User.query.filter_by(username=username).first() + if not user: + # No user with this username + user = User(username, password, email) + db.session.add(user) + db.session.commit() + return redirect(url_for("login")) + else: + flash("User already exists", "danger") + + return render_template("user/register.html", form=form) + else: + return redirect(url_for("index")) -- cgit v1.2.3