tracktolib 0.66.3__py3-none-any.whl → 0.67.0__py3-none-any.whl

tracktolib/s3/niquests.py

@@ -272,9 +272,12 @@ class S3Session:
         key: str,
         on_chunk: Callable[[bytes], None] | None = None,
         chunk_size: int = 1024 * 1024,
+        on_start: OnDownloadStartFn | None = None,
     ) -> AsyncIterator[bytes]:
         """Download a file from S3 with streaming support."""
-        async for chunk in s3_download_file(self._s3, self.http_client, bucket, key, chunk_size=chunk_size):
+        async for chunk in s3_download_file(
+            self._s3, self.http_client, bucket, key, chunk_size=chunk_size, on_start=on_start
+        ):
             if on_chunk:
                 on_chunk(chunk)
             yield chunk
@@ -433,10 +436,10 @@ async def s3_put_object(
 
     See: https://docs.aws.amazon.com/AmazonS3/latest/API/API_PutObject.html
     """
+
     obj_params: S3ObjectParams = kwargs
     presigned_params = build_s3_presigned_params(bucket, key, obj_params)
     headers = build_s3_headers(obj_params)
-
     url = s3.generate_presigned_url(
         ClientMethod="put_object",
         Params=presigned_params,
@@ -478,6 +481,9 @@ async def s3_get_object(
     return resp.content
 
 
+type OnDownloadStartFn = Callable[[niquests.AsyncResponse], None]
+
+
 async def s3_download_file(
     s3: botocore.client.BaseClient,
     client: niquests.AsyncSession,
@@ -485,6 +491,7 @@ async def s3_download_file(
     key: str,
     *,
     chunk_size: int = 1024 * 1024,
+    on_start: OnDownloadStartFn | None = None,
 ) -> AsyncIterator[bytes]:
     """Download an object from S3 with streaming support."""
     url = s3.generate_presigned_url(
@@ -493,6 +500,8 @@ async def s3_download_file(
     )
     resp = await client.get(url, stream=True)
     resp.raise_for_status()
+    if on_start:
+        on_start(resp)
     async for chunk in await resp.iter_content(chunk_size):
         yield chunk
 

{tracktolib-0.66.3.dist-info → tracktolib-0.67.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracktolib
-Version: 0.66.3
+Version: 0.67.0
 Summary: Utility library for python
 Keywords: utility
 Author-email: julien.brayere@tracktor.fr

{tracktolib-0.66.3.dist-info → tracktolib-0.67.0.dist-info}/RECORD

@@ -12,10 +12,10 @@ tracktolib/pg_sync.py,sha256=PDTN37kU0BxkSZetwSAtqcW2aA8Nn4gUI2mC54gSJhg,6750
 tracktolib/pg_utils.py,sha256=ArYNdf9qsdYdzGEWmev8tZpyx8_1jaGGdkfYkauM7UM,2582
 tracktolib/s3/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tracktolib/s3/minio.py,sha256=wMEjkSes9Fp39fD17IctALpD6zB2xwDRQEmO7Vzan3g,1387
-tracktolib/s3/niquests.py,sha256=HTKEDyD9W30Ru7jeNSl9gOHgpa3UHkEdQD6cBVp_l0Q,23206
+tracktolib/s3/niquests.py,sha256=9j3RxM3EfIYV1wEH0OpvT_uhJ68sXN4PwxlDAH3eBEE,23453
 tracktolib/s3/s3.py,sha256=Vi3Q6DLBm44gz6fXx6uzdbGEtJly6KzdgLYHJwU6r-U,4922
 tracktolib/tests.py,sha256=gKE--epQjgMZGXc5ydbl4zjOdmwztJS42UMV0p4hXEA,399
 tracktolib/utils.py,sha256=FP87gbL27zHXaI9My2VZYEG5ZJ7eL6SiljW5MyRutOY,6553
-tracktolib-0.66.3.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
-tracktolib-0.66.3.dist-info/METADATA,sha256=4ELI18eTlJ46kdn2LLwFXwXYB-d8ZvEr1qm02lY_x78,4045
-tracktolib-0.66.3.dist-info/RECORD,,
+tracktolib-0.67.0.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
+tracktolib-0.67.0.dist-info/METADATA,sha256=yJodZh9-6H2mIOQcQWguUE9zsbDqOPKLyTh3A_tHUAo,4045
+tracktolib-0.67.0.dist-info/RECORD,,