diff --git a/app.py b/app.py index c7436f2..83e9027 100755 --- a/app.py +++ b/app.py @@ -1,28 +1,52 @@ #!/usr/bin/env python -from flask import Flask, render_template, redirect, url_for, abort, request +from flask import Flask, render_template, redirect, url_for, request, session import openvpnstatus app = Flask(__name__) -@app.route('/status', methods=['GET', 'POST']) +@app.route("/") +def index(): + if "username" in session: + return render_template("home.html", username = session["username"]) + return render_template("login.html") + +@app.route("/login", methods = ["GET", "POST"]) +def login(): + if request.method == "POST": + session["username"] = request.form["username"] + if 'login' in request.referrer: + return redirect(url_for('index')) + else: + return redirect(request.referrer) + return render_template("login.html") + +@app.route("/logout") +def logout(): + session.pop("username", None) + return redirect(url_for("index")) + +@app.route("/status") def status(): - o = openvpnstatus.openvpnstatus() - return render_template('status.html', connections = o.connections, routes = o.routes) + if "username" in session: + o = openvpnstatus.openvpnstatus() + return render_template("status.html", connections = o.connections, routes = o.routes, username = session["username"]) + return render_template("login.html") -@app.route('/socialnerds') -def redir(): - return redirect('https://www.socialnerds.org') - -@app.route('/form', methods=['GET', 'POST']) +@app.route("/form", methods=["GET", "POST"]) def formular(): - something = "iatren" - if request.method == 'POST': - value = request.form['textfield'] - return render_template("footer.html", test=value, test2=value) - else: - return render_template("form.html") + if "username" in session: + something = "iatren" + if request.method == "POST": + value = request.form["textfield"] + return render_template("footer.html", test=value, test2=value) + else: + return render_template("form.html") + return render_template("login.html") -if __name__ == '__main__': +app.secret_key = 'A0Zr98j/3yX R~XHH!jmN]LWX/,?RT' + + +if __name__ == "__main__": app.run(debug=True) diff --git a/templates/footer.html b/templates/footer.html index c7c584f..5e0dc4d 100644 --- a/templates/footer.html +++ b/templates/footer.html @@ -5,6 +5,6 @@ Hello, my dear {{ test2 }}!