azure-storage-blob 12.26.0__py3-none-any.whl → 12.27.0__py3-none-any.whl

azure/storage/blob/__init__.py

@@ -125,8 +125,8 @@ def upload_blob_to_url(
     :return: Blob-updated property dict (Etag and last modified)
     :rtype: dict(str, Any)
     """
-    with BlobClient.from_blob_url(blob_url, credential=credential) as client:
-        return cast(BlobClient, client).upload_blob(data=data, blob_type=BlobType.BLOCKBLOB, **kwargs)
+    with BlobClient.from_blob_url(blob_url, credential=credential) as client:  # pylint: disable=not-context-manager
+        return client.upload_blob(data=data, blob_type=BlobType.BLOCKBLOB, **kwargs)
 
 
 def _download_to_stream(client: BlobClient, handle: IO[bytes], **kwargs: Any) -> None:
@@ -194,7 +194,7 @@ def download_blob_from_url(
     :rtype: None
     """
     overwrite = kwargs.pop('overwrite', False)
-    with BlobClient.from_blob_url(blob_url, credential=credential) as client:
+    with BlobClient.from_blob_url(blob_url, credential=credential) as client: # pylint: disable=not-context-manager
         if hasattr(output, 'write'):
             _download_to_stream(client, cast(IO[bytes], output), **kwargs)
         else:

azure/storage/blob/_blob_client.py

@@ -190,6 +190,22 @@ class BlobClient(StorageAccountHostsMixin, StorageEncryptionMixin):  # pylint: d
         self._client._config.version = get_api_version(kwargs)  # type: ignore [assignment]
         self._configure_encryption(kwargs)
 
+    def __enter__(self) -> Self:
+        self._client.__enter__()
+        return self
+
+    def __exit__(self, *args) -> None:
+        self._client.__exit__(*args)
+
+    def close(self) -> None:
+        """This method is to close the sockets opened by the client.
+        It need not be used when using with a context manager.
+
+        :return: None
+        :rtype: None
+        """
+        self._client.close()
+
     def _format_url(self, hostname: str) -> str:
         return _format_url(
             container_name=self.container_name,