From 1c2a9169e663909991b1d495d998524e22398d3d Mon Sep 17 00:00:00 2001 From: pluja Date: Thu, 27 Aug 2020 10:15:20 +0200 Subject: [PATCH] Fix #13 --- app/routes.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/app/routes.py b/app/routes.py index 37cd3a1..30659c8 100644 --- a/app/routes.py +++ b/app/routes.py @@ -437,15 +437,16 @@ def register(): form = RegistrationForm() if form.validate_on_submit(): - if isTwitterUser(form.username.data): - flash('This is username is taken! Choose a different one.') - else: - user = User(username=form.username.data) - user.set_password(form.password.data) - db.session.add(user) - db.session.commit() - flash('Congratulations, you are now a registered user!') - return redirect(url_for('login')) + if User.query.filter_by(username=form.username.data).first(): + flash("This username is taken! Try with another.") + return redirect(request.referrer) + + user = User(username=form.username.data) + user.set_password(form.password.data) + db.session.add(user) + db.session.commit() + flash('Congratulations, you are now a registered user!') + return redirect(url_for('login')) return render_template('register.html', title='Register', form=form) @app.route('/error/')