Quellcode durchsuchen

fix azure stream download (#6063)

tags/0.6.13
Jyong vor 1 Jahr
Ursprung
Commit
cbbe28f40d
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
1 geänderte Dateien mit 3 neuen und 5 gelöschten Zeilen
  1. 3
    5
      api/extensions/storage/azure_storage.py

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

from collections.abc import Generator from collections.abc import Generator
from contextlib import closing
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone


from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas from azure.storage.blob import AccountSasPermissions, BlobServiceClient, ResourceTypes, generate_account_sas


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)
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() return generate()


def download(self, filename, target_filepath): def download(self, filename, target_filepath):

Laden…
Abbrechen
Speichern