From 4e21167a6ba935b0b8c6e2094d1d8baf0ab499e7 Mon Sep 17 00:00:00 2001 From: gazorby Date: Sat, 3 Dec 2022 15:58:46 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20style(starlite):=20cosmetic=20fi?= =?UTF-8?q?xes?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sentry_sdk/integrations/starlite.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/sentry_sdk/integrations/starlite.py b/sentry_sdk/integrations/starlite.py index f6e87e4f4f..8b36e01878 100644 --- a/sentry_sdk/integrations/starlite.py +++ b/sentry_sdk/integrations/starlite.py @@ -1,4 +1,4 @@ -from typing import TYPE_CHECKING, Any, Awaitable, Dict, List, Optional, Union +from typing import TYPE_CHECKING, Any, Awaitable, Dict, Optional, Union from pydantic import BaseModel @@ -93,13 +93,11 @@ def injection_wrapper(self, *args, **kwargs): def patch_middlewares() -> None: old__resolve_middleware_stack = BaseRouteHandler.resolve_middleware - def resolve_middleware_wrapper(self: Any) -> List["Middleware"]: - return list( - map( - lambda middleware: enable_span_for_middleware(middleware), - old__resolve_middleware_stack(self), - ) - ) + def resolve_middleware_wrapper(self: Any) -> list["Middleware"]: + return [ + enable_span_for_middleware(middleware) + for middleware in old__resolve_middleware_stack(self) + ] BaseRouteHandler.resolve_middleware = resolve_middleware_wrapper @@ -143,7 +141,7 @@ async def _sentry_receive( new_receive = _sentry_receive if not receive_patched else receive # Creating spans for the "send" callback - async def _sentry_send(message: "Message") -> Awaitable[None]: + async def _sentry_send(message: "Message") -> None: hub = Hub.current with hub.start_span( op=OP.MIDDLEWARE_STARLITE_SEND, @@ -220,7 +218,7 @@ def event_processor(event: "Event", _: Dict[str, Any]) -> "Event": sentry_scope._name = StarliteIntegration.identifier sentry_scope.add_event_processor(event_processor) - await old_handle(self, scope, receive, send) + return await old_handle(self, scope, receive, send) HTTPRoute.handle = handle_wrapper