dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z d Zd Ze j j"ee j j$ee j j&eiZd Zd ZdZe j j"fdZe j j"dfdZdZdZdZdZdZy)z$Utilities for the cloudbuild v2 API.)absolute_import)division)unicode_literalsN)apis)base) resources)resource_property cloudbuildv2z)projects/.*/locations/.*/memberships/(.*)z/projects/.*/locations/.*/workerPoolSecondGen/.*z1projects/.*/locations/.*/workerPoolSecondGen/(.*)cDtjtt|S)zReturns the messages module for Cloud Build. Args: release_track: The desired value of the enum googlecloudsdk.calliope.base.ReleaseTrack. Returns: Module containing the definitions of messages for Cloud Build. )rGetMessagesModule _API_NAMERELEASE_TRACK_TO_API_VERSION) release_tracks 7lib/googlecloudsdk/api_lib/cloudbuild/v2/client_util.pyr r .s#    <] K MMTcJtjtt|| S)a.Returns an instance of the Cloud Build client. Args: release_track: The desired value of the enum googlecloudsdk.calliope.base.ReleaseTrack. use_http: bool, True to create an http object for this client. Returns: base_api.BaseApiClient, An instance of the Cloud Build client. )no_http)rGetClientInstancerr)ruse_https rrr<s'   "=1| rc t}t}|dk(rbtjj |dd|||d}|j j |j|j}|S|dk(rbtjj |dd|||d }|jj |j|j} | Sy ) zGet a PipelineRun/TaskRun. pipelinerunz*cloudbuild.projects.locations.pipelineRunsr ) projectsId locationsIdpipelineRunsId) collection api_versionparamsnametaskrunz&cloudbuild.projects.locations.taskRuns)rr taskRunsIdN) rr rREGISTRYParseprojects_locations_pipelineRunsGet1CloudbuildProjectsLocationsPipelineRunsGetRequest RelativeNameprojects_locations_taskRuns-CloudbuildProjectsLocationsTaskRunsGetRequest) projectregionrun_idrun_typeclientmessagespipeline_run_resource pipeline_runtask_run_resourcetask_runs rGetRunr5Ms   &  ( %..44?!!$ 5 99==BB&335 C 89L 9!**00;!!  1 1155>>"//1 ? 45H Orctjt|}|r|jdSt dtd)aPGet the name part of a cluster membership's full resource name. For example, "projects/123/locations/global/memberships/cluster2" returns "cluster2". Args: resource_name: A cluster's full resource name. Raises: ValueError: If the full resource name was not well-formatted. Returns: The cluster's short name. z1The cluster membership resource name must match "")researchCLUSTER_NAME_SELECTORgroup ValueError resource_namematchs rClusterShortNamerAos; )))= 9% ;;q>), --rct}t}|jj|j dj |S)zGet the list of supported Cloud Build locations. Args: project: The project to search. Returns: A CloudbuildProjectsLocationsListRequest object. z projects/{}r)rr projects_locationsList&CloudbuildProjectsLocationsListRequestformat)r+r/r0s r ListLocationsrGsN  &  (  " " ' '55##G,6 rctjt|}|r|jdSt dt d)aXGet the name part of a worker pool second gen's full resource name. E.g. "projects/abc/locations/def/workerPoolSecondGen/ghi" returns "ghi". Args: resource_name: A worker pool second gen's full resource name. Raises: ValueError: If the full resource name was not well-formatted. Returns: The worker pool's short name. r7z5The worker pool second gen resource name must match "r8)r9r:#WORKERPOOL_SECOND_GEN_NAME_SELECTORr<r="WORKERPOOL_SECOND_GEN_NAME_MATCHERr>s rWorkerPoolSecondGenShortNamerKs; ))7 G% ;;q>69 ::rcjg}|jD]}|j|j}|jr|s-|0t j |j}t |dr2t|D]#}|jdj||%|j||S)aProduce field paths from a message object. The result is used to create a FieldMask proto message that contains all field paths presented in the object. https://github.com/protocolbuffers/protobuf/blob/master/src/google/protobuf/field_mask.proto Args: msg: A user defined message object that extends the messages.Message class. https://github.com/google/apitools/blob/master/apitools/base/protorpclite/messages.py Returns: The list of field paths. all_fieldsz{}.{}) rMget_assigned_valuer repeatedr ConvertToSnakeCasehasattrMessageToFieldPathsappendrF)msgfieldsfieldvr fs rrRrRs &~~e uzz*A ~~a}  1 1%** =d L !$Q'A --tQ/ 0(  d  -r)__doc__ __future__rrrr9googlecloudsdk.api_lib.utilrgooglecloudsdk.calliopergooglecloudsdk.corergooglecloudsdk.core.resourcer rGA_API_VERSION ReleaseTrackGABETAALPHArr;rJrIr rr5rArGrKrRrrres+&' ,():   .N^  E6#9$ %)$5$5$8$8 M%)$5$5$8$84"D-,&:*r