Fix/file stream azure blob (#6196)

This commit is contained in:
Jyong 2024-07-11 17:01:03 +08:00 committed by GitHub
parent 63e34e5227
commit 678ad6b7eb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -38,9 +38,8 @@ class AzureStorage(BaseStorage):
def generate(filename: str = filename) -> Generator: def generate(filename: str = filename) -> Generator:
blob = client.get_blob_client(container=self.bucket_name, blob=filename) blob = client.get_blob_client(container=self.bucket_name, blob=filename)
blob_data = blob.download_blob() blob_data = blob.download_blob()
for chunk in blob_data.chunks(): yield from blob_data.chunks()
yield from chunk return generate(filename)
return generate()
def download(self, filename, target_filepath): def download(self, filename, target_filepath):
client = self._sync_client() client = self._sync_client()