DEV-19025: codestyle

huntflow · Feb 8, 2025 · f35d67d · f35d67d
1 parent 66f10bc
commit f35d67d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)