Skip to content

Commit

Permalink
🎨 style(starlite): cosmetic fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
gazorby committed Jan 8, 2023
1 parent 50da815 commit 4e21167
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions 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

Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -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,
Expand Down Expand Up @@ -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

Expand Down

0 comments on commit 4e21167

Please sign in to comment.