From f35d67daf27f8fd13567112b344d1a2a18a95df1 Mon Sep 17 00:00:00 2001 From: Aleksey Reznikov Date: Sat, 8 Feb 2025 12:49:48 +0300 Subject: [PATCH] DEV-19025: codestyle --- src/huntflow_base_metrics/web_frameworks/litestar.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/huntflow_base_metrics/web_frameworks/litestar.py b/src/huntflow_base_metrics/web_frameworks/litestar.py index af66836..6755b33 100644 --- a/src/huntflow_base_metrics/web_frameworks/litestar.py +++ b/src/huntflow_base_metrics/web_frameworks/litestar.py @@ -59,14 +59,14 @@ async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None: self.count_request_exceptions(ctx, exception_type) @staticmethod - def _get_send_wrapper(send: Send, span: RequestContext) -> Callable: + def _get_send_wrapper(send: Send, ctx: RequestContext) -> Callable: @wraps(send) async def wrapped_send(message: Message) -> None: if message["type"] == "http.response.start": - span.status_code = message["status"] + ctx.status_code = message["status"] if message["type"] == "http.response.body": - span.end_time = time.perf_counter() + ctx.end_time = time.perf_counter() await send(message)