fixing django admin redirect for django v4.8.0 and above

eadwinCode · eadwinCode · commit be79705e81f5 · 2024-03-05T00:20:11.000+01:00
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -14,7 +14,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: 3.8
+          python-version: 3.10
       - name: Install Flit
         run: pip install flit
       - name: Install Dependencies
diff --git a/ellar_django/middleware.py b/ellar_django/middleware.py
@@ -0,0 +1,34 @@
+import typing as t
+
+from ellar.common.types import ASGIApp, TMessage, TReceive, TScope, TSend
+from starlette.datastructures import MutableHeaders
+
+
+class DjangoAdminRedirectMiddleware:
+    def __init__(self, app: ASGIApp, path_prefix: str) -> None:
+        self.app = app
+        self.path_prefix = path_prefix
+
+    async def __call__(self, scope: TScope, receive: TReceive, send: TSend) -> t.Any:
+        if scope["type"] != "http":
+            return await self.app(scope, receive, send)
+
+        async def sender(message: TMessage) -> None:
+            if message["type"] == "http.response.start":
+                headers = message.get("headers")
+                if headers:
+                    working_headers = MutableHeaders(
+                        raw=[
+                            (key.decode("latin-1").lower().encode("latin-1"), value)
+                            for key, value in headers
+                        ]
+                    )
+                    location = working_headers.get("location")
+                    if location and not location.startswith(self.path_prefix):
+                        working_headers["Location"] = (
+                            f"{self.path_prefix}{working_headers['Location']}"
+                        )
+                        message["headers"] = working_headers.raw
+            await send(message)
+
+        return await self.app(scope, receive, sender)
diff --git a/ellar_django/module.py b/ellar_django/module.py
@@ -8,6 +8,7 @@
 from starlette.routing import Mount
 
 from .commands import django_command_wrapper
+from .middleware import DjangoAdminRedirectMiddleware
 
 _router = ModuleRouter()
 
@@ -28,7 +29,12 @@ def setup(cls, settings_module: str, path_prefix: str = "/dj") -> "DynamicModule
             path_prefix,
             routes=[
                 _router_as_mount,
-                Mount("/", app=get_asgi_application()),
+                Mount(
+                    "/",
+                    app=DjangoAdminRedirectMiddleware(
+                        get_asgi_application(), path_prefix
+                    ),
+                ),
             ],
         )
         return DynamicModule(cls, routers=[mount])
