File size: 2,201 Bytes
1106e39 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
from flask import Flask, render_template, request, redirect, url_for, session, flash
from werkzeug.security import generate_password_hash, check_password_hash
app = Flask(__name__)
app.secret_key = "supersecretkey" # change this in production!
# Simple in-memory "database"
users = {}
@app.route("/")
def home():
if "user" in session:
return redirect(url_for("dashboard"))
return redirect(url_for("login"))
@app.route("/signup", methods=["GET", "POST"])
def signup():
if request.method == "POST":
username = request.form["username"]
password = request.form["password"]
confirm_password = request.form.get("confirmPassword")
if password != confirm_password:
flash("Passwords do not match.", "danger")
return redirect(url_for("signup"))
if username in users:
flash("Username already exists!", "danger")
return redirect(url_for("signup"))
users[username] = generate_password_hash(password)
flash("Signup successful! Please login.", "success")
return redirect(url_for("login"))
return render_template("signup.html")
@app.route("/login", methods=["GET", "POST"])
def login():
if request.method == "POST":
username = request.form["username"]
password = request.form["password"]
if username in users and check_password_hash(users[username], password):
session["user"] = username
return redirect(url_for("dashboard"))
else:
flash("Invalid username or password", "danger")
return render_template("login.html")
@app.route("/dashboard")
def dashboard():
if "user" not in session:
return redirect(url_for("login"))
return render_template("dashboard.html", username=session["user"])
@app.route("/portfolio")
def portfolio():
if "user" not in session:
return redirect(url_for("login"))
return render_template("portfolio.html")
@app.route("/logout")
def logout():
session.pop("user", None)
flash("Logged out successfully.", "info")
return redirect(url_for("login"))
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860, debug=True)
|