Skip to content

Commit

Permalink
fix logout route
Browse files Browse the repository at this point in the history
  • Loading branch information
unhackableandy committed Oct 19, 2023
1 parent af2ad43 commit d803292
Showing 1 changed file with 16 additions and 1 deletion.
17 changes: 16 additions & 1 deletion app.py
Expand Up @@ -18,13 +18,15 @@
app.config["SECRET_KEY"] = "your_secret_key_here"
app.config["SESSION_TYPE"] = "filesystem"


class LoginForm(FlaskForm):
username = StringField(
"Username", validators=[DataRequired(), Length(min=2, max=20)]
)
password = PasswordField("Password", validators=[DataRequired()])
submit = SubmitField("Login")


class DashboardForm(FlaskForm):
command = StringField(
"Enter command to reset server:",
Expand All @@ -33,21 +35,27 @@ class DashboardForm(FlaskForm):
)
submit = SubmitField("Submit")


@app.route("/")
def home():
return render_template("home.html")


@app.route("/login", methods=["GET", "POST"])
def login():
form = LoginForm()
if form.validate_on_submit():
if form.username.data == ADMIN_USERNAME and form.password.data == ADMIN_PASSWORD:
if (
form.username.data == ADMIN_USERNAME
and form.password.data == ADMIN_PASSWORD
):
session["logged_in"] = True
return redirect(url_for("dashboard"))
else:
flash("Login Unsuccessful. Please check username and password", "danger")
return render_template("login.html", title="Login", form=form)


@app.route("/dashboard", methods=["GET", "POST"])
def dashboard():
if not session.get("logged_in"):
Expand Down Expand Up @@ -79,5 +87,12 @@ def dashboard():

return render_template("dashboard.html", form=form, stats=stats, output=output)


@app.route("/logout")
def logout():
session.pop("logged_in", None)
return redirect(url_for("home"))


if __name__ == "__main__":
app.run(debug=True)

0 comments on commit d803292

Please sign in to comment.