port-ocean 0.25.3__py3-none-any.whl → 0.25.5__py3-none-any.whl

port_ocean/helpers/retry.py

@@ -4,9 +4,10 @@ import time
 from datetime import datetime
 from functools import partial
 from http import HTTPStatus
-from typing import Any, Callable, Coroutine, Iterable, Mapping, Union
+from typing import Any, Callable, Coroutine, Iterable, Mapping, Union, cast
 import httpx
 from dateutil.parser import isoparse
+import logging
 
 _ON_RETRY_CALLBACK: Callable[[httpx.Request], httpx.Request] | None = None
 
@@ -148,6 +149,9 @@ class RetryTransport(httpx.AsyncBaseTransport, httpx.BaseTransport):
                 response = self._retry_operation(request, send_method)
             else:
                 response = transport.handle_request(request)
+
+            self._log_response_size(request, response)
+
             return response
         except Exception as e:
             if not self._is_retryable_method(request) and self._logger is not None:
@@ -172,6 +176,8 @@ class RetryTransport(httpx.AsyncBaseTransport, httpx.BaseTransport):
             else:
                 response = await transport.handle_async_request(request)
 
+            await self._log_response_size_async(request, response)
+
             return response
         except Exception as e:
             # Retyable methods are logged via _log_error
@@ -246,6 +252,67 @@ class RetryTransport(httpx.AsyncBaseTransport, httpx.BaseTransport):
                 f" {type(error).__name__} - {str(error) or 'No error message'}, retrying in {sleep_time} seconds."
             )
 
+    def _should_log_response_size(self, request: httpx.Request) -> bool:
+        return self._logger is not None and not request.url.host.endswith("getport.io")
+
+    def _get_content_length(self, response: httpx.Response) -> int | None:
+        content_length = response.headers.get("Content-Length") or response.headers.get(
+            "content-length"
+        )
+        if content_length:
+            return int(content_length)
+        return None
+
+    async def _log_response_size_async(
+        self, request: httpx.Request, response: httpx.Response
+    ) -> None:
+        """Log the size of the response."""
+        if not self._should_log_response_size(request):
+            return
+
+        # Try to get content length from headers first
+        content_length = self._get_content_length(response)
+        if content_length is not None:
+            size_info = content_length
+        else:
+            # If no Content-Length header, try to get actual content size
+            try:
+                actual_size = len(await response.aread())
+                size_info = actual_size
+            except Exception as e:
+                cast(logging.Logger, self._logger).error(
+                    f"Error getting response size: {e}"
+                )
+                return
+
+        cast(logging.Logger, self._logger).info(
+            f"Response for {request.method} {request.url} - Size: {size_info} bytes"
+        )
+
+    def _log_response_size(
+        self, request: httpx.Request, response: httpx.Response
+    ) -> None:
+        if not self._should_log_response_size(request):
+            return
+
+        content_length = self._get_content_length(response)
+        if content_length is not None:
+            size_info = content_length
+        else:
+            # If no Content-Length header, try to get actual content size
+            try:
+                actual_size = len(response.read())
+                size_info = actual_size
+            except Exception as e:
+                cast(logging.Logger, self._logger).error(
+                    f"Error getting response size: {e}"
+                )
+                return
+
+        cast(logging.Logger, self._logger).info(
+            f"Response for {request.method} {request.url} - Size: {size_info} bytes"
+        )
+
     async def _should_retry_async(self, response: httpx.Response) -> bool:
         return response.status_code in self._retry_status_codes
 

{port_ocean-0.25.3.dist-info → port_ocean-0.25.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.25.3
+Version: 0.25.5
 Summary: Port Ocean is a CLI tool for managing your Port projects.
 Home-page: https://app.getport.io
 Keywords: ocean,port-ocean,port
@@ -28,6 +28,7 @@ Requires-Dist: confluent-kafka (>=2.10.1,<3.0.0)
 Requires-Dist: cookiecutter (>=2.1.1,<3.0.0) ; extra == "cli"
 Requires-Dist: fastapi (>=0.115.3,<0.116.0)
 Requires-Dist: httpx (>=0.28.1,<0.29.0)
+Requires-Dist: jinja2 (>=3.1.6)
 Requires-Dist: jinja2-time (>=0.2.0,<0.3.0) ; extra == "cli"
 Requires-Dist: jq (>=1.8.0,<2.0.0)
 Requires-Dist: loguru (>=0.7.0,<0.8.0)

{port_ocean-0.25.3.dist-info → port_ocean-0.25.5.dist-info}/RECORD

@@ -142,7 +142,7 @@ port_ocean/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 port_ocean/helpers/async_client.py,sha256=0MkFY46MuEYkju7gJ8HYeUTMBsjwAzESvATryJy2DaM,1698
 port_ocean/helpers/metric/metric.py,sha256=Aacz7bOd8ZCwEPpXAdwLbKRXf28Z4wiViG_GXiV_xWg,14529
 port_ocean/helpers/metric/utils.py,sha256=1lAgrxnZLuR_wUNDyPOPzLrm32b8cDdioob2lvnPQ1A,1619
-port_ocean/helpers/retry.py,sha256=gmS4YxM6N4fboFp7GSgtOzyBJemxs46bnrz4L4rDS6Y,16136
+port_ocean/helpers/retry.py,sha256=VHAp6j9-Vid6aNR5sca3S0aW6b1S2oYw9vT9hi1N22U,18556
 port_ocean/log/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 port_ocean/log/handlers.py,sha256=ncVjgqrZRh6BhyRrA6DQG86Wsbxph1yWYuEC0cWfe-Q,3631
 port_ocean/log/logger_setup.py,sha256=0K3zVG0YYrYOWEV8-rCGks1o-bMRxgHXlqawu9w_tSw,2656
@@ -203,8 +203,8 @@ port_ocean/utils/repeat.py,sha256=U2OeCkHPWXmRTVoPV-VcJRlQhcYqPWI5NfmPlb1JIbc,32
 port_ocean/utils/signal.py,sha256=mMVq-1Ab5YpNiqN4PkiyTGlV_G0wkUDMMjTZp5z3pb0,1514
 port_ocean/utils/time.py,sha256=pufAOH5ZQI7gXvOvJoQXZXZJV-Dqktoj9Qp9eiRwmJ4,1939
 port_ocean/version.py,sha256=UsuJdvdQlazzKGD3Hd5-U7N69STh8Dq9ggJzQFnu9fU,177
-port_ocean-0.25.3.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-port_ocean-0.25.3.dist-info/METADATA,sha256=MAUx2nbIcxCElrak8jRvin6MeNPvB40mh2bNqU3xEQM,6855
-port_ocean-0.25.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-port_ocean-0.25.3.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
-port_ocean-0.25.3.dist-info/RECORD,,
+port_ocean-0.25.5.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+port_ocean-0.25.5.dist-info/METADATA,sha256=z6eqWLmVOCtooAj5NDNhgK76ZNsAtBXWufhYAz2ee80,6887
+port_ocean-0.25.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+port_ocean-0.25.5.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
+port_ocean-0.25.5.dist-info/RECORD,,