v( dZddlmZddlmZddlmZddlmZddlmZddlZddlm Z m Z m Z ddl Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z#dZ$e jJdGddejLZ'e jJdGddejPZ)e jJdGddejPZ*e jJdGdd ejLZ+e jJdGd!d"ejLZ,e jJdGd#d$ejPZ-e jJdGd%d&ejPZ.e jJdGd'd(ejLZ/e jJd)d*Gd+d,e'Z0e jJdGd-d.e'Z1e jJdGd/d0e'Z2d1Z3 dkd2Z4 dld3Z5e jJdGd4d5ejLZ6e jJdGd6d7ejLZ7e jJdGd8d9ejLZ8e jJdGd:d;ejLZ9dejLZ;e jJdGd?d@ejLZ<e jJdGdAdBejLZ=e jJdGdCdDejLZ>e jJd)d*GdEdFe'Z?e jJd)d*GdGdHe'Z@e jJdGdIdJe'ZAe jJdGdKdLe'ZBe jJdGdMdNejPZCe jJdGdOdPejPZDe jJdGdQdRejZFe jJdGdSdTejLZGe jJdGdUdVejLZHe jJdGdWdXejLZIe jJdGdYdZejLZJe jJdGd[d\e'ZKed]eLeLd^ZMe jJdGd_d`e'ZNe jJdGdadbejLZOe jJdGdcddejLZPe jJdGdedfejPZQe jJdGdgdhejPZRe jJdGdidjejPZSy)mzBClass for representing various changes to a Cloud Run V2 resource.)absolute_import) annotations)division)print_function)unicode_literalsN)IterableMappingSequence) TypedDict) duration_pb2)launch_stage_pb2)base)config_changes) exceptions)flags)resource_name_conversion)volumes)instance_split)k8s_min)vendor_settings) worker_poolc6|D]}|j|}|S)zApply ConfigChangers to resource. It's undefined whether the input resource is modified. Args: resource: The Cloud Run V2 resource to modify. changes: List of ConfigChangers. Returns: Changed resource. )Adjust)resourcechanges config_changes 7lib/googlecloudsdk/command_lib/run/v2/config_changes.py WithChangesr+s#m##H-H /T)frozencdeZdZUdZdZded<dZded<ej d dZ d Z d Z y) ContainerConfigChangerzAn abstract class representing worker pool container configuration changes. Attributes: container_name: Name of the container to modify. If None the first container is modified. N str | Nonecontainer_nameFboolnon_ingress_typecy)zMutate the given container. This method is called by this class's Adjust method and should apply the desired changes directly to container. Args: container: the container to adjust. Nself containers rAdjustContainerz&ContainerConfigChanger.AdjustContainerHsrcJ|j|}|j||S)aNReturns a modified Cloud Run V2 resource. Adjusts Cloud Run V2 resource by applying changes to the container specified by self.container_name if present or the first container otherwise. Calls AdjustContainer to apply changes to the selected container. Args: resource: The Cloud RUn V2 resource to modify. )_FindContainerr,)r*rr+s rrzContainerConfigChanger.AdjustVs'##H-I# Orc|jj}|jX|D]}|j|jk(s|cS|j t j |j|dS|s't j d}|j ||jr|dS|D]}|js|cS|dS)a Find the container to adjust. 1. Make ''(empty string) name referenceable. 2. If name is specified, but not found, create one with the name. 3. No name specified, no containers, add one with name ''. 4. No name specified, we have containers, return the first one for non-ingress types. 5. For ingress types, return the ingress container. If not found, fall-back to the first one. Args: resource: The Cloud Run V2 resource to modify. Returns: The container to adjust. )namer) template containersr$r0appendr Containerr&ports)r*rr4cr+s rr.z%ContainerConfigChanger._FindContainerds"""--J &! 66T(( (())t/B/BCD ^ %%2. )$   !}A WWH!}rr+zk8s_min.Container) __name__ __module__ __qualname____doc__r$__annotations__r&abcabstractmethodr,rr.r(rrr"r"<sH $.*# D  "   )rr"c,eZdZUdZded<ded<dZy)SetAnnotationChangezRepresents the user intent to set an annotation. Attributes: key: Annotation to set. value: Annotation value to set. strkeyvaluecL|j|j|j<|SN)rErrDr*rs rrzSetAnnotationChange.Adjusts%)ZZH" OrNr:r;r<r=r>rr(rrrBrB ( *rrBc"eZdZUdZded<dZy)DeleteAnnotationChangezdRepresents the user intent to delete an annotation. Attributes: key: Annotation to delete. rCrDcT|j}|j|vr ||j=|SrG)rrDr*rrs rrzDeleteAnnotationChange.Adjusts*&&K xx; dhh  OrNrIr(rrrLrL  (rrLc,eZdZUdZded<ded<dZy)SetTemplateAnnotationChangezRepresents the user intent to set a template annotation. Attributes: key: Template annotation to set. value: Annotation value to set. rCrDrEc`|j|jj|j<|SrG)rEr3rrDrHs rrz"SetTemplateAnnotationChange.Adjusts%.2jjH!!$((+ OrNrIr(rrrQrQrJrrQc"eZdZUdZded<dZy)DeleteTemplateAnnotationChangezuRepresents the user intent to delete a template annotation. Attributes: key: Template annotation to delete. rCrDch|jj}|j|vr ||j=|SrG)r3rrDrNs rrz%DeleteTemplateAnnotationChange.Adjusts0##//K xx; dhh  OrNrIr(rrrTrTrOrrTc"eZdZUdZded<dZy)SetLaunchStageChangez]Sets launch stage on a resource. Attributes: launch_stage: The launch stage to set. base.ReleaseTrack launch_stagec|jtjjk7r8tj j |jj|_|SrG)rYr ReleaseTrackGAr LaunchStageValueidrHs rrzSetLaunchStageChange.AdjustsK D--000.::@@     h OrNrIr(rrrWrWs "!rrWc,eZdZUdZded<ded<dZy)SetClientNameAndVersionChangezSets the client name and version. Attributes: client_name: Client name to set. client_version: Client version to set. rC client_nameclient_versioncz|j|j|_|j|j|_|SrG)rbclientrcrHs rrz$SetClientNameAndVersionChange.Adjusts: #((ho & $ 3 3h OrNrIr(rrrarasrrac"eZdZUdZded<dZy)ServiceAccountChangezRepresents the user intent to change service account for the revision. Attributes: service_account: The service account to set. rCservice_accountc<|j|j_|S)zCMutates the given config's service account to match what's desired.)rhr3rHs rrzServiceAccountChange.Adjusts(,(<(rqr, __classcell__rus@rrlrl s *-!rrlc.eZdZUdZdZded<dZddZy)AddDigestToImageChangezAdd image digest that comes from source build. Attributes: image_digest: The image digest to set in the adjusted container. Nr# image_digestc d|vS)z0Return true if the given image url is by-digest.z@sha256:r()r*urls r _IsDigestz AddDigestToImageChange._IsDigest's  rc|j|jry|jr"|jdz|jz|_yy)N@)rrmr|r)s rr,z&AddDigestToImageChange.AdjustContainer+s> ~~ioo&  !#-0A0AAiorr9)r:r;r<r=r|r>rr,r(rrr{r{s ", !Brr{cXeZdZUdZdZded<dZded<dZded<d fd Zd dZ xZ S) ResourceLimitsChangea\Represents the user intent to update resource limits. Attributes: memory: Updated memory limit to set in the container. Specified as string ending in 'Mi' or 'Gi'. If None the memory limit is not changed. cpu: Updated cpu limit to set in the container if not None. gpu: Updated gpu limit to set in the container if not None. Nr#memorycpugpuc t|di|tj|d|tj|d|tj|d|y)Nrrrr(ro)r*rrrrtrus rrqzResourceLimitsChange.__init__AsI Gv tXv. tUC( tUC(rc|j#|j|jjd<|j#|j|jjd<|jZ|jdk(r'|jjj ddy|j|jjd<yy)zCMutates the given config's resource limits to match what's desired.Nrr0znvidia.com/gpu)r resourceslimitsrrpopr)s rr,z$ResourceLimitsChange.AdjustContainerGs {{-1[[i  * xx*.((i  ' xx S""&&'7>7;xx ""#34 r)NNNr9) r:r;r<r=rr>rrrqr,rxrys@rrr2s3&*#z#z) @rrcz|jjD]"}|jdk(s|js |cSy)z7Returns the Cloud SQL volume for the given worker pool.cloudsqlN)r3rr0cloud_sql_instance)rvolumes r_GetCloudSQLVolumerTs6!!))f {{j V%>%> m* rc|jd}t|dk(r|St|dk(r8|stjd|stjd|d|d|Stjdj |)zZValidates and augments the given Cloud SQL instance with the project and region if needed.:zJTo specify a Cloud SQL instance by plain name, you must specify a project.zITo specify a Cloud SQL instance by plain name, you must specify a region.z&Malformed CloudSQL instance string: {})splitlenr CloudSQLErrorformat) instance_strprojectregioninstances r_AugmentSQLInstancer\s    $(]a  8}   $ $     $ $  Yaxq //  " "077 E rcB|Dcgc]}t|||c}Scc}w)z[Validates and augments the given Cloud SQL instances with the project and region if needed.)r) instancesrrrs r_AugmentSQLInstancesrzs1FO EN(GV4Y  scdeZdZUdZdZded<dZded<eje Z ded<d Z y) AddCloudSQLChangesaARepresents the intent to append the given Cloud SQL instances to the current list. Attributes: project: Project to use as the default project for Cloud SQL instances. region: Region to use as the default region for Cloud SQL instances add_cloudsql_instances: List of Cloud SQL instances to append. Nr#rrdefault_factory Sequence[str]add_cloudsql_instancesc t|}|{tjdtjt |j |j |j}|jjj||S|jjjt |j |j |j|SNr)r)r0r)rrVolumeCloudSqlInstancerrrrr3rr5rrextendr*rcloud_sql_volumes rrzAddCloudSQLChanges.Adjusts)(3 $55,--t||T[[ &&'78 O ))33:: ))4<<  Or) r:r;r<r=rr>r dataclassesfieldlistrrr(rrrrs?':&**;+*;*;+-rrcdeZdZUdZdZded<dZded<eje Z ded<d Z y) RemoveCloudSQLChangesaFRepresents the intent to remove the given Cloud SQL instances from the current list. Attributes: project: Project to use as the default project for Cloud SQL instances. region: Region to use as the default region for Cloud SQL instances remove_cloudsql_instances: List of Cloud SQL instances to remove. Nr#rrrrremove_cloudsql_instancesct|}|tt|j|j|j }|j jDcgc]}||vr| c}|j _|j jsB|jjDcgc]}|jdk7r|c}|j_ |Scc}wcc}wNr) rsetrrrrrrr3rr0)r*rrinstances_to_removerrs rrzRemoveCloudSQLChanges.Adjusts)(3# ,,dllDKK +==GG7Gh 0 0 G7))3  0 0 : :#++33% 3{{j( 3% ! O7% s  C'C) r:r;r<r=rr>rrrrrrr(rrrrs?':&*->[->->.]rrceZdZdZdZy)ClearCloudSQLChangesa;Represents the intent to clear the current list of Cloud SQL instances. Attributes: project: Project to use as the default project for Cloud SQL instances. region: Region to use as the default region for Cloud SQL instances clear_cloudsql_instances: Whether to clear the Cloud SQL instances. c|jjDcgc]}|jdk7r|c}|j_|Scc}wr)r3rr0)r*rrs rrzClearCloudSQLChanges.AdjustsN ''//!/F ;;* $ /!H O !sANr:r;r<r=rr(rrrrs rrcdeZdZUdZdZded<dZded<eje Z ded<d Z y) SetCloudSQLChangesaIRepresents the intent to replace the current list of Cloud SQL instances with the given list. Attributes: project: Project to use as the default project for Cloud SQL instances. region: Region to use as the default region for Cloud SQL instances set_cloudsql_instances: List of Cloud SQL instances to set. Nr#rrrrset_cloudsql_instancesc t|}|{tjdtjt |j |j |j}|jjj||St |j |j |j|j_ |Sr) rrrrrrrrr3rr5rrrs rrzSetCloudSQLChanges.Adjusts)(3 $55,--t||T[[ &&'78 O7K  % %t||T[[7))3 Or) r:r;r<r=rr>rrrrrrr(rrrrs?':&**;+*;*;+-rr?c"eZdZUdZded<dZy)RevisionNameChangezRepresents the user intent to change revision name. Attributes: revision_suffix: Suffix to append to the revision name. rCrevision_suffixc"|jsd|j_|Stt |jz dz }t j |j^}}dj|d||j|j_|S)zAMutates the given config's revision name to match what's desired.r2rz{}-{}N) rr3revision_MAX_RESOURCE_NAME_LENGTHrrGetInfoFromFullNamer0r)r*rmax_prefix_length_r0s rrzRevisionNameChange.Adjusts   #%h o "C(<(<$==A(;;HMMJHQ!(  $"6"6"H OrNrIr(rrrrs  rrc,eZdZUdZded<ded<dZy) MeshChangeaRepresents the user intent to enable/disable Cloud Service Mesh. Attributes: project: The project to use for the mesh when not specified in mesh_name. mesh_name: Mesh resource name in the format of MESH_NAME or projects/PROJECT/locations/global/meshes/MESH_NAME. rCr mesh_namec|jrd|jvr |jnd|jd|j|jj_|S)N/z projects/z/locations/global/meshes/)rrr3 service_meshmeshrHs rrzMeshChange.Adjust9sT~~!6 &??O P ""' OrNrIr(rrrr,s , .rrc"eZdZUdZded<dZy) GpuTypeChangezfSets the gpu-type on the resource template. Attributes: gpu_type: The gpu_type value to set. rCgpu_typec|jr'|j|jj_|Sd|jj_|SrG)rr3 node_selector acceleratorrHs rrzGpuTypeChange.AdjustLsC }}48MMh%%1 O59h%%1 OrNrIr(rrrrBs -rrc"eZdZUdZded<dZy)GpuZonalRedundancyChangezSets the gpu-zonal-redundancy-disabled field on the resource template. Attributes: gpu_zonal_redundancy: The gpu_zonal_redundancy value to set. r%gpu_zonal_redundancyc>|j |j_|SrG)rr3gpu_zonal_redundancy_disabledrHs rrzGpuZonalRedundancyChange.Adjust^s#  % %% 3 OrNrIr(rrrrTs rrc6eZdZUdZded<fdZddZxZS)ContainerCommandChangezRepresents the user intent to change the 'command' for the container. Attributes: command: The command to set in the adjusted container. rCcommandc Rt|di|tj|d|y)Nrr(ro)r*rrtrus rrqzContainerCommandChange.__init__os% Gv tY0rc&|j|_yrG)rr)s rr,z&ContainerCommandChange.AdjustContainerss Irr9rwrys@rrres ,1%rrc6eZdZUdZded<fdZddZxZS)ContainerArgsChangezRepresents the user intent to change the 'args' for the container. Attributes: args: The args to set in the adjusted container. list[str]argsc Rt|di|tj|d|y)Nrr(ro)r*rrtrus rrqzContainerArgsChange.__init__s% Gv tVT*rc&|j|_yrG)rr)s rr,z#ContainerArgsChange.AdjustContainers YYINrr9rwrys@rrrws /+rrceZdZUdZej eZded<ej e Z ded<dZ ded <d d Z y ) EnvVarLiteralChangeszRepresents the user intent to modify environment variables string literals. Attributes: updates: Updated env var names and values to set. removes: Env vars to remove. clear_others: If true clear all non-updated env vars. rdict[str, str]updatesrremovesFr% clear_othersct|j}|jr+|Dcgc] }d|vs|j|jvr|"}}g}|D]}d|vrX|j|jvr.t j dj|j|j|_|j|jvrx|j|jvrC|j|j|_ |jj|j|j||jjD]+\}}|jtj||-||_ycc}w)aMutates the given config's env vars literals to match the desired changes. Args: container: container to adjust Returns: The adjusted container Raises: ConfigurationError if there's an attempt to replace the source of an existing environment variable whose source is of a different type (e.g. env var's secret source can't be replaced with a config map source). value_sourcezpCannot update environment variable [{}] to string literal because it has already been set with a different type.)r0rEN)renvrr0rrConfigurationErrorrr5rrEritemsrEnvVar)r*r+current_env_varsenv_varupdated_env_vars env_var_name env_var_values rr,z$EnvVarLiteralChanges.AdjustContainersc IMM* *)g w &',,$,,*F ) # 7 " <<4<< '--HHN,,I   (  %  % W\\2  &g&-$2(,||'9'9';# m ..l- @(<%IMIs%FNr9) r:r;r<r=rrdictrr>rrrr,r(rrrrsF.K--dC'>C({((>'9>,7%rrceZdZUdZej eZded<ej e Z ded<dZ ded <d d Z dd Z y )SecretsEnvVarChangeszRepresents the user intent to modify environment variable secrets. Attributes: updates: Updated env var names and values to set. removes: Secret env vars to remove. clear_others: If true clear all non-updated secret env vars. rrrrrFr%rcP|jd}t|dk(r%tjdj |t|dk(r|\}}n$tjdj |t j t j||S) zFBuilds a secret env var source from the given secret name and version.rrz^No secret version specified for {secret}. Use {secret}:latest to reference the latest version.)secretz#Invalid secret name and version: {})rversion)secret_key_ref)rrrrrr EnvVarSourceSecretKeySelector)r*rparts secret_namesecret_versions r_BuildSecretEnvVarSourcez-SecretsEnvVarChanges._BuildSecretEnvVarSources LL E 5zQ  ) ) AAGBHB   Uq$)!k>  ) ) / 6 6v >    00"  rc Tt|j}|jr+|Dcgc] }d|vs|j|jvr|"}}g}|D]}d|vrX|j|jvr.t j dj|j|j|_|j|jvrx|j|jvrR|j|j|j|_ |jj|j|j||jjD]:\}}|jtj||j|<||_ycc}w)aMutates the given config's secrets env vars to match the desired changes. Args: container: container to adjust Returns: The adjusted container Raises: ConfigurationError if there's an attempt to replace the source of an existing environment variable whose source is of a different type (e.g. env var's secret source can't be replaced with a config map source). rzwCannot update environment variable [{}] to secret source env var because it has already been set with a different type.)r0rN)rrrr0rrrrr5rrrrrrr)r*r+rrrrrs rr,z$SecretsEnvVarChanges.AdjustContainers IMM* *)g 7 *glldll.J ) # w & <<4<< '--HHN,,I   (  %  %#<< LL &  &g&1$6!% 2 2 4 f ..88@ !5%IMSs%F%N)rrCreturnzk8s_min.EnvVarSourcer9)r:r;r<r=rrrrr>rrrrr,r(rrrrsK.K--dC'>C({((>'9>,.<%rrc"eZdZUdZded<dZy)DescriptionChangezuRepresents the user intent to change resource description. Attributes: description: The description to set. rC descriptionc(|j|_|S)zAMutates the given resource's description to match what's desired.)r rHs rrzDescriptionChange.Adjust?s++H OrNrIr(rrr r 5s rr cBeZdZUdZdZded<dZded<dZded<d Zy) BinaryAuthorizationChangezRepresents the user intent to change binary authorization. Attributes: policy: The binauthz policy to set. breakglass_justification: The breakglass justification to set. clear_binary_authorization: Whether to clear binary authorization. Nr#policybreakglass_justificationFr%clear_binary_authorizationc|jr5d|j_d|j_d|j_|S|jdk(rd|j_n8|jr,d|j_|j|j_|jr|j|j_|S)zrrrr(rrrrEs.&*)-J-%*d*rrceZdZUdZej eZded<ej e Z ded<dZ ded <e d Z d Zy ) LabelChangeaGRepresents the user intent to modify metadata labels. Attributes: additions: {str: str}, any label values to be updated subtractions: List[str], any labels to be removed clear_labels: bool, whether to clear the labels Raises: ValueError: if both subtractions and clear are specified at the same time. rr additionsr subtractionsFr% clear_labelscy)NTr()r*s radjusts_templatezLabelChange.adjusts_templateys rc.|jr|jr td|jr>|jj |j jj |jj |j|j jj |j|jD]D}|jj|d|j jj|dF|S)NzCCannot specify both subtractions and clear_labels at the same time.) rr ValueErrorlabelsclearr3updaterr)r*r label_keys rrzLabelChange.Adjust}s T..  O   oo$$& OO4>>* ##DNN3&& oo)T*""9d3' OrN)r:r;r<r=rrrrr>rrrpropertyrrr(rrrrhsX 0k//E)^E-K--dC, C,  rrceZdZUdZdZded<dZded<dZded<eje Z ded <d Z d ed <d Z d ed <dZy)VpcAccessChangesaRepresents the user intent to modify VPC access. Attributes: vpc_egress: str, the vpc egress setting (all traffic, private ranges only). network: str, vpc network to set. subnet: str, vpc subnet to set. network_tags: List[str], vpc network tags to set. clear_network: bool, whether to clear the network. clear_network_tags: bool, whether to clear the network tags. Nr# vpc_egressnetworksubnetrr network_tagsFr% clear_networkclear_network_tagsc|jrZd|jj_tj j j|jj_|S|jjjsK|jjjjtj j|jjjd}|jr|j|_ |jr|j|_ |jr|j|_|j rg|_|j"dk(s|j"dk(r?tj j j$|jj_|S|j"dk(r=tj j j&|jj_|S)Nrallz all-trafficzprivate-ranges-only)r*r3 vpc_accessnetwork_interfacesr VpcAccess VpcEgressVPC_EGRESS_UNSPECIFIEDegressr5NetworkInterfacer'r( subnetworkr)tagsr+r& ALL_TRAFFICPRIVATE_RANGES_ONLY)r*rnetwork_interfaces rrzVpcAccessChanges.Adjusts 8r'r(rrrr)r*r+rr(rrr%r%sX  *j':&*-K--dC, C-"d"rr%cleZdZUdZdZded<dZded<dZded<dZd ed <dZ d ed <dZ d ed <d Z y)CmekKeyChangesaRepresents the user intent to add cmek key related changes. Attributes: key: str, cmek key to set. post_key_revocation_action_type: str, vpc network to set. encryption_key_shutdown_hours: int, the number of hours to wait before an automatic shutdown server after CMEK key revocation is detected. clear_key: bool, whether to clear any previously set CMEK key reference. clear_post_key_revocation_action_type: bool, whether to clear any previously set post CMEK key revocation action type. clear_encryption_key_shutdown_hours: bool, whether to clear any previously set CMEK key shutdown hours setting. Nr#rDpost_key_revocation_action_typez int | Noneencryption_key_shutdown_hoursFr% clear_key%clear_post_key_revocation_action_type#clear_encryption_key_shutdown_hoursc|j|j|j_|jrR|jdk(rtj j ntj j|j_|j2tj|jdz|j_ |jrKd|j_tj j|j_d|j_ |jr:tj j|j_d|j_ |j rd|j_ |S)Nz prevent-newi)seconds)rDr3encryption_keyr<rEncryptionKeyRevocationAction PREVENT_NEWSHUTDOWN encryption_key_revocation_actionr=r Duration encryption_key_shutdown_durationr>,ENCRYPTION_KEY_REVOCATION_ACTION_UNSPECIFIEDr?r@rHs rrzCmekKeyChanges.Adjusts5 xx)-h& ++  1 1] B  7 7 C C<<EE8  ))5   884? 8  ~~)-h&  7 7 d d8<@h8 11  7 7 d d8<@h8 //;?h8 Or) r:r;r<r=rDr>r<r=r>r?r@rr(rrr;r;sN #z04!:4.22)T05'5.3%t3!rr;c,eZdZUdZded<ded<dZy)RemoveVolumeChangezxRemoves volumes from the service/job/worker-pool template. Attributes: removed_volumes: The volumes to remove. Iterable[str]removed_volumesr% clear_volumescj|jr&|jjj|S|jro|jjDcic]}|j |}}|jD] }||vs||= t |j|j_|Scc}wrG)rOr3rr rNr0rvalues)r*rrcurrent_volumes to_removes rrzRemoveVolumeChange.Adjust s %%' O   ,4,=,=,E,E,E&&++v ,E++)  'i(,#''='='?"@h OsB0NrIr(rrrLrLs !  rrLcXeZdZUdZded<ej jZded<dZ y)AddVolumeChangezUpdates Volumes set on the service/job/worker-pool template. Attributes: new_volumes: The volumes to add. release_track: The resource's release track. Used to verify volume types are supported in that release track. z Iterable[volumes_lib.VolumeDict] new_volumesrX release_trackc<|jjDcic]}|j|}}|jD]1}t j ||j }|||j<3t|j|j_|Scc}wrG) r3rr0rV volumes_libcreate_volume_v2rWrrQ)r*rrrR new_volumes rrzAddVolumeChange.Adjust)s+3*;*;*C*C*C V*C""//8J8JKj)3ojoo&#!%_%;%;%= >H OsBN) r:r;r<r=r>rr[ALPHArWrr(rrrUrUs+0/%)%6%6%<%<-"< rrUcVeZdZUdZej eZded<dZ ded<dZ y ) RemoveVolumeMountChangezRemoves Volume Mounts from the container. Attributes: removed_mounts: Volume mounts to remove from the adjusted container. clear_mounts: Whether to clear all volume mounts. rrMremoved_mountsFr% clear_mountsc|jr|jj|St|j}|jDcgc]}|j |vr|c}|_|Scc}wrG)r` volume_mountsr rr_ mount_path)r*r+r_ volume_mounts rr,z'RemoveVolumeMountChange.AdjustContainerAsz ##%  4../n(55!5l  $ $N : 5!i  !s A-N) r:r;r<r=rrrr_r>r`r,r(rrr^r^5s/#4+"3"3D"I.-I, rr^VolumeMountDict)r mount-pathcJeZdZUdZej eZded<ddZ y)AddVolumeMountChangezvUpdates Volume Mounts set on the container. Attributes: new_mounts: Mounts to add to the adjusted container. rzIterable[VolumeMountDict] new_mountsc6|jDcic]}|j|}}|jD]B}d|vsd|vrtjdt j |d|d||d<Dt|j|_|Scc}w)Nrrfz-nU<() !#>#8#8#:;I sBNr9) r:r;r<r=rrrrir>r,r(rrrhrhWs+ +<+*;*;+*'rrhc"eZdZUdZded<dZy)RemoveContainersChangezcRemoves the specified containers. Attributes: containers_to_remove: Containers to remove. rMcontainers_to_removect|j}|jjDcgc]}|j|vr|c}|j_|Scc}wrG)rrpr3r4r0)r*rrpr+s rrzRemoveContainersChange.Adjust~s`t889"**55$5I >>!5 5 5$H O $sANrIr(rrrorots &%rrocHeZdZUdZej eZded<dZ y)ContainerDependenciesChangeaFSets container dependencies. Updates container dependencies to add the dependencies in new_dependencies. Additionally, dependencies to or from a container which does not exist will be removed. Attributes: new_dependencies: A map of containers to their updated dependencies. Defaults to an empty map. rzMapping[str, Iterable[str]]new_dependenciesc t|jjDcgc]}|jc}}|jjDcic],}|j|jDcgc] }||vs| c}.}}}|j j D]V\}}t|}||z x}r+tjd|ddj|d|rt|||<T||=X|jjD]#}|j|jg|_%|Scc}wcc}wcc}}w)zUpdates container dependencies.z--depends_on for container z$ references nonexistent containers: ,.) rr3r4r0 depends_onrtr frozensetrrjoinsortedget) r*rr+current_containersr8current_dependenciesr$rxmissings rrz"ContainerDependenciesChange.Adjustsg)1):):)E)EF)EI)EF"**55 6I  ++ +!q4F/FA+  6  '+&;&;&A&A&C" Z(j!33 3 3++).)9:((+(9':! =   /5j/A^,  0'D&&11 155innbIi2 O/ G s#D6E/ D;9D;=E;EN) r:r;r<r=rrrrtr>rr(rrrsrss- 3D+2C2C3/rrscDeZdZUdZdZded<dZded<dZded<d dZy) WorkerPoolScalingChangea0Represents the user intent to adjust worker pool scaling. Attributes: min_instance_count: The minimum count of instances to set. max_instance_count: The maximum count of instances to set. scaling: Scaling flag value that either contains manual instance count or auto scaling mode. Nzflags.ScaleValue | Nonemin_instance_countmax_instance_countzflags.ScalingValue | Nonescalingc|jj}|js |jrH|js<|tj j jk(rtjd|jr|jr+|jjstjd|jjrd|j_n%|jj|j_|jr|jr+|jjstjd|jjrd|j_n%|jj|j_|jr|jjrFd|j_ tj j j|j_|Sd|j_d|j_tj j j|j_|jj|j_ |S)aAdjusts worker pool scaling. Args: worker_pool_resource: The worker pool resource to modify. Raises: ConfigurationError: If the user attempts to set min or max instance count without setting --scaling=auto when the current scaling mode is manual. ConfigurationError: If the user attempts to set min or max instance count along with a manual instance count using --scaling flag. Returns: The adjusted worker pool resource. zBNeed to specify --scaling=auto to swtich mode from manual to auto.zBCannot set --min when --scaling is set to a manual instance count.rzBCannot set --max when --scaling is set to a manual instance count.dN)r scaling_moderrrWorkerPoolScaling ScalingModeMANUALrr auto_scalingrestore_defaultinstance_countmanual_instance_count AUTOMATIC)r*worker_pool_resourcecurrent_scaling_modes rrzWorkerPoolScalingChange.Adjusts"077DD D$;$;  , , 8 8 ? ? @  ) ) N   dll77++ P    0 0:;$$7  # # 2 2 $$7  dll77++ P    0 0:=$$7  # # 2 2 $$7 ||  " "=A$$:  - - 9 9 C C $$1 ;?$$7:>$$7  - - 9 9 @ @ $$1 LL ' ' $$: r)rworker_pool_objects.WorkerPool) r:r;r<r=rr>rrrr(rrrrs215-404-4'+' $+G rrceZdZdZddZy)NoPromoteChangezRRepresents the user intent to block instance assignment for a new worker revision.c|jstjdtj|j }t jt|j||_ |S)znRemoves LATEST from the worker pools instance assignments and assign the percent to the latest ready revision.z;--no-promote not supported when creating a new worker pool.) generationrrrGetNameFromFullChildNamelatest_ready_revisioninstance_split_libZeroLatestAssignmentrinstance_splits)r*rlatest_ready_revision_names rrzNoPromoteChange.Adjustsl     ) ) G  !99  * *  2FF X % %&(B H OrNrrrr(rrrrs ZrrcXeZdZUdZdZded<ejeZ ded<d dZ y ) InstanceSplitChangezRepresents the user intent to change a worker pool's instance split assignments. Attributes: to_latest: Whether to assign all traffic to the latest revision. to_revisions: The new instance split percentages to set based on revision. Fr% to_latestrzdict[str, int] to_revisionscn|jrd|jj|jjt j tj jd|S|jr8tjt|j|j|_|S)Nr)typepercent) rrr r5r InstanceSplitInstanceSplitAllocationType%INSTANCE_SPLIT_ALLOCATION_TYPE_LATESTrrGetUpdatedSplitsrrHs rrzInstanceSplitChange.Adjust4s ~~$$&%%  & &!==cc  O   !3!D!D x'' ($*;*;"h OrNr) r:r;r<r=rr>rrrrrr(rrrr(s/)T!2!2!24!H,H rr)rrCrr#rr#rrC)rrrr#rr#rr)Tr= __future__rrrrrr?collections.abcrr r rtypingr cloudsdk.google.protobufr google.apir googlecloudsdk.calliopergooglecloudsdk.command_lib.runrrrrrrY!googlecloudsdk.command_lib.run.v2rr;googlecloudsdk.generated_clients.gapic_clients.run_v2.typesrrrworker_pool_objectsr dataclassTemplateConfigChangerr"NonTemplateConfigChangerrBrLrQrTrWrargrlr{rrrrrrrrrrrrrrrrrr r ConfigChangerrr%r;rLrUr^rCrerhrorsrrrr(rrrslI&"%' 771'(950CARVOWj"d#P^AAP$Phd# .AA $  d# ^DD $  d# ."F"F $  d# ^%I%I $  d#>BB$"d#N$K$K$&d# >?? $ E$/!(!0!"d#B3B$B(d#@1@$@B   <   d##==#$#Ld#%N@@%$%Pd#>??$(d#!==!$!Hd#==$2d#55$*d#N88$"d# ~CC $  E$/%3%0%"E$/00"d#D%1D%$D%Nd#`%1`%$`%Fd# ?? $ d# G G$Dd#&...&$&Rd#1~;;1$1hd#7^997$7td#==$2d#n::$0d#4$0d#1$8d#^AA$&d#*."F"F*$*bd#U nEEU $U pd#n==$(d#.AA$r