fix duration

Author: r-m-n

.pre-commit-config.yaml

@@ -16,3 +16,8 @@ repos:
       args: [ --fix ]
     # Run the formatter.
     - id: ruff-format
+
+- repo: https://github.com/pycqa/isort
+  rev: 5.13.2
+  hooks:
+    - id: isort

httpx_logging_client/__init__.py

@@ -1,4 +1,6 @@
-from .client import LoggingClient, AsyncLoggingClient
-
+from .client import (
+    AsyncLoggingClient,
+    LoggingClient,
+)
 
 __all__ = ("AsyncLoggingClient", "LoggingClient")

httpx_logging_client/client.py

@@ -1,13 +1,24 @@
 import typing
+
 import httpx
-from httpx import ASGITransport, WSGITransport
+from httpx import (
+    ASGITransport,
+    WSGITransport,
+)
+from httpx._config import (
+    DEFAULT_LIMITS,
+    Limits,
+    Proxy,
+)
+from httpx._types import (
+    CertTypes,
+    VerifyTypes,
+)
+
 from httpx_logging_client.transport import (
     AsyncHTTPLoggingTransport,
     HTTPLoggingTransport,
 )
-from httpx._types import CertTypes
-from httpx._types import VerifyTypes
-from httpx._config import Limits, Proxy, DEFAULT_LIMITS
 
 
 class LoggingClient(httpx.Client):

httpx_logging_client/transport.py

@@ -2,10 +2,11 @@
 
 import httpx
 import structlog
-from httpx import Request
+from httpx import (
+    Request,
+    Response,
+)
 from structlog.types import FilteringBoundLogger
-from httpx import Response
-
 
 logger = structlog.get_logger()
 
@@ -38,12 +39,13 @@ def handle_request(self, request: Request) -> Response:
         start_time = time.perf_counter()
         try:
             response = super().handle_request(request)
+            end_time = time.perf_counter()
             log = log_bind_response(log, response)
             response.read()
             log.info(
                 "httpx request",
                 response_body=response.content.decode(),
-                duration=time.perf_counter() - start_time,
+                duration=end_time - start_time,
             )
             return response
         except httpx.HTTPError as exc:
@@ -61,12 +63,13 @@ async def handle_async_request(
         start_time = time.perf_counter()
         try:
             response = await super().handle_async_request(request)
+            end_time = time.perf_counter()
             log = log_bind_response(log, response)
             await response.aread()
             log.info(
                 "httpx request",
                 response_body=response.content.decode(),
-                duration=time.perf_counter() - start_time,
+                duration=end_time - start_time,
             )
             return response
         except httpx.HTTPError as exc:

poetry.lock

@@ -13,7 +13,13 @@ structlog = "^24.2.0"
 [tool.poetry.group.dev.dependencies]
 mypy = "^1.10.1"
 ruff = "^0.5.0"
+isort = "^5.13.2"
 
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.isort]
+multi_line_output = 3
+include_trailing_comma = true
+force_grid_wrap = 2