pybiolib 1.2.257__tar.gz → 1.2.261__tar.gz

{pybiolib-1.2.257 → pybiolib-1.2.261}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybiolib
-Version: 1.2.257
+Version: 1.2.261
 Summary: BioLib Python Client
 Home-page: https://github.com/biolib
 License: MIT

{pybiolib-1.2.257 → pybiolib-1.2.261}/biolib/_internal/http_client.py

@@ -79,6 +79,7 @@ class HttpClient:
         timeout_in_seconds: Optional[int] = None,
         response_path: Optional[str] = None,
         retry_on_http_500: Optional[bool] = False,
+        max_content_length_in_bytes: Optional[int] = None,
     ) -> HttpResponse:
         if not HttpClient.ssl_context:
             HttpClient.ssl_context = _create_ssl_context()
@@ -107,6 +108,14 @@ class HttpClient:
                     context=HttpClient.ssl_context,
                     timeout=timeout_in_seconds,
                 ) as response:
+                    if max_content_length_in_bytes:
+                        content_length = response.getheader('Content-Length')
+                        if not content_length:
+                            raise ValueError('No Content-Length in response headers')
+
+                        if int(content_length) > max_content_length_in_bytes:
+                            raise ValueError(f'Content-Length exceeds {max_content_length_in_bytes} bytes')
+
                     return HttpResponse(response, response_path)
 
             except urllib.error.HTTPError as error:

{pybiolib-1.2.257 → pybiolib-1.2.261}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pybiolib"
-version = "1.2.257"
+version = "1.2.261"
 description = "BioLib Python Client"
 readme = "PYPI_README.md"
 license = "MIT"