o xi@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z d d lmZer>dd lmZdd lmZe d ddjZe dddjZGdddeZdS)z/Implementation of AzureContainerRegistry class.) annotationsN) TYPE_CHECKING)AzureEnvironment) LaunchError)"AZURE_CONTAINER_REGISTRY_URI_REGEX) get_module)AbstractRegistry)ContainerRegistryClient)ResourceNotFoundErrorzazure.containerregistryzThe azure-containerregistry package is required to use launch with Azure. Please install it with `pip install azure-containerregistry`.)requiredzazure.core.exceptionszmThe azure-core package is required to use launch with Azure. Please install it with `pip install azure-core`.c@sNeZdZdZ   ddddZedd d ZdddZdddZdddZ dS)AzureContainerRegistryz8Helper for accessing Azure Container Registry resources.Nuri str | None registry_name repo_namecCs|durF||_tdd||fDrtd|jdr&|jtdd|_t|j}|dur8td|j|d|_|d|_ dStd d||fDrUtd ||_||_ |jd |j |_dS) z%Initialize an AzureContainerRegistry.Ncss|]}|duVqdSN.0xrrf/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/registry/azure_container_registry.py +z2AzureContainerRegistry.__init__..zFPlease specify either a registry name and repo name or a registry URI.https://.Unable to parse Azure Container Registry URI: rcss|]}|duVqdSrrrrrrr9rzDPlease specify both a registry name and repo name or a registry URI. .azurecr.io/) ranyr startswithlenrmatchgrouprr)selfrrrr!rrr__init__"s,    zAzureContainerRegistry.__init__configdictreturncCs$|d}|dur td||dS)aCreate an AzureContainerRegistry from a config dict. Args: config (dict): The config dict. environment (AbstractEnvironment): The environment to use. verify (bool, optional): Whether to verify the registry. Defaults to True. Returns: AzureContainerRegistry: The registry. Raises: LaunchError: If the config is invalid. rNz=Please specify a registry name to use under the registry.uri.)r)getr)clsr%rrrr from_configAs z"AzureContainerRegistry.from_configtuple[str, str]cst)z1Get username and password for container registry.)NotImplementedErrorr#rrrget_username_password\sz,AzureContainerRegistry.get_username_password image_uristrboolc stt|}|durtd||d}|d}|d}t}td|d|}z |||WdSt yAYd St yT}ztd ||d}~ww) zCheck if image exists in container registry. Args: image_uri (str): Image URI to check. Returns: bool: True if image exists, False otherwise. Nrrrrz .azurecr.ioTFz=Unable to check if image exists in Azure Container Registry: ) rer!rrr"rget_credentialsr get_manifest_propertiesr Exception) r#r/r!registry repositorytag credentialclienterrrcheck_image_exists`s0      z)AzureContainerRegistry.check_image_existscs|jd|jS)Nr)rrr-rrr get_repo_uri}sz#AzureContainerRegistry.get_repo_uri)NNN)rrrrrr)r%r&r'r )r'r+)r/r0r'r1)r'r0) __name__ __module__ __qualname____doc__r$ classmethodr*r.r=r>rrrrr s    r )rB __future__rr3typingr.wandb.sdk.launch.environment.azure_environmentrwandb.sdk.launch.errorsrwandb.sdk.launch.utilsr wandb.utilrabstractr azure.containerregistryr azure.core.exceptionsr r rrrrs2