diff options
Diffstat (limited to 'app/routes/user.py')
| -rw-r--r-- | app/routes/user.py | 41 |
1 files changed, 22 insertions, 19 deletions
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")) |
