litestar-vite 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

litestar_vite/inertia/exception_handler.py

@@ -18,6 +18,7 @@ from litestar.exceptions.responses import (
 )
 from litestar.plugins.flash import flash
 from litestar.repository.exceptions import (
+    ConflictError,  # pyright: ignore[reportUnknownVariableType,reportAttributeAccessIssue]
     NotFoundError,  # pyright: ignore[reportUnknownVariableType,reportAttributeAccessIssue]
     RepositoryError,  # pyright: ignore[reportUnknownVariableType,reportAttributeAccessIssue]
 )
@@ -47,13 +48,14 @@ class _HTTPConflictException(HTTPException):
 def exception_to_http_response(request: Request[UserT, AuthT, StateT], exc: Exception) -> Response[Any]:
     """Handler for all exceptions subclassed from HTTPException."""
     inertia_enabled = getattr(request, "inertia_enabled", False) or getattr(request, "is_inertia", False)
-    if isinstance(exc, NotFoundError):
-        http_exc = NotFoundException
-    elif isinstance(exc, RepositoryError):
-        http_exc = _HTTPConflictException  # type: ignore[assignment]
-    else:
-        http_exc = InternalServerException  # type: ignore[assignment]
+
     if not inertia_enabled:
+        if isinstance(exc, NotFoundError):
+            http_exc = NotFoundException
+        elif isinstance(exc, (RepositoryError, ConflictError)):
+            http_exc = _HTTPConflictException  # type: ignore[assignment]
+        else:
+            http_exc = InternalServerException  # type: ignore[assignment]
         if request.app.debug and http_exc not in (PermissionDeniedException, NotFoundError):
             return cast("Response[Any]", create_debug_response(request, exc))
         return cast("Response[Any]", create_exception_response(request, http_exc(detail=str(exc.__cause__))))
@@ -88,16 +90,13 @@ def create_inertia_exception_response(request: Request[UserT, AuthT, StateT], ex
         return InertiaBack(request)
     if isinstance(exc, PermissionDeniedException):
         return InertiaBack(request)
-    if status_code == HTTP_401_UNAUTHORIZED or isinstance(exc, NotAuthorizedException):
-        if (
-            inertia_plugin.config.redirect_unauthorized_to is not None
-            and str(request.url) != inertia_plugin.config.redirect_unauthorized_to
-        ):
-            return InertiaRedirect(request, redirect_to=inertia_plugin.config.redirect_unauthorized_to)
-        if str(request.url) != inertia_plugin.config.redirect_unauthorized_to:
-            return InertiaResponse[Any](
-                media_type=preferred_type,
-                content=content,
-                status_code=status_code,
-            )
-    return InertiaBack(request)
+    if (status_code == HTTP_401_UNAUTHORIZED or isinstance(exc, NotAuthorizedException)) and (
+        inertia_plugin.config.redirect_unauthorized_to is not None
+        and str(request.url) != inertia_plugin.config.redirect_unauthorized_to
+    ):
+        return InertiaRedirect(request, redirect_to=inertia_plugin.config.redirect_unauthorized_to)
+    return InertiaResponse[Any](
+        media_type=preferred_type,
+        content=content,
+        status_code=status_code,
+    )

litestar_vite/inertia/request.py

@@ -114,3 +114,13 @@ class InertiaRequest(Request[UserT, AuthT, StateT]):
     def inertia_enabled(self) -> bool:
         """True if the route handler contains an inertia enabled configuration."""
         return bool(self.inertia.route_component is not None)
+
+    @property
+    def is_partial_render(self) -> bool:
+        """True if the route handler contains an inertia enabled configuration."""
+        return self.inertia.is_partial_render
+
+    @property
+    def partial_keys(self) -> set[str]:
+        """True if the route handler contains an inertia enabled configuration."""
+        return set(self.inertia.partial_keys)

litestar_vite/inertia/response.py

@@ -33,7 +33,6 @@ if TYPE_CHECKING:
     from litestar.connection.base import AuthT, StateT, UserT
     from litestar.types import ResponseCookies, ResponseHeaders, TypeEncodersMap
 
-    from litestar_vite.inertia.request import InertiaRequest
     from litestar_vite.inertia.routes import Routes
 
     from .plugin import InertiaPlugin
@@ -65,7 +64,10 @@ def error(
         connection.logger.warning(msg)
 
 
-def get_shared_props(request: ASGIConnection[Any, Any, Any, Any]) -> Dict[str, Any]:  # noqa: UP006
+def get_shared_props(
+    request: ASGIConnection[Any, Any, Any, Any],
+    partial_data: set[str] | None = None,
+) -> Dict[str, Any]:  # noqa: UP006
     """Return shared session props for a request
 
 
@@ -217,9 +219,11 @@ class InertiaResponse(Response[T]):
                 removal_in="3.0.0",
                 alternative="request.app",
             )
-        inertia_enabled = getattr(request, "inertia_enabled", False) or getattr(request, "is_inertia", False)
-        is_inertia = getattr(request, "is_inertia", False)
-
+        inertia_enabled = cast(
+            "bool",
+            getattr(request, "inertia_enabled", False) or getattr(request, "is_inertia", False),
+        )
+        is_inertia = cast("bool", getattr(request, "is_inertia", False))
         headers = {**headers, **self.headers} if headers is not None else self.headers
         cookies = self.cookies if cookies is None else itertools.chain(self.cookies, cookies)
         type_encoders = (
@@ -238,15 +242,18 @@ class InertiaResponse(Response[T]):
                 media_type=media_type,
                 status_code=self.status_code or status_code,
             )
+        is_partial_render = cast("bool", getattr(request, "is_partial_render", False))
+        partial_keys = cast("set[str]", getattr(request, "partial_keys", {}))
         vite_plugin = request.app.plugins.get(VitePlugin)
         template_engine = vite_plugin.template_config.to_engine()
         headers.update(
             {"Vary": "Accept", **get_headers(InertiaHeaderType(enabled=True))},
         )
-        shared_props = get_shared_props(request)
+        shared_props = get_shared_props(request, partial_data=partial_keys if is_partial_render else None)
+        shared_props["content"] = self.content
         page_props = PageProps[T](
             component=request.inertia.route_component,  # type: ignore[attr-defined] # pyright: ignore[reportUnknownArgumentType,reportUnknownMemberType,reportAttributeAccessIssue]
-            props={"content": self.content, **shared_props},  # pyright: ignore[reportArgumentType]
+            props=shared_props,  # pyright: ignore[reportArgumentType]
             version=template_engine.asset_loader.version_id,
             url=request.url.path,
         )
@@ -337,7 +344,7 @@ class InertiaRedirect(Redirect):
         """Initialize external redirect, Set status code to 409 (required by Inertia),
         and pass redirect url.
         """
-        referer = urlparse(request.headers.get("referer", str(request.base_url)))
+        referer = urlparse(request.headers.get("Referer", str(request.base_url)))
         redirect_to = urlunparse(urlparse(redirect_to)._replace(scheme=referer.scheme))
         super().__init__(
             path=redirect_to,
@@ -358,12 +365,8 @@ class InertiaBack(Redirect):
         """Initialize external redirect, Set status code to 409 (required by Inertia),
         and pass redirect url.
         """
-        referer = request.headers.get("referer", str(request.base_url))
-        inertia_enabled = getattr(request, "inertia_enabled", False) or getattr(request, "is_inertia", False)
-        if inertia_enabled:
-            referer = cast("InertiaRequest[Any, Any, Any]", request).inertia.referer or referer
         super().__init__(
-            path=request.headers.get("referer", str(request.base_url)),
+            path=request.headers.get("Referer", str(request.base_url)),
             status_code=HTTP_307_TEMPORARY_REDIRECT if request.method == "GET" else HTTP_303_SEE_OTHER,
             cookies=request.cookies,
             **kwargs,

{litestar_vite-0.2.8.dist-info → litestar_vite-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: litestar-vite
-Version: 0.2.8
+Version: 0.2.9
 Summary: Vite plugin for Litestar
 Project-URL: Changelog, https://cofin.github.io/litestar-vite/latest/changelog
 Project-URL: Discord, https://discord.gg/X3FJqy8d2j

{litestar_vite-0.2.8.dist-info → litestar_vite-0.2.9.dist-info}/RECORD

@@ -10,11 +10,11 @@ litestar_vite/template_engine.py,sha256=ffC4KPtUUNkuC0tJ0bD1Bu7c8lE33vKP0US1fWUY
 litestar_vite/inertia/__init__.py,sha256=RLeCWGHsHyBPpUjSXRhYOQlTh9cG33Q115zGH89M6XE,863
 litestar_vite/inertia/_utils.py,sha256=ijO9Lgka7ZPIAHkby9szbTGoSg0nDShC2bqWT9cDxi0,1956
 litestar_vite/inertia/config.py,sha256=ziEO-xQ5SoE7niri39PMXYgieM94Yf0y4BCFyaMs6P0,1125
-litestar_vite/inertia/exception_handler.py,sha256=S-55sCQ6zLSMICo8l9co0RbT5Dr_TzJlRQBg7YpsRXM,4774
+litestar_vite/inertia/exception_handler.py,sha256=qqhVmosjlGE4gs03AYF1kaNtPdhqGzWHYHjrHxfm900,4735
 litestar_vite/inertia/middleware.py,sha256=NEDcAoT7GMWA9hEGvANZ3MG5_p3MmZX57RF95T71les,1716
 litestar_vite/inertia/plugin.py,sha256=ebAG9XnDBahttuc7WIUgBd3o_Ys8MdPS273LPNs5H8A,2344
-litestar_vite/inertia/request.py,sha256=hk8m1pmDiMbWhVurRDHfDPD24nMHp56JzUKV6SBDeqA,3665
-litestar_vite/inertia/response.py,sha256=D3JqbO1Fa1g7G3FXqbd304cED3zQTTO5tsvO_UN-Ndk,15942
+litestar_vite/inertia/request.py,sha256=5IWZLkQxIOuvYA7RRW5uQZD9skjqr4lEbM4V7HCzCnY,4034
+litestar_vite/inertia/response.py,sha256=Au7jQnqjY3wJAHNmiwteCZxKTKGCpnkjMTSzBVRwkXs,15927
 litestar_vite/inertia/routes.py,sha256=QksJm2RUfL-WbuhOieYnPXXWO5GYnPtmsYEm6Ef8Yeo,1782
 litestar_vite/inertia/types.py,sha256=tLp0pm1N__hcWC875khf6wH1nuFlKS9-VjDqgsRkXnw,702
 litestar_vite/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,7 +24,7 @@ litestar_vite/templates/package.json.j2,sha256=0JWgdTuaSZ25EmCltF_zbqDdpxfvCLeYu
 litestar_vite/templates/styles.css.j2,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 litestar_vite/templates/tsconfig.json.j2,sha256=q1REIuVyXUHCy4Zi2kgTkmrhdT98vyY89k-WTrImOj8,843
 litestar_vite/templates/vite.config.ts.j2,sha256=FZ4OJaB8Kjby_nlx4_LCP8eCe1LRi8kW2GspCiVMfDY,1115
-litestar_vite-0.2.8.dist-info/METADATA,sha256=1X6A9SBiXIjw-P7AwYNqvp_4RH7p3bIoIfisX5juRHE,6180
-litestar_vite-0.2.8.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-litestar_vite-0.2.8.dist-info/licenses/LICENSE,sha256=HeTiEfEgvroUXZe_xAmYHxtTBgw--mbXyZLsWDYabHc,1069
-litestar_vite-0.2.8.dist-info/RECORD,,
+litestar_vite-0.2.9.dist-info/METADATA,sha256=Z32djOlmFMe6ZeWKPrAv0VjDeuklsutMmDEcBYoO3bU,6180
+litestar_vite-0.2.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+litestar_vite-0.2.9.dist-info/licenses/LICENSE,sha256=HeTiEfEgvroUXZe_xAmYHxtTBgw--mbXyZLsWDYabHc,1069
+litestar_vite-0.2.9.dist-info/RECORD,,