dZddlmZddlmZddlmZddlmZddlmZGddZGd d ejZ y ) z3Functions to add flags in cluster upgrade commands.)absolute_import)division)unicode_literals) arg_parsers)parser_argumentsceZdZdZdej fdZedZdZ dZ ddZ dej fd Z dej fd Z dej fd Zdej fd Zdd Zdej fdZdej fdZy)ClusterUpgradeFlagsz1Add flags to the cluster upgrade command surface.parserc||_yN_parser)selfr s Flib/googlecloudsdk/command_lib/container/fleet/clusterupgrade/flags.py__init__zClusterUpgradeFlags.__init__s DLc|jSr r rs rr zClusterUpgradeFlags.parsers <<rcB|jjddddy)z,Adds the --show-linked-cluster-upgrade flag.z--show-linked-cluster-upgrade store_trueNa^ Shows the cluster upgrade feature information for the current fleet as well as information for all other fleets linked in the same rollout sequence (provided that the caller has permission to view the upstream and downstream fleets). This displays cluster upgrade information for fleets in the current fleet's rollout sequence in order of furthest upstream to downstream. To view the cluster upgrade feature information for the rollout sequence containing the current fleet, run: $ {command} --show-linked-cluster-upgrade actiondefaulthelp)r add_argumentrs rAddShowLinkedClusterUpgradez/ClusterUpgradeFlags.AddShowLinkedClusterUpgrade#s'KK'  rcd|jjdtjdy)z(Adds the --default-upgrade-soaking flag.z--default-upgrade-soakinga Configures the default soaking duration for each upgrade propagating through the current fleet to become "COMPLETE". Soaking begins after all clusters in the fleet are on the target version, or after 30 days if all cluster upgrades are not complete. Once an upgrade state becomes "COMPLETE", it will automatically be propagated to the downstream fleet. Max is 30 days. To configure Rollout Sequencing for a fleet, this attribute must be set. To do this while specifying a default soaking duration of 7 days, run: $ {command} --default-upgrade-soaking=7d typerN)r rrDurationrs rAddDefaultUpgradeSoakingFlagz0ClusterUpgradeFlags.AddDefaultUpgradeSoakingFlag8s-KK#  ! ! #  rc|r=|jj}|j||j|y|j|jyr )r add_mutually_exclusive_group%_AddRemoveUpgradeSoakingOverridesFlag_AddUpgradeSoakingOverrideFlagsrwith_destructivegroups rAddUpgradeSoakingOverrideFlagsz2ClusterUpgradeFlags.AddUpgradeSoakingOverrideFlagsMsEkk668e 007 **51 **4;;7rr(c.|jddddy)zvAdds the --remove-upgrade-soaking-overrides flag. Args: group: The group that should contain the flag. z"--remove-upgrade-soaking-overridesrNa Removes soaking time overrides for all upgrades propagating through the current fleet. Consequently, all upgrades will follow the soak time configured by `--default-upgrade-soaking` until new overrides are configured with `--add_upgrade_soaking_override` and `--upgrade_selector`. To remove all configured soaking time overrides, run: $ {command} --remove-upgrade-soaking-overrides rrrr(s rr$z9ClusterUpgradeFlags._AddRemoveUpgradeSoakingOverridesFlagUs% ,  rcl|jd}|j||j|y)zAdds upgrade soaking override flags. Args: group: The group that should contain the upgrade soaking override flags. a4 Upgrade soaking override. Defines a specific soaking time override for a particular upgrade propagating through the current fleet that supercedes the default soaking duration configured by `--default-upgrade-soaking`. To set an upgrade soaking override of 12 hours for the upgrade with name, `k8s_control_plane`, and version, `1.23.1-gke.1000`, run: $ {command} --add-upgrade-soaking-override=12h --upgrade-selector=name="k8s_control_plane",version="1.23.1-gke.1000" )rN) add_group!_AddAddUpgradeSoakingOverrideFlag_AddUpgradeSelectorFlagr,s rr%z3ClusterUpgradeFlags._AddUpgradeSoakingOverrideFlagsns7 OO " O E **51  'rcR|jdtjddy)zrAdds the --add-upgrade-soaking-override flag. Args: group: The group that should contain the flag. z--add-upgrade-soaking-overrideTa7 Overrides the soaking time for a particular upgrade name and version propagating through the current fleet. Set soaking to 0 days to bypass soaking and fast-forward the upgrade to the downstream fleet. See `$ gcloud topic datetimes` for information on duration formats. rrequiredrN)rrr r,s rr/z5ClusterUpgradeFlags._AddAddUpgradeSoakingOverrideFlags. (  ! ! #  rc>|jdtddy)zfAdds the --ugprade-selector flag. Args: group: The group that should contain the flag. z--upgrade-selectorTz Name and version of the upgrade to be overridden where version is a full GKE version. Currently, name can be either `k8s_control_plane` or `k8s_node`. r2N)rUpgradeSelectorr,s rr0z+ClusterUpgradeFlags._AddUpgradeSelectorFlags(     rc|r=|jj}|j||j|y|j|jy)zAdds upstream fleet flags.N)r r#_AddUpstreamFleetFlag_AddResetUpstreamFleetFlagr&s rAddUpstreamFleetFlagsz)ClusterUpgradeFlags.AddUpstreamFleetFlagssEkk668e   ' %%e,   -rc4|jdtdy)zdAdds the --upstream-fleet flag. Args: group: The group that should contain the flag. z--upstream-fleeta The upstream fleet. GKE will finish upgrades on the upstream fleet before applying the same upgrades to the current fleet. To configure the upstream fleet, run: $ {command} --upstream-fleet={upstream_fleet} rN)rstrr,s rr7z)ClusterUpgradeFlags._AddUpstreamFleetFlags"    rc.|jddddy)zjAdds the --reset-upstream-fleet flag. Args: group: The group that should contain the flag. z--reset-upstream-fleetrNa Clears the relationship between the current fleet and its upstream fleet in the rollout sequence. To remove the link between the current fleet and its upstream fleet, run: $ {command} --reset-upstream-fleet rr+r,s rr8z.ClusterUpgradeFlags._AddResetUpstreamFleetFlags%    rN)F)__name__ __module__ __qualname____doc__rArgumentInterceptorrpropertyr rr!r)r$r%r/r0r9r7r8rrr r s9-AA  **8#772(#77(2#77*#77(.)9)M)M*#77rr c"eZdZdZfdZxZS)r5zGExtends the ArgDict type to properly parse --upgrade-selector argument.cLtt| ttdddgdy)N)nameversionrFrG)spec required_keys max_length)superr5rr;)r __class__s rrzUpgradeSelector.__init__s* /4)c *y)*r)r=r>r?r@r __classcell__)rMs@rr5r5sOrr5N) r@ __future__rrrgooglecloudsdk.callioperrr ArgDictr5rCrrrRs8:&'/4JJZk))r