o xi.!@sdZddlmZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZeeZe d d d Ze d d d e_e ddd ej_GdddeZdS)zasz2GoogleArtifactRegistry.__init__..z~The Google Artifact Registry must be specified with either the uri key or the repository, image-name, project and region keys.-docker.pkg.dev//zoThe Google Artifact Registry is missing the {} kwarg. Please specify it by name or as part of the uri argument. image-name)_loggerinfor anyrrmatchgrouprrrrformatr authdefault credentialsrefresh transportrequestsRequest)selfr rrrrr_missing_kwarg_msgrrr__init__-sP     $ zGoogleArtifactRegistry.__init__configdictcCsbhd}t||}|rtd|d|d||d|d|d|d|d d S) atCreate a Google Artifact Registry from a config. Arguments: config: A dictionary containing the following keys: repository: The repository name. image-name: The image name. environment: A GcpEnvironment configured for access to this registry. Returns: A GoogleArtifactRegistry. >r typerrrrz@The Google Artifact Registry config contains unacceptable keys: z5. Please remove these keys. The acceptable keys are: .r rrrr)r rrrr)setkeysrget)clsr,acceptable_keysunacceptable_keysrrr from_config|s"z"GoogleArtifactRegistry.from_configtuple[str, str]cs,|jjs|jtjjjd|jjfS)z|Get the username and password for the registry. Returns: A tuple of the username and password. oauth2accesstoken)r$tokenr%r r"r&r'r(r)rrrget_username_passwords z,GoogleArtifactRegistry.get_username_passwordstrcs$|jd|jd|jd|jS)zGet the URI for the given repository. Arguments: repo_name: The repository name. Returns: The repository URI. rr)rrrrr:rrr get_repo_uris z#GoogleArtifactRegistry.get_repo_uri image_uriboolc std|d|d\}}|IdH}||kr(td|d|jdd|jd |jd |j}t t j j j }||jd IdH}t |j}z|d |id IdHD] } || jvraWdSqVWdSt jjjy}} z td|jd| d} ~ ww)zCheck if the image exists. Arguments: image_uri: The image URI. Returns: True if the image exists, False otherwise. zChecking if image z exists:Nz The image z, does not match to the image uri repository r/z projects/z /locations/z/repositories/)r$parent)requestTz(The Google Artifact Registry repository zH does not exist. Please create it or modify your registry configuration.F)rrsplitr=rr rrrrr cloudartifactregistryArtifactRegistryClientr$list_docker_imagestagsapi_core exceptionsNotFound) r)r>repo_uritag self_repo_urirAartifact_registry_clientclient list_imagesimageerrrcheck_image_existss>    z)GoogleArtifactRegistry.check_image_exists)NNNNN) r r rr rr rr rr rr)r,r-rr )rr7)rr<)r>r<rr?) __name__ __module__ __qualname____doc__r+ classmethodr6r;r=rTrrrrr $s  O  ' r )rX __future__rlogging google.authr google.cloud.artifactregistrywandb.sdk.launch.errorsrwandb.sdk.launch.utilsrr wandb.utilrabstractr getLoggerrUrr"rDrEr rrrrs.