o xi(@spdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z d d lmZGd d d eZd S)z)Implementation of AzureEnvironment class.)HttpResponseError)DefaultAzureCredential) BlobClientBlobServiceClient)Self) LaunchError)AZURE_BLOB_REGEX)AbstractEnvironmentc@seZdZdZ  dddZeddededefd d Z ede fd d Z d e de ddfddZ d e de ddfddZde ddfddZdddZede dee e e ffddZdS)AzureEnvironmentz;AzureEnvironment is a helper for accessing Azure resources.returnNcCsdS)zInitialize an AzureEnvironment.Nselfrrb/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/environment/azure_environment.py__init__szAzureEnvironment.__init__TconfigverifycCs|S)z.Create an AzureEnvironment from a config dict.r)clsrrrrr from_configszAzureEnvironment.from_configc Cs2ztWSty}ztd||d}~ww)zGet Azure credentials.z`Could not get Azure credentials. Please make sure you have configured your Azure CLI correctly. N)r Exceptionr)rerrrget_credentialssz AzureEnvironment.get_credentialssource destinationc s||\}}}d|d|}|}z-td|d|||d}t|d} |j| ddWd Wd S1s;wYWd StyY} z t|d | j| d } ~ wtys} zt|d | j j d | | d } ~ ww) apUpload a file to Azure blob storage. Arguments: source (str): The path to the file to upload. destination (str): The destination path in Azure blob storage. Ex: https://.blob.core.windows.net// Raise: LaunchError: If the file could not be uploaded. zCould not upload file z to Azure blob https://.blob.core.windows.net credentialrbT) overwriteNz: ) parse_urirropen upload_blobrrmessager __class____name__) rrrstorage_accountstorage_containerpath _err_prefixcredsclientfrrrr upload_file%s*   &zAzureEnvironment.upload_filecst)z)Upload a directory to Azure blob storage.)NotImplementedError)rrrrrr upload_dir@szAzureEnvironment.upload_diruric sp|}||\}}}ztd|d|d}||WdSty7}z td|d|d|d}~ww)zsVerify that the given blob storage prefix exists. Args: uri (str): The URI to verify. rrrzCould not verify storage URI z in container .N)rr"rget_container_clientrr)rr2r,r(r)_r-rrrrverify_storage_uriDs" z#AzureEnvironment.verify_storage_urics|dS)z*Verify that the AzureEnvironment is valid.N)rrrrrrWs zAzureEnvironment.verifycCs>t|}|durtd|d|d|d|dfS)zParse an Azure blob storage URI into a storage account and container. Args: uri (str): The URI to parse. Returns: Tuple[str, str, prefix]: The storage account, container, and path. NzCould not parse Azure blob URI r3r r)r matchrgroup)r2r8rrrr"[s zAzureEnvironment.parse_uri)r N)T)r' __module__ __qualname____doc__r classmethoddictboolrrrrstrr/r1r6r staticmethodtupler"rrrrr s   "r N)r<azure.core.exceptionsrazure.identityrazure.storage.blobrrtyping_extensionsrerrorsrutilsr abstractr r rrrrs