How to redirect from the login portal using Flask

Question

I have been developing a pretty basic site, I have used tons of GPT but one issue both me and GPT have been unable to figure out is how after login, the site is not able to redirect. I am trying to redirect it to another page after a successful login. A JWT token is being generated but something I have not been able to figure out.

from flask import Flask, request, jsonify, render_template
from flask_sqlalchemy import SQLAlchemy
from flask_bcrypt import Bcrypt
from flask_jwt_extended import JWTManager, create_access_token, jwt_required, get_jwt_identity
from flask_cors import CORS
from flask_jwt_extended import verify_jwt_in_request, get_jwt_identity
import os

# Flask app configuration
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:#######@127.0.0.1/stockroom_db'
app.config['JWT_SECRET_KEY'] = 'your_jwt_secret_key'  # Replace with a strong secret key
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False  # Disable to save resources

# Initialize extensions
db = SQLAlchemy(app)
bcrypt = Bcrypt(app)
jwt = JWTManager(app)

# User model definition
class User(db.Model):
    __tablename__ = 'users'
    user_id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(50), unique=True, nullable=False)
    email = db.Column(db.String(100), unique=True, nullable=False)
    password_hash = db.Column(db.String(255), nullable=False)
    created_at = db.Column(db.DateTime, default=db.func.current_timestamp())

    def set_password(self, password):
        self.password_hash = bcrypt.generate_password_hash(password).decode('utf-8')

    def check_password(self, password):
        return bcrypt.check_password_hash(self.password_hash, password)

# Signup route for user registration
@app.route('/signup', methods=['POST'])
def signup():
    data = request.json
    username = data.get('username')
    email = data.get('email')
    password = data.get('password')

    # Check if username or email already exists
    if User.query.filter((User.username == username) | (User.email == email)).first():
        return jsonify({"error": "Username or Email already exists"}), 400

    # Create a new user instance
    new_user = User(username=username, email=email)
    new_user.set_password(password)
    
    # Save the new user to the database
    db.session.add(new_user)
    db.session.commit()

    return jsonify({"message": "User registered successfully"}), 201

# Login route for user authentication
@app.route('/login', methods=['POST'])
def login():
    data = request.json
    username = data.get('username')
    password = data.get('password')

    # Find the user by username
    user = User.query.filter_by(username=username).first()
    
    # Check if user exists and password is correct
    if user and user.check_password(password):
        # Create JWT token
        access_token = create_access_token(identity=user.user_id)
        return jsonify({"access_token": access_token}), 200

    return jsonify({"error": "Invalid username or password"}), 401

@app.before_request
def handle_options_request():
    if request.method == "OPTIONS":
        return jsonify({"status": "OK"}), 200


@app.route('/myStockroom', methods=['GET'])
@jwt_required()  # Enforce JWT validation
def my_stockroom():
    try:
        # Log the Authorization header
        auth_header = request.headers.get('Authorization')
        print(f"Authorization Header Received: {auth_header}")

        # Extract user ID from JWT
        user_id = get_jwt_identity()
        print(f"User ID from JWT: {user_id}")

        # Fetch user from the database
        user = User.query.get(user_id)
        if user:
            return jsonify({"username": user.username}), 200
        else:
            return jsonify({"error": "User not found"}), 404
    except Exception as e:
        print(f"Error in myStockroom route: {e}")
        return jsonify({"error": "Unauthorized access"}), 401

CORS(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True, expose_headers=["Authorization"])

# Main entry point for running the application
if __name__ == "__main__":
    # Create tables if they don't exist
    with app.app_context():
        db.create_all()
    app.run(host="127.0.0.1", port=5555, debug=True)

The above code is my app.py






Stockroom - Manage Your Inventory Seamlessly





    
        
    
    Manage your inventory, seamlessly.
    
      
      
  






  
    ×
    Sign Up
    Username
    
    Email
    
    Password
    
    
    
  




  
    ×
    Login
    Username
    
    Password

the above code is my login page






My Stockroom Inventory




    
        
        my stockroom
        inventory
        new order
        
            account
            homepage
            help
        
    
    
    
        My Stockroom
    
    
    
        Welcome, !
        
            
                DATE:
                
            
            
                ORDER#:
                
            
            
                CUSTOMERNAME:
                
            
            
                TOTAL ORDER VALUE (TOV):

This code is the page it should redirect to, please help me figure this redirection

How to redirect from the login portal using Flask

Answers (1)

Related Questions