dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z d Zd Zd Zd Zd Zej&dgdZej&dgdZdZdZdZdZddZdZdZy)z(Common utilities to operate with Docker.)absolute_import)division)unicode_literalsN)errors) local_util)logzcloudai-autogeneratedunnamedusPackage)script package_path python_moduleImage)name default_homedefault_workdirc|jddkDry|jdd}t|dk(rd|dvr t|S|dfS)aoParses out the repository and tag from a Docker image name. Args: image_name: (str) The full name of an image, expected to be in a format of "repository[:tag]" Returns: A (repository, tag) tuple representing the parsed result. None repository means the image name is invalid; tag may be None if it isn't present in the given image name. :)NN/N)countrsplitlentuple) image_namepartss 1lib/googlecloudsdk/command_lib/ai/docker/utils.py_ParseRepositoryTagr!)sWcQ    C #%Z1_E!H, < T c|jd}t|dk(rd|fSd|dvsd|dvr |d|ddfSd|fS)aParses a repository to an optional hostname and a list of path compoentes. Args: repository_name: (str) A name made up of slash-separated path name components, optionally prefixed by a registry hostname. Returns: A (hostname, components) tuple representing the parsed result. The hostname will be None if it isn't present; the components is a list of each slash-separated part in the given repository name. rrN.rr)splitr)repository_name componentss r _ParseRepositoryHostr(@sb$$S)*_  JqMSJqM1 a=*QR. (( z r"c|jdd}|djd}t|dk(r|dnd}||fS)aParses a registry hostname to a list of components and an optional port. Args: host: (str) The registry hostname supposed to comply with standard DNS rules, optionally be followed by a port number in the format like ":8080". Returns: A (hostcomponents, port) tuple representing the parsed result. The hostcomponents contains each dot-seperated component in the given hostname; port may be None if it isn't present. rrrr$rN)rr%r)hostrhostcomponentsports r _ParseHostPortr-YsF ++c1 %8>>#&.Z1_q$$  r"ct|\}}| tdt|tkDrtdj tt |\}}|r{t |\}}d}|D]2}tj||tdj |d} |r0tj| |tdj ||D]A} | s tdd} tj| | )td j | |rTt|tkDrtd j td } tj| | td yy) aaValidate the given image name is a valid repository/tag reference. As explained in https://docs.docker.com/engine/reference/commandline/tag/#extended-description, a valid repository/tag reference should following the below pattern: reference := name [ ":" tag ] name := [hostname '/'] component ['/' component]* hostname := hostcomponent ['.' hostcomponent]* [':' port-number] hostcomponent := /([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9-]*[a-zA-Z0-9])/ port-number := /[0-9]+/ component := alpha-numeric [separator alpha-numeric]* alpha-numeric := /[a-z0-9]+/ separator := /[_.]|__|[-]*/ tag := /[\w][\w.-]{0,127}/ Args: image_name: (str) Full name of a Docker image. Raises: ValueError if the image name is not valid. Nz#Unable to parse repository and tag.z4Repository name must not be more than {} characters.z5^(?:[a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9-]*[a-zA-Z0-9])$z.Invalid hostname/port "{}" in repository name.z^[0-9]+$z(Empty path component in repository name.z)^[a-z0-9]+(?:(?:[._]|__|[-]*)[a-z0-9]+)*$z/Invalid path component "{}" in repository name.z-Tag name must not be more than {} characters.z^[\w][\w.-]{0,127}$z Invalid tag.) r! ValueErrorr_MAX_REPOSITORY_LENGTHformatr(r-rematch_MAX_TAG_LENGTH) r repositorytaghostnamepath_componentsr+r,hostcomponent_regex hostcomponent port_regex componentcomponent_regex tag_regexs r ValidateRepositoryAndTagr?ms0( 3/*c : ;;_-- >EE " $ %%3:>(O )(3NDR' %} 5 = > E Eh OQ Q( J T*2  < C CH M OO#i  A BBBO xx+3  = D DY O QQ #  3x/! FMM  'I xx 3' ~ &&(  r"c>t|xst}tjjj d}dj t ||}|rC|rd}n|xst}dj |}dj ||jdd|S|S)zAGenerate a name for the Docker image built by AI platform gcloud.z%Y%m%d.%H.%M.%S.%fz{}/{}:{}zgcr.ioz{}-docker.pkg.devz{}/{}/{}rr) _SanitizeRepositoryName_DEFAULT_IMAGE_NAMEdatetimenowstrftimer1_AUTONAME_PREFIX_DEFAULT_REPO_REGIONreplace) base_nameprojectregionis_gcrsanitized_namer6rr5 region_prefixs r GenerateImageNamerOs*9+K8KL. (()=>#  !1>3G*  j4 4m&--mrks/&' 0@#* + !L N wJ L.2(B'J$K*>r"