GdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl mZ ddl m Zdd lmZdd lmZdd lmZdd lmZd ZdZdZdZdZdZdZdZdZ d?dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+d@d!Z,dAd"Z- dBd#Z. dCd$Z/ dDd%Z0 dDd&Z1d?d'Z2d(Z3d)Z4 d?d*Z5d+Z6dAd,Z7d-Z8d.Z9d/Z:d0Z;d1Zd4Z?d5Z@d6ZAd7ZB dAd8ZCd9eDd:eEd;eEfd<ZFd=ZGd>ZHy )EzUtility for making API calls.)absolute_import)division)unicode_literals) exceptions) http_wrapper) list_pager)iam)util)apis)iam_util) resourcesartifactregistryv1storagezstorage.objects.listz/cloudkms.projects.locations.keyRings.cryptoKeysc>tjttSN)r GetClientInstanceSTORAGE_API_NAMESTORAGE_API_VERSION4lib/googlecloudsdk/command_lib/artifacts/requests.pyGetStorageClientr+   02E FFrc>tjttSr)r GetMessagesModulerrrrrGetStorageMessagesr/rrc|jdk\rtjj|t j |S)z5Wrap http_wrapper.CheckResponse to skip retry on 501.i) status_codeapitools_exceptions HttpError FromResponser CheckResponse)responses rSkipRetryOn500Errorsr%3s8 S  ' ' 4 4X >>  # #H --rc\tjtt|}t|_|S)N)skip_activation_prompt)r rARTIFACTREGISTRY_API_NAMEARTIFACTREGISTRY_API_VERSIONr%check_response_func)r'clients r GetClientr,:s*  ! !"3 & 4& -rc>tjttSr)r rr(r)rrr GetMessagesr.Ds   9 < >>rc6tjtdSNv1beta2)r rr(rrrGetClientV1beta2r2I   9 ) ++rc6tjtdSr0)r rr(rrrGetMessagesV1beta2r5Nr3rc|j|}|jj|}t||js%t j dj||y)zDeletes a tag by its name.namezFailed to delete tag {}: {}N)FArtifactregistryProjectsLocationsRepositoriesPackagesTagsDeleteRequest-projects_locations_repositories_packages_tagsDelete isinstanceEmpty ar_exceptionsArtifactRegistryErrorformat)r+messagestagdelete_tag_reqerrs r DeleteTagrESsjbb c.<<CC # C (  - -%,,S#6 88 )rc|j|j|j}|j|j ||j }|j j|S)z7Creates a tag associated with the given docker version.)r8version)parentrBtagId)Tag GetTagNameGetVersionNameFArtifactregistryProjectsLocationsRepositoriesPackagesTagsCreateRequestGetPackageNamerBr:Create)r+rA docker_tagdocker_versionrBcreate_tag_reqs rCreateDockerTagrS^st  "N,I,I,K  M#bb  & & (ccI.  = = D D rc\|j|}|jj|S)zGets a tag by its name.r7)CArtifactregistryProjectsLocationsRepositoriesPackagesTagsGetRequestr:Get)r+rArB get_tag_reqs rGetTagrXhs2\\ ]+  = = A A+ NNrc\|j|}|jj|S)zDeletes a version by its name.r7)JArtifactregistryProjectsLocationsRepositoriesPackagesVersionsDeleteRequest1projects_locations_repositories_packages_versionsr;)r+rArGdelete_ver_reqs r DeleteVersionr]os7ff g.  A A H H rc\|j|}|jj|S)zDeletes a package by its name.r7)BArtifactregistryProjectsLocationsRepositoriesPackagesDeleteRequest(projects_locations_repositories_packagesr;)r+rApackagedelete_pkg_reqs r DeletePackagercws2^^ _.  8 8 ? ? OOrct}t}|j|}|jj |S)zGets a version by its name.r7)r,r.rUr:rV)r+rArG get_ver_reqs r GetVersionrf~sE ;& ](JJK  = = A A+ NNrcF|j|}|jj|}|jr't |jj ddk7r$t jdj||jj ddS)z"Gets a version name by a tag name.r7/ z!Internal error. Corrupted tag: {}) rUr:rVrGlensplitr>r?r@)r+rArBrW get_tag_ress rGetVersionFromTagrns\\ ]+DDHH+   K$7$7$=$=c$B Cr I  - -+2237 99    " "3 ' ++rNc |j||}ttj|j||ddS)z;Lists all tags under a package with the given package name.rHfilterpageSizetags batch_sizebatch_size_attributefield)DArtifactregistryProjectsLocationsRepositoriesPackagesTagsListRequestlistr YieldFromListr:)r+rAra page_size server_filter list_tags_reqs rListTagsr~sM__ ]`,-   > > )  rc |j|dj|}ttj|j ||ddS)z-Lists tags associated with the given version.z version="{}"rprrrsrt)rxr@ryrrzr:)r+rArarGr{r}s rListVersionTagsrsY__ -44W=`?-   > > )  rc |j|||}ttj|j|||ddS)z&Lists all packages under a repository.rHorderByrqrrpackageslimitrurvrw)@ArtifactregistryProjectsLocationsRepositoriesPackagesListRequestryrrzr`)r+rArepor{order_byrr| list_pkgs_reqs r ListPackagesrsYOOx P?   9 9 )  rc |}||||kr|}|j||||} ttj|j| ||ddS)z#Lists all versions under a package.)rHviewrrqrrversionsr)HArtifactregistryProjectsLocationsRepositoriesPackagesVersionsListRequestryrrzr[) r+rApkg version_viewr{rrr| page_limit list_vers_reqs r ListVersionsrsr* ]y,U1BJcc |XmdM-   B B )  rc t}t}|j|||}tt j |j ||ddS)z'Lists all repositories under a project.rrr repositoriesrt)r,r.8ArtifactregistryProjectsLocationsRepositoriesListRequestryrrzprojects_locations_repositories)projectr{rr|r+rAlist_repos_reqs rListRepositoriesrse ;& ](GG(=HB   0 0 )   !!rc |j|||}ttj|j||ddS)z#Lists all files under a repository.)rHrqrrrfilesrt)=ArtifactregistryProjectsLocationsRepositoriesFilesListRequestryrrz%projects_locations_repositories_files)r+rArr|r{rlist_files_reqs r ListFilesrsVLLmXM?   6 6 )  rct|}t}|j|}|jj |}|S)z#Gets the repository given its name.r7)r,r.7ArtifactregistryProjectsLocationsRepositoriesGetRequestrrV)rr'r+rA get_repo_req get_repo_ress r GetRepositoryrsI + ,& ](QQ R,77;;LI, rct}t}|j|}|jj |}|S)z1Gets the IAM policy for the specified repository.)resource)r,r.@ArtifactregistryProjectsLocationsRepositoriesGetIamPolicyRequestr GetIamPolicy)repo_resr+rAget_iam_policy_reqget_iam_policy_ress rrrsJ ;& ](``a==JJ rct}t}|j||j|}|jj |S)z1Sets the IAM policy for the specified repository.)policy)rsetIamPolicyRequest)r,r.@ArtifactregistryProjectsLocationsRepositoriesSetIamPolicyRequestSetIamPolicyRequestr SetIamPolicy)rrr+ ar_messagesset_iam_policy_reqs rrrsW ;& +#cc%999Hd  / / < <=O PPrct|}t}|jdj|||jj dd|}|j j|S)ajCreates the repository given its parent. Args: project: str: The project to create the repository in. location: str: The region to create the repository in. repository: messages.Repository to create. skip_activation_prompt: bool: If true, do not prompt for service activation Returns: The resulting operation from the create request. zprojects/{}/locations/{}rhrj)rH repositoryId repository)r,r.:ArtifactregistryProjectsLocationsRepositoriesCreateRequestr@r8rlrrO)rlocationrr'r+rArequests rCreateRepositoryrso + ,& ](  O O ' . .w A??((-b1 P '  / / 6 6w ??rct}t}|j|}|jj |}|S)z Gets the package given its name.r7)r,r.?ArtifactregistryProjectsLocationsRepositoriesPackagesGetRequestr`rV)rar+rAget_package_reqget_package_ress r GetPackager,sH ;& ](\\ ]/CCGG/ rct}t}|jd|z}tj|j ||dd}t |Dcgc]}|jc}Scc}w)z(Lists all locations for a given project. projects/r7rr locationsrt)r2r5,ArtifactregistryProjectsLocationsListRequestrrzprojects_locationssorted locationId) project_idr{r+rA list_locs_reqrlocs r ListLocationsr7sx  &  !(GG  #H%-&& %  ) 959C95 665sA1ct}t}|j|t|}|jj |S)zETests storage IAM permission for a given bucket for the user project.)bucket permissions userProject)rr'StorageBucketsTestIamPermissionsRequest_GCR_PERMISSIONbucketsTestIamPermissions)rrr+rAtest_reqs rTestStorageIAMPermissionrFsD  &  !(  = = g>G(  * *8 44rcvtjj|t}t j |S)z+Gets the IAM policy for a given crypto key. relative_name collection)r REGISTRYParseRelativeNameCRYPTO_KEY_COLLECTIONkms_iamGetCryptoKeyIamPolicy)kms_keycrypto_key_refs rGetCryptoKeyPolicyrOs4%%77(=8?.  & &~ 66rcztjj|t}t j ||dS)z;Adds Encrypter/Decrypter role to the given service account.rz*roles/cloudkms.cryptoKeyEncrypterDecrypter)r rrrrAddPolicyBindingToCryptoKey)rservice_accountrs rAddCryptoKeyPermissionrVs>%%77(=8?.  , ,o2 44rctj\}}|jj|j t j |S)z)Gets the service account given its email.r7)iam_apiGetClientAndMessagesprojects_serviceAccountsrV$IamProjectsServiceAccountsGetRequestr EmailToAccountResourceName)rr+rAs rGetServiceAccountr_sR113&(  ( ( , ,3322?C4E FFrct}t}|jd|zdz}|jj |S)Nr/projectSettingsr7)r,r.1ArtifactregistryProjectsGetProjectSettingsRequestprojectsGetProjectSettings)rr+rAget_settings_reqs rrrgsK ;& ](OO  #&8 8P:  + +,< ==rct}t}|jd|zdz|zdz}|jj |S)z/Gets VPC SC Config on the project and location.r /locations/ /vpcscConfigr7)r,r.6ArtifactregistryProjectsLocationsGetVpcscConfigRequestrGetVpcscConfig)r location_idr+rA get_vpcsc_reqs rGetVPCSCConfigrosY ;& ](QQ  #m 3k A R-  " " 1 1- @@rct}t}|jd|zdz|zdz|jjj}|j d|zdz|zdz|}|j j|S)z=Allows requests in Remote Repository inside VPC SC perimeter.rrrr8 vpcscPolicyr8 vPCSCConfig)r,r. VPCSCConfigVpcscPolicyValueValuesEnumALLOW9ArtifactregistryProjectsLocationsUpdateVpcscConfigRequestrUpdateVpcscConfig)rrr+rAvcupdate_vpcsc_reqs rAllowVPCSCConfigrys ;& ](  #m 3k A &&AAGGI"WW  #m 3k A X  " " 4 45E FFrct}t}|jd|zdz|zdz|jjj}|j d|zdz|zdz|}|j j|S)z=Denies requests in Remote Repository inside VPC SC perimeter.rrrrr)r,r.rrDENYrrr)rrr+rArrs rDenyVPCSCConfigr s ;& ](  #m 3k A &&AAFFH"TT  #m 3k A U-  " " 4 4] CCrcjt}t||jjjSr)r.SetUpgradeRedirectionStateProjectSettings%LegacyRedirectionStateValueValuesEnumREDIRECTION_FROM_GCR_IO_ENABLEDrrAs rEnableUpgradeRedirectionrs- ]( #(**PP&& ((rcjt}t||jjjSr)r.r rr REDIRECTION_FROM_GCR_IO_DISABLEDrs rDisableUpgradeRedirectionrs- ]( #(**PP'' ))rcjt}t||jjjSr)r.r rr!REDIRECTION_FROM_GCR_IO_FINALIZEDrs rFinalizeUpgradeRedirectionrs- ]( #(**PP(( **rct}t}|j|}d}|r||_|j d|zdz||}|j j |S)z >s CCrct}t}|r]|j|jjj |j ||j }nj|r]|j|jjj |j ||j }n t d|jj|S)z&Exports an artifact by version or tag.)gcsPath sourceVersion)rexportArtifactRequest)r7 sourceTagz(Either version or tag must be specified.) r,r.BArtifactregistryProjectsLocationsRepositoriesExportArtifactRequestParent RelativeNameExportArtifactRequest ValueErrorrExportArtifact)rGrBgcs_destinationr+rAr5s rr@r@s ;& ](  U U>>#**,99;&<<#!..0=  V C  U U::<&&(557&<<#&&(=  V C ? @@  / / > >s CCr)F)NNr)NNNN)NNNNN)NNN)I__doc__ __future__rrrapitools.base.pyrr rr googlecloudsdk.api_lib.artifactsr>googlecloudsdk.api_lib.cloudkmsr rgooglecloudsdk.api_lib.iamr rgooglecloudsdk.api_lib.utilr googlecloudsdk.command_lib.iamr googlecloudsdk.corer r(r)rrrrrrr%r,r.r2r5rErSrXr]rcrfrnr~rrrrrrrrrrrrrrrrrrr rrrr strintr+r2r@rrrrMsn$&'>)'H:6,3)/#(IGG.> + + 8OPO ,  48:>  6:JN&)-26!"59'+  Q;@@. 7574F>A G D()*15D,SS& DDr