login improvements

routes/auth.py

@@ -20,6 +20,9 @@ def init_routes(auth_bp):
                 return redirect(url_for('auth.login'))
                 
             login_user(user, remember=remember)
+            next_page = request.args.get('next')
+            if next_page:
+                return redirect(next_page)
             return redirect(url_for('main.dashboard'))
             
         return render_template('auth/login.html')
@@ -50,8 +53,8 @@ def init_routes(auth_bp):
             db.session.add(new_user)
             db.session.commit()
             
-            flash('Registration successful! Please login.', 'success')
-            return redirect(url_for('auth.login'))
+            login_user(new_user)
+            return redirect(url_for('main.dashboard'))
             
         return render_template('auth/register.html')
 
@@ -59,4 +62,4 @@ def init_routes(auth_bp):
     @login_required
     def logout():
         logout_user()
-        return redirect(url_for('main.home')) 
+        return redirect(url_for('auth.login'))