maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

safer check for existing user in tutorial

Browse source 
Co-authored-by: David Lord <davidism@gmail.com>
This commit is contained in:
Angeline 2021-06-07 22:09:58 +08:00 • committed by David Lord
parent 50b7dcbab3
commit 5119657547
No known key found for this signature in database
GPG key ID: 7A1C87E3F5BC42A8
2 changed files with 46 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

26
examples/tutorial/flaskr/auth.py
View file

@ -60,21 +60,21 @@ def register():
error = "Username is required."
elif not password:
error = "Password is required."
elif (
db.execute("SELECT id FROM user WHERE username = ?", (username,)).fetchone()
is not None
):
error = f"User {username} is already registered."
if error is None:
# the name is available, store it in the database and go to
# the login page
db.execute(
"INSERT INTO user (username, password) VALUES (?, ?)",
(username, generate_password_hash(password)),
)
db.commit()
return redirect(url_for("auth.login"))
try:
db.execute(
"INSERT INTO user (username, password) VALUES (?, ?)",
(username, generate_password_hash(password)),
)
db.commit()
except db.IntegrityError:
# The username was already taken, which caused the
# commit to fail. Show a validation error.
error = f"User {username} is already registered."
else:
# Success, go to the login page.
return redirect(url_for("auth.login"))
flash(error)