anydi 0.62.0__tar.gz → 0.64.0__tar.gz

{anydi-0.62.0 → anydi-0.64.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anydi
-Version: 0.62.0
+Version: 0.64.0
 Summary: Dependency Injection library
 Keywords: dependency injection,dependencies,di,async,asyncio,application
 Author: Anton Ruhlov
@@ -42,6 +42,7 @@ Modern, lightweight Dependency Injection library using type annotations.
 [![CI](https://github.com/antonrh/anydi/actions/workflows/ci.yml/badge.svg)](https://github.com/antonrh/anydi/actions/workflows/ci.yml)
 [![Coverage](https://codecov.io/gh/antonrh/anydi/branch/main/graph/badge.svg)](https://codecov.io/gh/antonrh/anydi)
 [![Documentation](https://readthedocs.org/projects/anydi/badge/?version=latest)](https://anydi.readthedocs.io/en/latest/)
+[![CodSpeed](https://img.shields.io/endpoint?url=https://codspeed.io/badge.json)](https://codspeed.io/antonrh/anydi?utm_source=badge)
 
 </div>
 
@@ -229,7 +230,7 @@ Configure Django (`settings.py`):
 
 ```python
 INSTALLED_APPS = [
-    ...
+    ...,
     "anydi_django",
 ]
 

{anydi-0.62.0 → anydi-0.64.0}/README.md

@@ -7,6 +7,7 @@ Modern, lightweight Dependency Injection library using type annotations.
 [![CI](https://github.com/antonrh/anydi/actions/workflows/ci.yml/badge.svg)](https://github.com/antonrh/anydi/actions/workflows/ci.yml)
 [![Coverage](https://codecov.io/gh/antonrh/anydi/branch/main/graph/badge.svg)](https://codecov.io/gh/antonrh/anydi)
 [![Documentation](https://readthedocs.org/projects/anydi/badge/?version=latest)](https://anydi.readthedocs.io/en/latest/)
+[![CodSpeed](https://img.shields.io/endpoint?url=https://codspeed.io/badge.json)](https://codspeed.io/antonrh/anydi?utm_source=badge)
 
 </div>
 
@@ -194,7 +195,7 @@ Configure Django (`settings.py`):
 
 ```python
 INSTALLED_APPS = [
-    ...
+    ...,
     "anydi_django",
 ]
 

{anydi-0.62.0 → anydi-0.64.0}/anydi/_container.py

@@ -272,6 +272,10 @@ class Container:
         # Register the scope
         self._scopes[scope] = tuple({scope, "singleton"} | set(parents))
 
+    def has_scope(self, scope: str) -> bool:
+        """Check if a scope is registered."""
+        return scope in self._scopes
+
     def get_context_scopes(self, scopes: set[Scope] | None = None) -> list[str]:  # noqa: C901
         """Return scopes that require context management in dependency order."""
         # Build execution order: singleton -> request -> custom (by depth)
@@ -431,6 +435,10 @@ class Container:
         parameters: list[ProviderParameter] = []
         scope_provider: dict[Scope, Provider] = {}
 
+        # Precompute constant checks
+        is_scoped = scope not in ("singleton", "transient")
+        has_defaults = defaults is not None
+
         for parameter in signature.parameters.values():
             if parameter.annotation is inspect.Parameter.empty:
                 raise TypeError(
@@ -450,10 +458,32 @@ class Container:
             )
             has_default = default is not NOT_SET
 
+            # Check if provider exists before attempting to register (for scoped only)
+            was_auto_registered = (
+                is_scoped and parameter.annotation not in self._providers
+            )
+
             try:
                 sub_provider = self._get_or_register_provider(parameter.annotation)
             except LookupError as exc:
-                if parameter.name in defaults if defaults else False or has_default:
+                if (has_defaults and parameter.name in defaults) or has_default:
+                    continue
+                # For request/custom scopes, allow unregistered dependencies
+                # They might be provided via context.set()
+                if is_scoped:
+                    # Add to unresolved list to provide better error messages
+                    # and prevent infinite recursion
+                    self._resolver.add_unresolved(parameter.annotation)
+                    parameters.append(
+                        ProviderParameter(
+                            name=parameter.name,
+                            annotation=parameter.annotation,
+                            default=default,
+                            has_default=has_default,
+                            provider=None,  # Will check context at runtime
+                            shared_scope=True,  # Same scope, check context
+                        )
+                    )
                     continue
                 unresolved_parameter = parameter
                 unresolved_exc = exc
@@ -463,6 +493,10 @@ class Container:
             if sub_provider.scope not in scope_provider:
                 scope_provider[sub_provider.scope] = sub_provider
 
+            # If provider was auto-registered and has same scope, mark as unresolved
+            if was_auto_registered and sub_provider.scope == scope:
+                self._resolver.add_unresolved(parameter.annotation)
+
             parameters.append(
                 ProviderParameter(
                     name=parameter.name,

{anydi-0.62.0 → anydi-0.64.0}/anydi/_resolver.py

@@ -228,11 +228,15 @@ class Resolver:
                 )
                 create_lines.append(f"        arg_{idx} = defaults['{name}']")
                 create_lines.append("    else:")
-                create_lines.append("        cached = NOT_SET_")
-                create_lines.append("        if context is not None:")
-                create_lines.append(
-                    f"            cached = context.get(_param_annotations[{idx}])"
-                )
+                # Direct dict access for shared scope params (avoids method call)
+                if param_shared_scopes[idx]:
+                    create_lines.append(
+                        f"        cached = (context._instances.get("
+                        f"_param_annotations[{idx}], NOT_SET_) "
+                        f"if context is not None else NOT_SET_)"
+                    )
+                else:
+                    create_lines.append("        cached = NOT_SET_")
                 create_lines.append("        if cached is NOT_SET_:")
                 create_lines.append(
                     f"            if _param_annotations[{idx}] in "

{anydi-0.62.0 → anydi-0.64.0}/anydi/ext/fastapi.py

@@ -8,8 +8,8 @@ from typing import Annotated, Any, cast
 
 from fastapi import Depends, FastAPI, params
 from fastapi.dependencies.models import Dependant
-from fastapi.routing import APIRoute
-from starlette.requests import Request
+from fastapi.routing import APIRoute, APIWebSocketRoute
+from starlette.requests import HTTPConnection
 
 from anydi import Container, Inject
 from anydi._marker import Marker, extend_marker
@@ -19,9 +19,8 @@ from .starlette.middleware import RequestScopedMiddleware
 __all__ = ["install", "get_container", "Inject", "RequestScopedMiddleware"]
 
 
-def get_container(request: Request) -> Container:
-    """Get the AnyDI container from a FastAPI request."""
-    return cast(Container, request.app.state.container)
+def get_container(connection: HTTPConnection) -> Container:
+    return cast(Container, connection.app.state.container)
 
 
 class FastAPIMarker(params.Depends, Marker):
@@ -39,7 +38,9 @@ class FastAPIMarker(params.Depends, Marker):
         return await container.aresolve(self.interface)
 
 
-# Configure Inject() and Provide[T] to use FastAPI-specific marker
+# Configure Inject() and Provide[T] to use FastAPI-specific marker at import time
+# This is also called in install() to ensure it's set correctly even if other
+# extensions have overwritten it
 extend_marker(FastAPIMarker)
 
 
@@ -51,7 +52,9 @@ def _iter_dependencies(dependant: Dependant) -> Iterator[Dependant]:
 
 
 def _validate_route_dependencies(
-    route: APIRoute, container: Container, patched: set[tuple[Any, ...]]
+    route: APIRoute | APIWebSocketRoute,
+    container: Container,
+    patched: set[tuple[Any, ...]],
 ) -> None:
     for dependant in _iter_dependencies(route.dependant):
         if dependant.cache_key in patched:
@@ -71,8 +74,14 @@ def _validate_route_dependencies(
 def install(app: FastAPI, container: Container) -> None:
     """Install AnyDI into a FastAPI application."""
     app.state.container = container  # noqa
+
+    # Register websocket scope with request as parent if not already registered
+    if not container.has_scope("websocket"):
+        container.register_scope("websocket", parents=["request"])
+
+    # Validate routes (both HTTP and WebSocket)
     patched: set[tuple[Any, ...]] = set()
     for route in app.routes:
-        if not isinstance(route, APIRoute):
+        if not isinstance(route, APIRoute | APIWebSocketRoute):
             continue
         _validate_route_dependencies(route, container, patched)

anydi-0.64.0/anydi/ext/starlette/middleware.py

@@ -0,0 +1,49 @@
+"""Starlette middleware for AnyDI scoped contexts."""
+
+from __future__ import annotations
+
+from contextlib import AsyncExitStack
+
+from starlette.requests import Request
+from starlette.types import ASGIApp, Receive, Scope, Send
+from starlette.websockets import WebSocket
+
+from anydi import Container
+
+
+class RequestScopedMiddleware:
+    """ASGI middleware for managing request-scoped AnyDI context."""
+
+    def __init__(self, app: ASGIApp, container: Container) -> None:
+        self.app = app
+        self.container = container
+
+    async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+        # Only handle HTTP requests or WebSocket connections
+        if scope["type"] not in {"http", "websocket"}:
+            await self.app(scope, receive, send)
+            return
+
+        async with AsyncExitStack() as stack:
+            # Create request context first (parent scope)
+            request_context = await stack.enter_async_context(
+                self.container.arequest_context()
+            )
+
+            # For WebSocket connections, create websocket context (child scope)
+            websocket_context = None
+            if scope["type"] == "websocket" and self.container.has_scope("websocket"):
+                websocket_context = await stack.enter_async_context(
+                    self.container.ascoped_context("websocket")
+                )
+
+            if scope["type"] == "http":
+                request = Request(scope, receive=receive, send=send)
+                request_context.set(Request, request)
+            else:
+                websocket = WebSocket(scope, receive=receive, send=send)
+                request_context.set(WebSocket, websocket)
+                if websocket_context is not None:
+                    websocket_context.set(WebSocket, websocket)
+
+            await self.app(scope, receive, send)

{anydi-0.62.0 → anydi-0.64.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "anydi"
-version = "0.62.0"
+version = "0.64.0"
 description = "Dependency Injection library"
 authors = [{ name = "Anton Ruhlov", email = "antonruhlov@gmail.com" }]
 requires-python = ">=3.10.0, <3.15"
@@ -55,6 +55,7 @@ dev = [
     "pytest>=8.4.0,<9",
     "pytest-cov>=7.0.0",
     "pytest-mock>=3.14.1",
+    "pytest-codspeed>=4.2.0",
     "starlette>=0.37.2",
     "fastapi>=0.100.0",
     "httpx>=0.26.0",

anydi-0.62.0/anydi/ext/starlette/middleware.py

@@ -1,23 +0,0 @@
-"""Starlette RequestScopedMiddleware."""
-
-from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
-from starlette.requests import Request
-from starlette.responses import Response
-from starlette.types import ASGIApp
-
-from anydi import Container
-
-
-class RequestScopedMiddleware(BaseHTTPMiddleware):
-    """Starlette middleware for managing request-scoped AnyDI context."""
-
-    def __init__(self, app: ASGIApp, container: Container) -> None:
-        super().__init__(app)
-        self.container = container
-
-    async def dispatch(
-        self, request: Request, call_next: RequestResponseEndpoint
-    ) -> Response:
-        async with self.container.arequest_context() as context:
-            context.set(Request, request)
-            return await call_next(request)