Advertisement
Guest User

Register.py

a guest
Nov 30th, 2017
89
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.63 KB | None | 0 0
  1. @app.route("/register", methods=["GET", "POST"])
  2. def register():
  3. """Register user."""
  4.  
  5. if request.method == "POST":
  6.  
  7. # ensure username was submitted
  8. if not request.form.get("username"):
  9. return apology("Missing username!")
  10.  
  11. # ensure password was submitted
  12. elif not request.form.get("password"):
  13. return apology("Missing password!")
  14.  
  15. # ensure password and confirmation match
  16. elif not request.form.get("password") == request.form.get("2nd_password"):
  17. return apology("Your password and confirmation doesn't match")
  18.  
  19. # to store password in hash
  20. hashed_password = pwd_context.hash(request.form.get("password"))
  21.  
  22. # check if username already exists
  23. result = db.execute("SELECT * FROM users WHERE username = :username", username=request.form.get("username"))
  24. if result[0]["username"] == request.form.get("username"):
  25. return apology("Username already exists")
  26. else:
  27. db.execute("INSERT INTO users (username, hash) VALUES(:username, :hash)", username=request.form.get("username"), hash=hashed_password)
  28.  
  29. row = db.execute("SELECT * FROM users WHERE username = :username", username=request.form.get("username"))
  30.  
  31. # store registered user in session
  32. session["user_id"] = row[0]["id"]
  33.  
  34. # redirect user to home page
  35. return redirect(url_for("index"))
  36.  
  37.  
  38. # else if user reached route via GET (as by clicking a link or via redirect)
  39. else:
  40. return render_template("register.html")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement