from flask import request, redirect, url_for, flash
from werkzeug.security import generate_password_hash
from flask_login import login_user
 
@app.route("/register", methods=["GET", "POST"])
def register():
    if request.method == "POST":
        username = request.form.get("username", "").strip()
        email = request.form.get("email", "").strip().lower()
        password = request.form.get("password", "")
 
        if not username or not email or not password:
            flash("All fields are required", "error")
            return redirect(url_for("register"))
 
        if User.query.filter_by(username=username).first():
            flash("Username already taken", "error")
            return redirect(url_for("register"))
 
        if User.query.filter_by(email=email).first():
            flash("Email already registered", "error")
            return redirect(url_for("register"))
 
        user = User(
            username=username,
            email=email,
            password_hash=generate_password_hash(password),
        )
        db.session.add(user)
        db.session.commit()
 
        login_user(user)
        flash("Welcome! Your account was created.", "success")
        return redirect(url_for("dashboard"))
 
    return "Register page"

from flask import request, redirect, url_for, flash
from werkzeug.security import generate_password_hash
from flask_login import login_user
 
@app.route("/register", methods=["GET", "POST"])
def register():
    if request.method == "POST":
        username = request.form.get("username", "").strip()
        email = request.form.get("email", "").strip().lower()
        password = request.form.get("password", "")
 
        if not username or not email or not password:
            flash("All fields are required", "error")
            return redirect(url_for("register"))
 
        if User.query.filter_by(username=username).first():
            flash("Username already taken", "error")
            return redirect(url_for("register"))
 
        if User.query.filter_by(email=email).first():
            flash("Email already registered", "error")
            return redirect(url_for("register"))
 
        user = User(
            username=username,
            email=email,
            password_hash=generate_password_hash(password),
        )
        db.session.add(user)
        db.session.commit()
 
        login_user(user)
        flash("Welcome! Your account was created.", "success")
        return redirect(url_for("dashboard"))
 
    return "Register page"