from django.shortcuts import redirect
from django.contrib.auth import login

class AdminLoginRedirectMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        response = self.get_response(request)

        # Check if user is authenticated and just logged in via /admin/login/
        if request.path == "/admin/login/" and request.user.is_authenticated:
            # Set session ID in the session
            request.session["sessionid"] = request.session.session_key
            request.session.modified = True  # Ensure session is saved

            # Redirect user to '/'
            return redirect("/")

        return response