fix types for extras

Author: lesnik512

lite_bootstrap/bootstrappers/fastapi_bootstrapper.py

@@ -110,7 +110,7 @@ def bootstrap(self) -> None:
         )
 
 
-class FastAPIBootstrapper(BaseBootstrapper[fastapi.FastAPI]):
+class FastAPIBootstrapper(BaseBootstrapper["fastapi.FastAPI"]):
     __slots__ = "bootstrap_config", "instruments"
 
     instruments_types: typing.ClassVar = [

lite_bootstrap/bootstrappers/faststream_bootstrapper.py

@@ -1,4 +1,3 @@
-from __future__ import annotations
 import dataclasses
 import json
 import typing
@@ -53,8 +52,8 @@ def __call__(self, msg: typing.Any | None) -> faststream.BaseMiddleware: ...  #
 
 @dataclasses.dataclass(kw_only=True, slots=True, frozen=True)
 class FastStreamConfig(HealthChecksConfig, LoggingConfig, OpentelemetryConfig, PrometheusConfig, SentryConfig):
-    application: AsgiFastStream = dataclasses.field(default_factory=AsgiFastStream)
-    broker: BrokerUsecase[typing.Any, typing.Any] | None = None
+    application: "AsgiFastStream" = dataclasses.field(default_factory=AsgiFastStream)
+    broker: typing.Optional["BrokerUsecase[typing.Any, typing.Any]"] = None
     opentelemetry_middleware_cls: type[FastStreamTelemetryMiddlewareProtocol] | None = None
     prometheus_middleware_cls: type[FastStreamPrometheusMiddlewareProtocol] | None = None
 
@@ -65,7 +64,7 @@ class FastStreamHealthChecksInstrument(HealthChecksInstrument):
 
     def bootstrap(self) -> None:
         @handle_get
-        async def check_health(_: object) -> AsgiResponse:
+        async def check_health(_: object) -> "AsgiResponse":
             return (
                 AsgiResponse(
                     json.dumps(self.render_health_check_data()).encode(), 200, headers={"content-type": "text/plain"}
@@ -137,7 +136,7 @@ def bootstrap(self) -> None:
             )
 
 
-class FastStreamBootstrapper(BaseBootstrapper[AsgiFastStream]):
+class FastStreamBootstrapper(BaseBootstrapper["AsgiFastStream"]):
     __slots__ = "bootstrap_config", "instruments"
 
     instruments_types: typing.ClassVar = [
@@ -158,5 +157,5 @@ def __init__(self, bootstrap_config: FastStreamConfig) -> None:
         if self.bootstrap_config.broker:
             self.bootstrap_config.application.broker = self.bootstrap_config.broker
 
-    def _prepare_application(self) -> AsgiFastStream:
+    def _prepare_application(self) -> "AsgiFastStream":
         return self.bootstrap_config.application

lite_bootstrap/bootstrappers/litestar_bootstrapper.py

@@ -112,7 +112,7 @@ class LitestarPrometheusController(PrometheusController):
         self.bootstrap_config.application_config.middleware.append(litestar_prometheus_config.middleware)
 
 
-class LitestarBootstrapper(BaseBootstrapper[litestar.Litestar]):
+class LitestarBootstrapper(BaseBootstrapper["litestar.Litestar"]):
     __slots__ = "bootstrap_config", "instruments"
 
     instruments_types: typing.ClassVar = [
@@ -131,5 +131,5 @@ def is_ready(self) -> bool:
     def __init__(self, bootstrap_config: LitestarConfig) -> None:
         super().__init__(bootstrap_config)
 
-    def _prepare_application(self) -> litestar.Litestar:
+    def _prepare_application(self) -> "litestar.Litestar":
         return litestar.Litestar.from_config(self.bootstrap_config.application_config)

pyproject.toml

@@ -132,7 +132,7 @@ python_version = "3.10"
 strict = true
 
 [tool.ruff]
-fix = true
+fix = false
 unsafe-fixes = true
 line-length = 120
 target-version = "py310"