Browse Source

fix azure stream download (#6063)

Jyong 9 tháng trước cách đây
mục cha
commit
cbbe28f40d
1 tập tin đã thay đổi với 3 bổ sung5 xóa
  1. 3 5
      api/extensions/storage/azure_storage.py

+ 3 - 5
api/extensions/storage/azure_storage.py

@@ -1,5 +1,4 @@
 from collections.abc import Generator
-from contextlib import closing
 from datetime import datetime, timedelta, timezone
 
 from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas
@@ -38,10 +37,9 @@ class AzureStorage(BaseStorage):
 
         def generate(filename: str = filename) -> Generator:
             blob = client.get_blob_client(container=self.bucket_name, blob=filename)
-            with closing(blob.download_blob()) as blob_stream:
-                while chunk := blob_stream.readall():
-                    yield chunk
-
+            blob_data = blob.download_blob()
+            for chunk in blob_data.chunks():
+                yield from chunk
         return generate()
 
     def download(self, filename, target_filepath):