'dZddlmZddlmZddlmZddlmZgdZgdZgdZ d Z d Z d Z d Z d ZdZdZdZdZdZy)aHandles the updating of PolicyControllerPolicyControllerDeploymentConfig. Each function updates a single value, diving to the appropriate depth, updating and returning the updated object. Note that while client-side validation could occur here, it is deferred to the API layer. )absolute_import)division)unicode_literals) exceptions)z cpu-limitz cpu-requestz memory-limitzmemory-request tolerationz replica-countz pod-affinity) NoSchedulePreferNoSchedule NoExecute) admissionauditmutationc|jjjjDcic]}|j|j c}Scc}w)zExtracts a dictionary of deployment configuration by component name. Args: spec: A hub membership spec. Returns: A dictionary mapping component name to configuration object. )policycontrollerpolicyControllerHubConfigdeploymentConfigsadditionalPropertieskeyvalue)speccfgs Ulib/googlecloudsdk/command_lib/container/fleet/policycontroller/deployment_configs.pyget_configurationsr5sN&&@@RRgg g# ggsyyg  sAc<| d|_|St||_|S)zDConfigures a replica count for the current deployment configuration.N) replicaCountint)currentrs rupdate_replica_countrDs) ]G .u:G .c8|j}|j |j}|j}|j |j}||_|j|j d}||_|j|j d}||_|S)aConfigures a cpu limit for the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. value: The value to set the cpu limit to. If None, the limit will be removed. If this is the only limit, limit requirements will be removed. If this is the only requirement, requirements will be removed. Returns: The modified deployment configuration object. N)$PolicyControllerResourceRequirementscontainerResourcesPolicyControllerResourceListlimitscpumemoryrequestsmessagesrr requirements resource_lists rupdate_cpu_limitr+Ms>>@, +--L779-$ ''M-=#7#7#?M%, \%:%:%BL+' .rc:|j |j}n|j}|j}|j |j}||_|j |jd}||_|j|j d}||_|S)aConfigures a memory limit for the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. value: The value to set the memory limit to. If None, the limit will be removed. If this is the only limit, limit requirements will be removed. If this is the only requirement, requirements will be removed. Returns: The modified deployment configuration object. N)r!r r"r#r%r$r&r's rupdate_mem_limitr-js +--L@@BL779-$ ''M-=#7#7#?M%, \%:%:%BL+' .rc:|j |j}n|j}|j}|j |j}||_|j|j d}||_|j |jd}||_|S)aConfigures a cpu request for the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. value: The value to set the cpu request to. If None, the request will be removed. If this is the only request, request requirements will be removed. If this is the only requirement, requirements will be removed. Returns: The modified deployment configuration object. N)r!r r"r&r$r%r#r's rupdate_cpu_requestr/s +--L@@BL779-& ))M-=#7#7#?M', \%:%:%BL+' .rc:|j |j}n|j}|j}|j |j}||_|j |jd}||_|j |jd}||_|S)aConfigures a memory request for the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. value: The value to set the memory request to. If None, the request will be removed. If this is the only request, request requirements will be removed. If this is the only requirement, requirements will be removed. Returns: The modified deployment configuration object. N)r!r r"r&r%r$r#r's rupdate_mem_requestr1s +--L@@BL779-& ))M-=#7#7#?M', \%:%:%BL+' .rc|jd}t|dkDr$tjdj ||d}t|dk(r|dnd}t|dk(rdnd}|||fS)N=z)Illegal value for toleration key-value={}rExistsEqual)splitlenrErrorformat) key_valuesplit_key_valuerroperators r_parse_key_valuer?sOOC(/A   3::9E  # #O 4 9/! t%_-2X( eX rc|j}t|\}}}||_||_|||_|||_g}|j |j }|j|||_|S)aAdds a toleration to the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. key_value: the key-and-optional-value string specifying the toleration key and value. effect: Optional. If included, will set the effect value on the toleration. Returns: The modified deployment configuration object. )PolicyControllerTolerationr?r>rreffectpodTolerationsappend) r(rr<rBrrrr> tolerationss radd_tolerationrFs224*))4#uh **. J J+ '((K Z &' .rc|j}t|\fd}|Dcgc] }||r |c}|_|Scc}w)a Removes a toleration from the current deployment configuration. A toleration must match exactly to be removed - it is not enough to match the key, or even key-value. The effect must also match the toleration being removed. Args: current: the deployment configuration object being modified. key_value: the key-and-optional-value string specifying the toleration key and value. effect: Optional. If included, will set the effect value on the toleration. Returns: The modified deployment configuration object. c|jk(xr1|jk(xr |jk(xr|jk(S)N)rrr>rB)rrBrr>rs rmatchz remove_toleration..matchsO 3  *    & *  H , *  & ( r)rCr?) rr<rBcurrent_tolerationsrItrr>rs ` @@@rremove_tolerationrLsS  ..))4#uh(;K':!%(A':K' .Ls AAc|dk(r'|jjj|_|S||dk(r'|jjj|_|St j dj|)aConfigures the pod affinity for the current deployment configuration. Args: messages: the set of proto messages for this feature. current: the deployment configuration object being modified. value: The value to set the pod affinity to. If the value is the string "none" or value `None`, the pod affinity will be NO_AFFINITY. Returns: The modified deployment configuration object. antinonez)invalid pod affinity option {} specified.)0PolicyControllerPolicyControllerDeploymentConfigPodAffinityValueValuesEnum ANTI_AFFINITY podAffinity NO_AFFINITYrr:r;)r(rrs rupdate_pod_affinityrUs f_AA\\jj  . }AA\\hh  .   3::5A rN)__doc__ __future__rrrgooglecloudsdk.corerSUPPORTED_PROPERTIESK8S_SCHEDULING_OPTIONSG8R_COMPONENTSrrr+r-r/r1r?rFrLrUrrr]sb ''* :<<< >Br