+dZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZGddeZGddeZdZdZdZ e jBe jDe jFjHGdde jJZ&ddde&_'e jDe jFjPGdde&Z)e jDe jFjTGd d!e&Z+y")#zUpgrade cluster command.)absolute_import)division)unicode_literals) exceptions) api_adapter)util)base)container_command_util)flags)log) properties) console_attr) console_io) iso_duration)times)SemVerc eZdZdZdZdZdZdZy)UpgradeHelpTextz%Upgrade available help text messages.z8 * - There is an upgrade available for your cluster(s). zZ ** - The current version of your cluster(s) will soon be out of support, please upgrade. zN *** - The current version of your cluster(s) is unsupported, please upgrade. zc To upgrade nodes to the latest available version, run $ gcloud container clusters upgrade {name}N)__name__ __module__ __qualname____doc__UPGRADE_AVAILABLESUPPORT_ENDING UNSUPPORTEDUPGRADE_COMMAND)lib/surface/container/clusters/upgrade.pyrr&s"-.+0/rrc&eZdZdZdZdZdZdZdZy)VersionVerifierzACompares the cluster and master versions for upgrade availablity.rc||k(r |jSt|}t|}|j|\}}}|dk7s|dkDr |jS|dkDr |jS|j S)z=Compares the cluster and master versions and returns an enum.rr#r") UP_TO_DATErDistancerrr)selfcurrent_master_versioncurrent_cluster_versionmaster_versioncluster_versionmajorminor_s rComparezVersionVerifier.Compare@sy!88 __23N45O$--o>OE5! zUQY        # ##rN) rrrrr&rrrr0rrrr!r!9sI*.+ $rr!c Ttj|j|j|j |j|j |j |j}|jr8tjtj|j|_|S)z;Parses the flags provided with the cluster upgrade command.)version update_master update_nodes node_pool image_typeimage image_project)seconds) rUpdateClusterOptionsr,masterr5r6r7r8control_plane_soak_durationrFormatDurationForJsonrDuration)argsoptss rParseUpgradeOptionsBaserAOs  ) )""KK O JJ&& $ %%','B'Bd&F&FG(D$ +rc|jdddtj|d|jdd|jd d d |jd tdddtj|tj |dtj |dtj|dtj|dy)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. nameNAMEz#The name of the cluster to upgrade.)metavarhelpaThe GKE release version to which to upgrade the cluster's node pools or master. If desired cluster version is omitted, *node pool* upgrades default to the current *master* version and *master* upgrades default to the default cluster version, which can be found in the server config. You can find the list of allowed versions for upgrades by running: $ gcloud container get-server-config )rFz --node-poolzThe node pool to upgrade.z--masterz\Upgrade the cluster's master. Node pools cannot be upgraded at the same time as the master. store_true)rFactionz --timeoutiTz;Timeout (seconds) for waiting on the operation to complete.)typedefaulthiddenrFzcluster/node pool)rKN) add_argumentr AddClusterVersionFlagint AddAsyncFlagAddImageTypeFlag AddImageFlagAddImageProjectFlagAddControlPlaneSoakDurationFlagparsers r_ArgsrVas  f#HJ    m*EF &     H J V!45VD)F40''t| dd |jA|} |jC|| } |jHs|jK| d j#|jL|jN  |j|}|jPD]} tjR|  tjV|yy#tjtjtjf$rD}tj dj#t%j&|d}Yd}~d}~wwxYw#tjtjtjf$rD}tj dj#t%j&|d} Yd}~Wd}~wwxYw#tjD$r)}tj|tjFd}~wwxYw#tjtjtjf$rP}tj tjTj#t%j&|Yd}~d}~wwxYw)zThis 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: Some value that we want to have printed later. r location_getT)requiredzbProblem loading details of cluster to upgrade: {} You can still attempt to upgrade the cluster. NzRProblem loading server config: {} You can still attempt to upgrade the cluster. z--control-plane-soak-durationzA--control-plane-soak-duration can only be specified with --master)rC server_confr[r;r<node_pool_name new_versionnew_image_type)messagethrow_if_unattended cancel_on_noz Upgrading {0}) timeout_s),context ParseClusterrCr VALUEScoreprojectGet GetClusterr HttpExceptionapitools_exceptionsHttpForbiddenErrorrErrorr warningformatrSafeTextGetServerConfigr<r;InvalidArgumentExceptionr ClusterUpgradeMessager5r,r6r]rPromptContinuerg UpdateCluster HttpErrorHTTP_ERROR_FORMATasync_WaitForOperation clusterIdtimeout nodePoolsCheckForCgroupModeV1CGROUPV1_CHECKING_FAILURE_MSGUpdatedResource)r(r?adapterrilocation cluster_ref project_idr[errorrkupgrade_messageoptionsop_refr5s rRunz Upgrade.Runs&ll=)G<</LD!H&&tyy(;K""''//33T3BJ"";/g++JAk ''  / / ) M OO-BB YY{{$($D$D~~(((O {{(TN&&t,GD$$[':f ;;    !6!6 7LL"  $$[1 **I  # #I .+ +&# Y  $ $&9&L&L JJ  kkEFLf#,,U3G56g   $ $&9&L&L JJ  kkEFLf#,,U3G56k @  ( (D  $ $UD,B,B CCD&&(;(N(Njj   . . 5 5%%e,    sa8H J$K;:7L: 2I?;9I::I?2K849K33K8;L7$L22L7:2N<,AN77N<N)rrrr staticmethodrdrgrrrrr_r_s%I)O'rr_a Upgrades the Kubernetes version of an existing container cluster. This command upgrades the Kubernetes version of the *node pools* or *master* of a cluster. Note that the Kubernetes version of the cluster's *master* is also periodically upgraded automatically as new releases are available. If desired cluster version is omitted, *node pool* upgrades default to the current *master* version and *master* upgrades default to the default cluster version, which can be found in the server config. *During node pool upgrades, nodes will be deleted and recreated.* While persistent Kubernetes resources, such as Pods backed by replication controllers, will be rescheduled onto new nodes, a small cluster may experience a few minutes where there are insufficient nodes available to run all of the scheduled Kubernetes resources. *Please ensure that any data you wish to keep is stored on a persistent* *disk before upgrading the cluster.* Ephemeral Kubernetes resources--in particular, Pods without replication controllers--will be lost, while persistent Kubernetes resources will get rescheduled. a Upgrade the node pool `pool-1` of `sample-cluster` to the Kubernetes version of the cluster's master. $ {command} sample-cluster --node-pool=pool-1 Upgrade the node pool `pool-1` of `sample-cluster` to Kubernetes version 1.14.7-gke.14: $ {command} sample-cluster --node-pool=pool-1 --cluster-version="1.14.7-gke.14" Upgrade the master of `sample-cluster` to the default cluster version: $ {command} sample-cluster --master ) DESCRIPTIONEXAMPLESc eZdZdZedZy) UpgradeBetar`ct|yrbrcrTs rrdzUpgradeBeta.Args,rerN)rrrrrrdrrrrr(sHrrc&eZdZdZedZdZy) UpgradeAlphar`cDt|tj|yrb)rVr !AddSecurityProfileForUpgradeFlagsrTs rrdzUpgradeAlpha.Args5s &M ++F3rc`t|}|j|_|j|_|Srb)rAsecurity_profilesecurity_profile_runtime_rules)r(r?opss rrgz UpgradeAlpha.ParseUpgradeOptions:s- !$ 'C00C)-)L)LC& JrN)rrrrrrdrgrrrrr1sH44rrN),r __future__rrrapitools.base.pyrr{ googlecloudsdk.api_lib.containerrrgooglecloudsdk.callioper $googlecloudsdk.command_lib.containerr r googlecloudsdk.corer r googlecloudsdk.core.consolerrgooglecloudsdk.core.utilrrgooglecloudsdk.core.util.semverrobjectrr!rArVr]DefaultUniverseOnly ReleaseTracks ReleaseTrackGACommandr_ detailed_helpBETArALPHArrrrrs,&'>81(.G6#*421*20f0&$f$,$*=Z . D%%(()Y'dllY'*Y'| 0 5)XD%%**+',D%%++, 7 - r