\dZddlmZddlmZddlmZddlZddlZddlmZ ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlZdZdZdZej>j@eej>jBeej>jDeiZ#dZ$dZ%dZ&dZ'dZ(GddejRZ*ej>j@fdZ+ej>j@fdZ,ej>j@ddfdZ-dZ.dZ/d Z0d!Z1d"Z2d#Z3d6d$Z4d%Z5d&Z6 d7d'Z7 d7d(Z8 d7d)Z9 d6d*Z: d6d+Z;d,Zd8d/Z?d8d0Z@d1ZAd2ZBd3ZCd4ZDd5ZEy)9z!Utilities for the cloudbuild API.)absolute_import)division)unicode_literalsN)messages)encoding)cloudbuild_exceptions)apis)base) exceptions)yaml)resource_property)files cloudbuildv1v1beta1z'projects/.*/locations/.*/workerPools/.*z)projects/.*/locations/.*/workerPools/(.*)z)projects/.*/locations/(.*)/workerPools/.*globali@ceZdZdZdZdZy)WorkerpoolTypesrN)__name__ __module__ __qualname__UNKNOWNPRIVATEHYBRID8lib/googlecloudsdk/api_lib/cloudbuild/cloudbuild_util.pyrr8s ' ' &rrcDtjtt|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. )r GetMessagesModule _API_NAMERELEASE_TRACK_TO_API_VERSION release_tracks rr!r!>s#    <] K MMrcDtjtt|S)zReturns the client class for Cloud Build. Args: release_track: The desired value of the enum googlecloudsdk.calliope.base.ReleaseTrack. Returns: base_api.BaseApiClient, Client class for Cloud Build. )r GetClientClassr"r#r$s rr'r'Ls#   Y9-H JJrTFcLtjtt|| |S)aReturns 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. skip_activation_prompt: bool, True to skip prompting for service activation. Should be used only if service activation was checked earlier in the command. Returns: base_api.BaseApiClient, An instance of the Cloud Build client. )no_httpskip_activation_prompt)r GetClientInstancer"r#)r%use_httpr*s rr+r+Zs*$   "=1|3  rc`|sytj||jjdS)NT) sort_items)apitools_encodingDictToAdditionalPropertyMessageBuildSubstitutionsValue) substitutionsrs rEncodeSubstitutionsr4ts.    : :X^^664 IIrc|syg}ttj|D]'\}}|j|j ||)||S)NkeyvalueadditionalProperties)sortedsix iteritemsappendAdditionalProperty)r3 value_typesubstitution_propertiesr7r8s rEncodeTriggerSubstitutionsrB|sY  3==78jc5""%%#U%;=9 )@ AArc|s|Si}|r*|jD]}|j||j<tj|D] \}}|||< g}t |j D];\}}|j|jjj||=|jj|S)a6Encodes the trigger substitutions for the update command. Args: old_substitutions: The existing substitutions to be updated. substitutions: The substitutions to be added to the existing substitutions. messages: A Cloud Build messages module. Returns: The updated trigger substitutions. r6r9) r:r8r7r<r=r;itemsr> BuildTriggerr2r?)old_substitutionsr3rsubstitution_mapsubr7r8updated_substitutionss r!EncodeUpdatedTriggerSubstitutionsrJs   55"%))sww6MM-0jc5!S1+1134jc5  00CC5 D 5    1 10 2 rc|syg}|rj|jD][}|j|vs|j|jjj |j|j ]|s3|j|jjj |jj |S)a Removes existing substitutions for the update command. Args: old_substitutions: The existing substitutions. substitutions_to_be_removed: The substitutions to be removed if exist. messages: A Cloud Build messages module. Returns: The updated trigger substitutions. Nr6r9)r:r7r>rEr2r?r8)rFsubstitutions_to_be_removedrrArHs rRemoveTriggerSubstitutionsrMs %  55 3 3&&  ! ! 4 4 G GGG399 H  6 !""00CCE    1 12 2 rc|jjjg}|jj|S)Nr9)rEr2r?)rrAs rEncodeEmptyTriggerSubstitutionsrOsE..AAC    1 12 2 rc|jd}|s|Sd}|D] }|s|dz } n|r||d}|sd|dz zSd|z|dz|d<dj|dd|ddDcgc]}|jc}zScc}w)zChange a snake_case string into a camelCase string. Args: snake: str, the string to be transformed. Returns: str, the transformed string. _rrN)splitjoin capitalize)snakepartsleading_blankspss rSnakeToCamelStringr[s ++c %  L . a n   ./ "E NQ& ''^#eAh.E!H reABi@iallni@@ AA@sA= c |g}t|trBtj|Dcic]!\}}t |||vr t ||n|#c}}St|t r|Dcgc]}t ||c}S|Scc}}wcc}w)aRecursively transform all keys and values from snake_case to camelCase. If a key is in skip, then its value is left alone. Args: msg: dict, list, or other. If 'other', the function returns immediately. skip: contains dict keys whose values should not have camel case applied. Returns: Same type as msg, except all strings that were snake_case are now CamelCase, except for the values of dict keys contained in skip. ) isinstancedictr<r=r[ SnakeToCamellist)msgskipr7valelems rr_r_s \ DT c* +HC 3$'tOc4  =*  #t14 5Lt $ 55 J  6s &B+B cg}|jD]}|j|j}|jr|s-|0|jdk(rd}n1|jdk(rd}nt 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. privatePoolV1Configprivate_pool_v1_confighybridPoolConfighybrid_pool_config all_fieldsz{}.{}) rjget_assigned_valuenamerepeatedr ConvertToSnakeCasehasattrMessageToFieldPathsr>format)rafieldsfieldvrlfs rrprps &~~e uzz*A ~~a} , ,' ::+ +# 33EJJ? L !$Q'A --tQ/ 0(  d% & -rcNtj||}fd||S)aPStuff a dict into a proto message, and fail if there are unused values. Args: obj: dict(), The structured data to be reflected into the message type. msg_type: type, The proto message type. Raises: ValueError: If there is an unused value in obj. Returns: Proto message, The message that was created from obj. c6t|tjr|j}|r?t |dkDrddj t |z}n|d}td|z|jD]!} t||j}|#t|trt|D]\}} |yy#t$r}td|j|d}~wwxYw#t$r}td||fzd}~wwxYw) z8Check for any unused fields in nested messages or lists.rz{%s},rz .%s: unused.Nz[%d]%s) r]proto_messagesMessageall_unrecognized_fieldslenrTr; ValueErrorrjgetattrrlr` enumerate) obj unused_fields unused_msg used_fieldrseiitem_CheckForUnusedFieldss rrz1_UnpackCheckUnused.._CheckForUnusedFieldsCs#~--.113m  }  ! )> ??*%Q'*344(* ;#z/%  &) #ts^'!T .  %$ ;jooq9: : ;  .8q!f,- - .s0CC: C7C22C7: DDD)r/ DictToMessage)rmsg_typerars @r_UnpackCheckUnusedr4s+ ''X6#.6 *rc|g} tj||}t |tstj|dt|||||S#tj$r%}tj||j d}~wwxYw)aLoad a proto message from a stream of JSON or YAML text. Args: stream: file-like object containing the JSON or YAML data to be decoded. msg_type: The protobuf message type to create. msg_friendly_name: A readable name for the message type, for use in error messages. skip_camel_case: Contains proto field names or map keys whose values should not have camel case applied. path: str or None. Optional path to be used in error messages. Raises: ParserError: If there was a problem parsing the stream as a dict. ParseProtoException: If there was a problem interpreting the stream as the given message type. Returns: Proto message, The message that got decoded. N file_hintz Could not parse as a dictionary.) r loadErrorr ParserError inner_errorr]r^_YamlToMessage)streamrmsg_friendly_nameskip_camel_casepathstructured_datars rLoadMessageFromStreamrcs0OAii$7O OT *  + +D,N PP 3D' // A  + +D!-- @@AsAB % BB c |g} tj||}|Dcgc]}t |||||c}S#tj$r%}tj||j d}~wwxYwcc}w)aLoad multiple proto message from a stream of JSON or YAML text. Args: stream: file-like object containing the JSON or YAML data to be decoded. msg_type: The protobuf message type to create. msg_friendly_name: A readable name for the message type, for use in error messages. skip_camel_case: Contains proto field names or map keys whose values should not have camel case applied. path: str or None. Optional path to be used in error messages. Raises: ParserError: If there was a problem parsing the stream. ParseProtoException: If there was a problem interpreting the stream as the given message type. Returns: Proto message list of the messages that got decoded. Nr)r load_allrrrrr)rrrrrrrrs rLoadMessagesFromStreamrs0OAmmFd;O " !$T8%6N!  A  + +D!-- @@A s:A5A2 A--A2ct||} t||}|S#t$r}tj||d|zd}~wwxYw)azLoad a proto message from a file containing JSON or YAML text. Args: structured_data: Dict containing the decoded YAML data. msg_type: The protobuf message type to create. msg_friendly_name: A readable name for the message type, for use in error messages. skip_camel_case: Contains proto field names or map keys whose values should not have camel case applied. path: str or None. Optional path to be used in error messages. Raises: ParseProtoException: If there was a problem interpreting the file as the given message type. Returns: Proto message, The message that got decoded. z%sN)r_r ExceptionrParseProtoException)rrrrrrars rrrs[2!/B/> _h 7C * > 3 3D:K481H >> >s  A?Acvtj|5}t|||||cdddS#1swYyxYw)aLoad a proto message from a file containing JSON or YAML text. Args: path: The path to a file containing the JSON or YAML data to be decoded. msg_type: The protobuf message type to create. msg_friendly_name: A readable name for the message type, for use in error messages. skip_camel_case: Contains proto field names or map keys whose values should not have camel case applied. Raises: files.MissingFileError: If the file does not exist. ParserError: If there was a problem parsing the file as a dict. ParseProtoException: If there was a problem interpreting the file as the given message type. Returns: Proto message, The message that got decoded. N)r FileReaderrrrrrrus rLoadMessageFromPathrs4.  H.?!0$ 8/8cvtj|5}t|||||cdddS#1swYyxYw)aLoad a proto message from a file containing JSON or YAML text. Args: path: The path to a file containing the JSON or YAML data to be decoded. msg_type: The protobuf message type to create. msg_friendly_name: A readable name for the message type, for use in error messages. skip_camel_case: Contains proto field names or map keys whose values should not have camel case applied. Raises: files.MissingFileError: If the file does not exist. ParseProtoException: If there was a problem interpreting the file as the given message type. Returns: Proto message list of the messages that got decoded. N)rrrrs rLoadMessagesFromPathrs4,  !!X/@"14 9rcHttjt|S)zDetermine if the provided full resource name is a worker pool. Args: resource_name: str, The string to test. Returns: bool, True if the string is a worker pool's full resource name. )boolrematchWORKERPOOL_NAME_MATCHER) resource_names r IsWorkerPoolrs bhh. > ??rctjt|}|r|jdSt dt d)aBGet the name part of a worker pool's full resource name. For example, "projects/abc/locations/def/workerPools/ghi" returns "ghi". Args: resource_name: A worker pool's full resource name. Raises: ValueError: If the full resource name was not well-formatted. Returns: The worker pool's short name. r*The worker pool resource name must match "")rsearchWORKERPOOL_NAME_SELECTORgroupr~rrrs rWorkerPoolShortNamers; )),m <% ;;q>+. //rctjt|}|r|jdSt dt d)aQGet the region part of a worker pool's full resource name. For example, "projects/abc/locations/def/workerPools/ghi" returns "def". Args: resource_name: str, A worker pool's full resource name. Raises: ValueError: If the full resource name was not well-formatted. Returns: str, The worker pool's region string. rrr)rrWORKERPOOL_REGION_SELECTORrr~rrs rWorkerPoolRegionr0s; )). >% ;;q>+. //rctt}|j}|j|_|j|_|j |j |_|s|j|j|_ |jX|j}|j|_ |j|_|j|j|_||_|S|j#}|j$|_|j(|_|j,|_|j0|_||_|S)a.Construct the GitHubEnterpriseConfig resource from the command line args. Args: args: An argparse namespace. All the arguments that were provided to this command invocation. update: bool, if the args are for an update. Returns: A populated GitHubEnterpriseConfig message. )r!GitHubEnterpriseConfighost_urihostUrlapp_idappId webhook_key webhookKeypeered_network peeredNetwork gcs_bucket GCSLocationbucket gcs_objectobject generation appConfigJsonGitHubEnterpriseSecretsprivate_key_nameprivateKeyNamewebhook_secret_namewebhookSecretNameoauth_secret_nameoauthSecretNameoauth_client_id_nameoauthClientIdNamesecrets)argsupdaterghe gcs_locationsecret_locations rGitHubEnterpriseConfigFromArgsrEs ('')# #+kk#) !%%CN D''3++C __ '')L//L//L " $l$C * 668O%)%:%:O"(,(@(@O%&*&<&%)%C%C"-E* m$".. * +/F -rcf|ytj|}t|j S)NT)r/ MessageToDictanyvalues)message message_dicts rrrs1 _ "009, $$& ' ''rc|duxrL|jduxr<|jjduxr"|jjjduSN)optionspoolrl build_configs rWorkerPoolIsSpecifiedrs\$5   d *5    # #4 /5    # # ( ( 4 rc|duxrL|jduxr<|jjduxr"|jjjduSr )r r workerConfigrs rWorkerPoolConfigIsSpecifiedrs\$=   d *=    # #4 /=    # # 0 0 < rc|sy|tzS)zConverts bytes to GB. Args: size: a size in GB Does not require size to be a multiple of 1 GB unlike utils.BytesToGb from from googlecloudsdk.api_lib.compute Returns: size in bytes. N)BYTES_IN_ONE_GB)sizes r BytesToGbrs     rr )NN)F)F__doc__ __future__rrrenumrapitools.base.protorpcliterrzapitools.base.pyrr/!googlecloudsdk.api_lib.cloudbuildrgooglecloudsdk.api_lib.utilr googlecloudsdk.callioper r rgooglecloudsdk.corer googlecloudsdk.core.resourcer googlecloudsdk.core.utilrr<r"_GA_API_VERSION_BETA_API_VERSION ReleaseTrackGABETAALPHAr#rrrDEFAULT_REGIONrEnumrr!r'r+r4rBrJrMrOr[r_rprrrrrrrrrrrrrrrrrrrr,s(&' A:C,(>$:*   /O_  EGI dii %)$5$5$8$8 M"&!2!2!5!5 J##&&  4IB FBBB6"J, d+/# $/T,0 $ #R$( % V)-8>*.96 @/*/*" J) X!H( !r