initial login logic

routes.py

@@ -0,0 +1,64 @@
+from flask import Blueprint, render_template, redirect, url_for, flash, jsonify, current_app
+from flask_login import login_user, login_required, current_user
+
+from models import db, User
+from forms import SignupForm
+from altcha_utils import get_altcha
+
+main = Blueprint('main', __name__)
+
+
+@main.route('/altcha/challenge')
+def altcha_challenge():
+    """Generate a new Altcha challenge for the signup form."""
+    altcha = get_altcha(current_app.config['ALTCHA_HMAC_KEY'])
+    challenge = altcha.create_challenge()
+    return jsonify(challenge)
+
+
+@main.route('/')
+def index():
+    if current_user.is_authenticated:
+        return redirect(url_for('main.dashboard'))
+    return redirect(url_for('main.signup'))
+
+
+@main.route('/signup', methods=['GET', 'POST'])
+def signup():
+    if current_user.is_authenticated:
+        return redirect(url_for('main.dashboard'))
+
+    form = SignupForm()
+
+    if form.validate_on_submit():
+        # Verify Altcha solution
+        altcha = get_altcha(current_app.config['ALTCHA_HMAC_KEY'])
+        if not altcha.verify_solution(form.altcha.data):
+            flash('CAPTCHA verification failed. Please try again.', 'error')
+            return render_template('signup.html', form=form)
+
+        # Check if user already exists
+        existing_user = User.query.filter_by(email=form.email.data.lower()).first()
+        if existing_user:
+            flash('An account with this email already exists.', 'error')
+            return render_template('signup.html', form=form)
+
+        # Create new user
+        user = User(email=form.email.data.lower())
+        user.set_password(form.password.data)
+
+        db.session.add(user)
+        db.session.commit()
+
+        # Auto-login after signup
+        login_user(user)
+        flash('Account created successfully!', 'success')
+        return redirect(url_for('main.dashboard'))
+
+    return render_template('signup.html', form=form)
+
+
+@main.route('/dashboard')
+@login_required
+def dashboard():
+    return render_template('dashboard.html')