def login():
form = forms.LoginForm()
if form.validate_on_submit():
# there is no use to display below info
# flash('Login requested for user {}, remember_me={}'.format(
# form.username.data, form.remember_me.data))
if login_valid(username=form.username.data, password=form.password.data, connection=connection):
flash(
"Welcome, {}! Current date and time is {}".format(form.username.data, mysql_helper.display_datetime()))
flash(f"Last logged in on {get_user_last_logged_in(form.username.data, connection)}.")
update_login_time_to_now(form.username.data, connection)
# superadmin
user = form.username.data
if user in superadmin: # TODO remove hard-coded list and use MySQL queries
flash("You have full access to database."
"Please be cautious while performing any actions")
flash(sysusenotification)
elif user in dataadmin:
flash("You have all privileges to "
"database except delete commands")
flash(sysusenotification)
else:
flash("Ha! You have no power here!")
flash(sysusenotification)
return redirect(url_for('main'))
return render_template('login.html', title="Sign in", form=form)
def login():
form = forms.LoginForm()
if form.validate_on_submit():
flash('Login requested for user {}, remember_me={}'.format(
form.username.data, form.remember_me.data))
if login_valid(username=form.username.data,
password=form.password.data,
connection=connection):
flash("Login is correct. Welcome, {}!".format(form.username.data))
update_login_time_to_now(form.username.data, connection)
return redirect(url_for('main'))
return render_template('login.html', title="Sign in", form=form)
def validate(self):
rv = FlaskForm.validate(self)
ret = True
if not rv:
ret = False
if not user_exists(self.username.data, connection):
self.username.errors.append("User does not exist.")
flash(" Please check the userID {}! ".format(self.username.data))
ret = False
elif not login_valid(self.username.data, self.password.data,
connection):
self.password.errors.append('Invalid password.')
flash("Hi, {}! you have entered an incorrect password".format(
self.username.data))
login_invalid(self.username.data, self.password.data, connection)
ret = False
if not ret:
flash("Login is incorrect.")
return ret
@app.route('/login', methods=['GET', 'POST'])
def login():
form = forms.LoginForm()
if form.validate_on_submit():
flash('Login requested for user {}, remember_me={}'.format(
form.username.data, form.remember_me.data))
if login_valid(username=form.username.data,
password=form.password.data,
connection=connection):
flash("Login is correct. Welcome, {}!".format(form.username.data))
update_login_time_to_now(form.username.data, connection)
return redirect(url_for('main'))
return render_template('login.html', title="Sign in", form=form)
if __name__ == '__main__':
app.run()
# Sanity checks.
assert (login_valid("henry", "iliketofarm", connection))
assert (not login_valid("henry", "notafarmer", connection))
assert (user_exists("henry", connection))
assert (not user_exists("tiffany_the_lord_of_squids", connection))