` and re-run."
)
app = Flask(__name__)
app.secret_key = os.environ.get("FLASK_SECRET_KEY") or secrets.token_hex(32)
app.permanent_session_lifetime = timedelta(days=30)
def login_required(view):
@wraps(view)
def wrapped(*args, **kwargs):
if not session.get("logged_in"):
return redirect(url_for("login", next=request.path))
return view(*args, **kwargs)
return wrapped
@app.route("/login", methods=["GET", "POST"])
def login():
error = None
if request.method == "POST":
u = request.form.get("username", "")
p = request.form.get("password", "")
if hmac.compare_digest(u, USERNAME) and hmac.compare_digest(p, PASSWORD):
session["logged_in"] = True
session.permanent = True
return redirect(request.args.get("next") or url_for("index"))
error = "Invalid username or password."
elif session.get("logged_in"):
return redirect(url_for("index"))
return render_template_string(LOGIN_TEMPLATE, error=error)
@app.route("/logout")
def logout():
session.clear()
return redirect(url_for("login"))
@app.route("/")
@login_required
def index():
if not DASHBOARD_PATH.exists():
abort(
503,
description=(
"dashboard.html has not been generated yet. "
"Run `python run.py` first."
),
)
return send_file(DASHBOARD_PATH, mimetype="text/html")
@app.route("/healthz")
def healthz():
return ("ok", 200) if DASHBOARD_PATH.exists() else ("dashboard missing", 503)
LOGIN_TEMPLATE = """
Sign in - Funnel Tracker
"""
if __name__ == "__main__":
port = int(os.environ.get("PORT", 5000))
app.run(host="0.0.0.0", port=port, debug=False)