IU>dZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZddl mZdd lmZdd lmZdd lmZd ddZdZej.ej0ej2j4ej6Gddej8Zej0ej2j<GddeZej0ej2j@GddeZ!ee_"y)zUpdate node pool command.)absolute_import)division)unicode_literals) exceptions)utils) api_adapter)util)base)flags)log) console_iozW *{command}* updates a node pool in a Google Kubernetes Engine cluster. z To turn on node autoupgrade in "node-pool-1" in the cluster "sample-cluster", run: $ {command} node-pool-1 --cluster=sample-cluster --enable-autoupgrade ) DESCRIPTIONEXAMPLESctj|dtj|dtj||j dt dddy) zRegister flags for this command. Args: parser: An argparse.ArgumentParser-like object. It is mocked out in order to capture some information, but behaves like an ArgumentParser. zThe name of the node pool.zThe name of the cluster.z --timeoutiTzTHIS ARGUMENT NEEDS HELP TEXT.)typedefaulthiddenhelpN)r AddNodePoolNameArgAddNodePoolClusterFlag AddAsyncFlag add_argumentint)parsers *lib/surface/container/node_pools/update.py_Argsr.sV6#?@v'ABV   + -c,eZdZdZedZdZdZy)Update)Updates a node pool in a running cluster.ct||jd}tj||j d}tj |dtj |dtj|ddddtj|}tj|dtj||j d}tj|tj|dtj|dtj|tj|dtj |tj"|dtj$|dtj&|dtj(|dd tj*|dd tj,|tj.|tj0|dtj2|dd tj4|dd tj6|dtj8|tj:|tj<|dtj>|tj@|tjB|dd tjD|tjF|tjH|tjJ|tjL||j d }tjN|d tjP|tjR|tjT|tjV|y)NTrequiredNode management for_node_poolFenable_gpu_partitionenable_gpu_sharingenable_gpu_driver_installationrrUpgrade settingsr& for_updater& is_updater& for_create Node configupdate),radd_mutually_exclusive_groupr AddNodePoolLocationsFlagadd_argument_groupAddEnableAutoRepairFlagAddEnableAutoUpgradeFlagAddAcceleratorArgsAddClusterAutoscalingFlagsAddNodePoolAutoprovisioningFlagAddWorkloadMetadataFlagAddEnableSurgeUpgradeFlagAddSurgeUpgradeFlagAddMaxUnavailableUpgradeFlagAddEnableBlueGreenUpgradeFlagAddStandardRolloutPolicyFlagAddAutoscaledRolloutPolicyFlagAddNodePoolSoakDurationFlagAddSystemConfigFlag AddLabelsFlagAddNodeLabelsFlagAddNodeTaintsFlagAddTagsNodePoolUpdateAddEnableGvnicFlagAddEnableImageStreamingFlagAddEnableConfidentialNodesFlagAddConfidentialNodeTypeFlag AddNetworkPerformanceConfigFlagsAddNodePoolEnablePrivateNodesAddEnableFastSocketFlagAddLoggingVariantFlagAddWindowsOsVersionFlagAddContainerdConfigFlagAddStoragePoolsFlag$AddResourceManagerTagsNodePoolUpdateAddQueuedProvisioningFlagAddMaxRunDurationFlagAddFlexStartFlag AddEnableKubeletReadonlyPortFlagAddMachineTypeFlagAddDiskTypeFlagAddDiskSizeFlag$AddBootDiskProvisionedThroughputFlagAddBootDiskProvisionedIopsFlagrgroupnode_management_groupautoscaling_groupupgrade_settings_groupnode_config_groups rArgsz Update.ArgsGs &M  / / / >E ""5)!445FG !!"7tL ""#8M  !'+ 88? ))*;EJ !!%("556HI ##$:; 4DI &&d4 ''(>? &&d4 (()?@ %%d4 eE2 T2 E$G E$G & U# %%e4@ (( TT3 %%e44P **5? ''. !!%( T: !!%( !!%(  Te5 ..u5 ##E* & 5! **5100? .t< +, +, ../@A (():;rc @tj|tjd0id|jd|j d|j d|jd|jd|jd|jd|jd |jd |jd |jd |jd |j d|j"d|j$d|j&d|j(d|j*d|j,d|j.d|j0d|j2d|j4d|j6d|j8d|j:d|j<d|j>d|j@d|jBd|jDd |jFd!|jHd"|jJd#|jLd$|jNd%|jPd&|jRd'|jTd(|jVd)|jXd*|jZd+t]|d,rt_j`|jbndd-|jdd.|jfd/|jhS)1N acceleratorsenable_autorepairenable_autoupgradeenable_autoscalingnode_locations max_nodes min_nodestotal_max_nodestotal_min_nodeslocation_policyworkload_metadataworkload_metadata_from_nodeenable_autoprovisioningmax_surge_upgrademax_unavailable_upgradesystem_config_from_filelabels node_labels node_taintstagsgvnicenable_image_streamingnetwork_performance_configenable_confidential_nodesconfidential_node_typeenable_blue_green_upgradeenable_surge_upgradenode_pool_soak_durationstandard_rollout_policyautoscaled_rollout_policyenable_private_nodesenable_fast_socketlogging_variantwindows_os_versioncontainerd_config_from_file storage_poolsenable_queued_provisioningmax_run_duration flex_start machine_type disk_type%enable_insecure_kubelet_readonly_port disk_size_gb disk_sizeresource_manager_tagsboot_disk_provisioned_iops boot_disk_provisioned_throughput)5r ValidateSurgeUpgradeSettingsrUpdateNodePoolOptions acceleratorrirjrkrlrmrnrorprqrrrsrtrurvrwrxryrzr{ enable_gvnicr}network_performance_configsrrrrrrrrrrrrrrrrrrrhasattrr BytesToGbrrrr)selfargss rParseUpdateNodePoolOptionsz!Update.ParseUpdateNodePoolOptionss( &&t,  , ,1 %%1 001  221  22 1  ** 1  .. 1 ..1 ,,1 ,,1 ,,1 001 %)$D$D1 !% < <1 001 !% < <1  !% < "66?1 @ 22A1 B,,C1 D 22E1 F%)$D$DG1 H((I1 J$(#B#BK1 L..M1 N??O1 P&&Q1 R..S1 T/3.X.XU1 X 4 %__T^^4 [1 \#88]1 ^$(#B#B_1 `*.)N)Na1 1rc|jd}|jd}||}|j|j|}|j|}|j7t j dj|jddd|j7t j d j|jd  dd|j7t j d j|jd dd|jrtj|j |j||}|j r9|j#|}|j%dstj&|_|S|j)|dj|j*|j,|j/|} tj0| t=j>|| S#t2j4$r)} t7j8| tj:d} ~ wwxYw)a4This is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: Cluster message for the successfully updated node pool. Raises: util.Error, if creation failed. r location_getNzSThe previous user-specified labels on this node pool will be replaced by '{labels}'ry)rxT)messagethrow_if_unattended cancel_on_nozSThe previous user-specified taints on this node pool will be replaced by '{taints}'rz)taintszOThe previous user-specified tags on this node pool will be replaced by '{tags}'r{)r{formatzUpdating node pool {0}) timeout_s) context ParseNodePoolnamerryr PromptContinuerGetValuerzr{r}r 2CheckForContainerFileSystemApiEnablementWithPrompt projectIdUpdateNodePoolasync_ GetOperation IsSpecifiedOPERATIONS_FORMATWaitForOperation nodePoolIdtimeout GetNodePoolCheckForCgroupModeV1apitools_exceptions HttpErrorr HttpExceptionHTTP_ERROR_FORMATr UpdatedResource) rradapterrlocationpool_refoptions operation_refoppoolerrors rRunz Update.Runsll=)G<</LD!H$$TYY9H--d3G&)*0&}5+1+7" &)*0&}5+1+7" ||,-3Vv9NV-O" %% ==   D,,Xw?m   ! !- 0)..$+    " ) )(*=*= >LL"  *d %! K  ( (D  $ $UD,B,B CCDs AH AHI'$I  IN)__name__ __module__ __qualname____doc__ staticmethodrfrrrrrrrAs&2><><@4lDrrc&eZdZdZedZdZy) UpdateBetar ct||jd}|jd}tj|dtj |dtj |ddddtj|}tj|d|jd}tj|tj|dtj|dtj|tj|dtj|tj|dtj |d tj"|tj$|dtj&|dtj(|dd tj*|dd tj,|tj.|tj0|dtj2|tj4|dtj6|dtj8|dd tj:|dd tj<|tj>|dtj@|tjB|tjD|tjF|dd tjH|tjJ|tjL|tjN||jd }tjP|dtjR|tjT|tjV|tjX|yNTr"r$r%Fr'r+r,)use_moder-r/r1r3r4-rr6r8r r9r:r;r<r=r?r@rArBrCrDrEr>r7rFrGrHrIrJrPAddEnableGcfsFlagrKrLrOrMrNrQrRrSrVrTrUrWrXrYrZr[r\r]r^r_r`s rrfzUpdateBeta.Args  &M  / / / >E!445FG !!"7tL ""#8M  !'+ 88? ))*;EJ"556HI ##$:; 4DI &&d4 ''(>? &&d4 (()?@ %%d4 !!%%8 ""5) eE2 T2 E$G E$G & ''. E6 U# %%e4@ **5? (( TT3 %%e44P !!%( T: !!%( ..u5 !!%(  Te5 ##E* & 5! **5100? .t< +, +, ../@A (():;rc ^tj|tjd1id|jd|j d|j d|jd|jd|jd|jd|jd |jd |jd |jd |jd |j d|j"d|j$d|j&d|j(d|j*d|j,d|j.d|j0d|j2d|j4d|j6d|j8d|j:d|j<d|j>d|j@d|jBd|jDd |jFd!|jHd"|jJd#|jLd$|jNd%|jPd&|jRd'|jTd(|jVd)|jXd*|jZd+|j\d,|j^d-ta|d.rtcjd|jfndd/|jhd0|jj}|S2Nrhrirjrkrmrnrorprqrtrrrsrlrurvrwrxryrzr{r enable_gcfsr|r}rrrrrr~rrrrrrrrrrrrrrrrrrr6r rrrrrirjrkrmrnrorprqrtrrrsrlrurvrwrxryrzr{rrrr}rrrrrrrrrrrrrrrrrrrrrrrrrrrropss rrz%UpdateBeta.ParseUpdateNodePoolOptionsN? &&t,  + +2 %%2 002  222  22 2  .. 2  .. 2 ,,2 ,,2 ,,2 !% < <2 002 %)$D$D2 **2 002 !% < <2  !% < #'"@"@?2 @ $::A2 B 22C2 D,,E2 F 22G2 H#88I2 J%)$D$DK2 L((M2 N$(#B#BO2 P..Q2 R??S2 T&&U2 V..W2 X/3.X.XY2 \ 4 %__T^^4 _2 `$(#B#Ba2 b*.)N)Nc2 Cf JrNrrrrrrfrrrrrr1B<B|dtj@|tjB|tjD|tjF|dd tjH|tjJ|tjL|tjN||jd }tjP|dtjR|tjT|tjV|tjX|yrrr`s rrfzUpdateAlpha.Argsrrc ^tj|tjd1id|jd|j d|j d|jd|jd|jd|jd|jd |jd |jd |jd |jd |j d|j"d|j$d|j&d|j(d|j*d|j,d|j.d|j0d|j2d|j4d|j6d|j8d|j:d|j<d|j>d|j@d|jBd|jDd |jFd!|jHd"|jJd#|jLd$|jNd%|jPd&|jRd'|jTd(|jVd)|jXd*|jZd+|j\d,|j^d-ta|d.rtcjd|jfndd/|jhd0|jj}|Srrrs rrz&UpdateAlpha.ParseUpdateNodePoolOptionsrrNrrrrrrrrrN)#r __future__rrrapitools.base.pyrrgooglecloudsdk.api_lib.computer googlecloudsdk.api_lib.containerrr googlecloudsdk.callioper $googlecloudsdk.command_lib.containerr googlecloudsdk.corer googlecloudsdk.core.consoler DETAILED_HELPrDefaultUniverseOnly ReleaseTracks ReleaseTrackGAUniverseCompatible UpdateCommandrBETArALPHAr detailed_helprrrrs &'>081(.6#2   -&D%%(()~T  ~*~BD%%**+~~,~BD%%++,~&~-~B%r