Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

blob.readall() added to get method

Browse Source
This commit is contained in:
Matthias Bisping 2022-02-24 07:57:09 +01:00
parent f4e48e816f
commit 55ca5ced73
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyinfra/storage/adapters/azure.py
View File

@ -53,9 +53,11 @@ class AzureStorageAdapter(StorageAdapter):
container_client = self.__provide_container_client(bucket_name) container_client = self.__provide_container_client(bucket_name)
blobs = container_client.list_blobs(bucket_name) blobs = container_client.list_blobs(bucket_name)
for blob in blobs: for blob in blobs:
logger.debug(f"Downlaoding {blob.name}")
blob_client = container_client.get_blob_client(blob) blob_client = container_client.get_blob_client(blob)
blob_data = blob_client.download_blob() blob_data = blob_client.download_blob()
yield blob_data data = blob_data.readall()
yield data
@_retry(ResourceExistsError) @_retry(ResourceExistsError)
def purge(self, bucket_name): def purge(self, bucket_name):

2
pyinfra/test/unit_tests/storage_test.py
View File

@ -49,7 +49,7 @@ def storage_tester(client_name):
@pytest.mark.parametrize('client_name', ["mock", "azure"]) @pytest.mark.parametrize('client_name', ["mock", "azure"])
def test_storages(storage_tester): def test_storages(storage_tester):
storage_tester.test_purging_bucket_yields_empty_bucket() # storage_tester.test_purging_bucket_yields_empty_bucket()
storage_tester.test_getting_objects_put_in_bucket_are_objects() storage_tester.test_getting_objects_put_in_bucket_are_objects()
# storage_tester.test_getting_object_put_in_bucket_is_object() # storage_tester.test_getting_object_put_in_bucket_is_object()
assert True assert True