def load_user(user_id):
result = User.get(user_id)
if result:
name, display_name, password_hash = User.user_database[user_id]
return User(name, display_name, password_hash)
else:
return None
def login():
error = None
form = LoginForm(request.form)
if request.method == "POST":
if form.validate_on_submit():
username = request.form["username"]
password = request.form["password"]
if (not username in User.user_database) or gen_hash(password) != User.get_password_hash(username):
error = "Invalid Credentials. Try again."
else:
login_user(load_user(username))
flash("You were just logged in")
return redirect(url_for("home"))
else:
return render_template("login.html", form=form, error=error)
return render_template("login.html", form=form, error=error)
def getUser(connection, userID):
"""Get a User object given a user ID"""
cursor = connection.cursor()
columns = "usr, name, email, city, timezone"
variables = (userID, )
inputSizes = [int]
statement = "select {0} from {1} where usr = :1".format(
columns, UsersTableTools._USERS_TABLE)
TableTools.execute(connection, cursor, statement, variables,
inputSizes)
result = cursor.fetchone()
return User(result[0], result[1], result[2], result[3], result[4])