}dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl mZ ddl mZdd lmZdd lmZdd lZ dd Zd Zy )z=Create requests for rolling-action restart/recreate commands.)absolute_import)division)unicode_literals)managed_instance_groups_utils)flags)scope)update_instances_utils) exceptions)timesNctj}tjj}t j |}|j||||}|jdvr(tdj|j|jjjj} tj dd|j"|j} t%j&||} | j(xs'|jj+| j,g} t/j0t3j4t2j6} t9| D]\}}d|| fz|_|jj|| || }t=|d r|j>|_ t=|d r1tjB|jD|j}||_#tI|| |||jjKd || }|jd k(rX|jLjN}|jjQ|jS||jT|jV}nW|jLjX}|jj[|jS||jT|j\}|d|fS)aCreate request helper for compute instance-groups managed rolling-action. Args: args: argparse namespace client: The compute client resources: The compute resources minimal_action: MinimalActionValueValuesEnum value max_surge: InstanceGroupManagerUpdatePolicy.maxSurge value Returns: ComputeInstanceGroupManagersPatchRequest or ComputeRegionInstanceGroupManagersPatchRequest instance Raises: ValueError: if instance group manager collection path is unknown ) default_scope scope_lister)compute.instanceGroupManagersz#compute.regionInstanceGroupManagerszUnknown reference type {0}z--max-unavailablezmax-unavailable)instanceTemplatez%d/%s)maxSurgemaxUnavailable minimalActiontype min_readyreplacement_methodN)r updatePolicyversionsr)instanceGroupManagerinstanceGroupManagerResourceprojectzone)rrrregionPatch)/instance_groups_flags%MULTISCOPE_INSTANCE_GROUP_MANAGER_ARG compute_scope ScopeEnumZONErGetDefaultScopeListerResolveAsResource Collection ValueErrorformatmessages InstanceGroupManagerUpdatePolicyTypeValueValuesEnum PROACTIVEr ParseFixedOrPercentmax_unavailablerGetInstanceGroupManagerOrThrowrInstanceGroupManagerVersionrsix text_typer NowUTC enumeratenamehasattrr minReadySecParseReplacementMethodrreplacementMethod$ValidateAndFixUpdaterAgainstStatefulInstanceGroupManagerapitools_clientinstanceGroupManagers(ComputeInstanceGroupManagersPatchRequestNamerrregionInstanceGroupManagers.ComputeRegionInstanceGroupManagersPatchRequestr)argsclient resourcesminimal_action max_surge resource_argr rigm_refupdate_policy_typer.igm_inforcurrent_time_striversion update_policyr igm_resourceservicerequests Plib/googlecloudsdk/command_lib/compute/instance_groups/managed/rolling_action.py CreateRequestrTs*'LL,))..-,,V4,  * * I] + O' " 1889K9K9MN OOHH++II*>>,d.B.B oo/+ I I v( oo11#4426$(]]599UYY#78h'jaa!122GL(//BB$"  C-  T; $M T'(/FF 2&8M#&}hM55-(6L, <<$$::GooFF$\\^%1 \\ GG $$@@GooLL$\\^%1~~ MG 7G $$c^tj|syt|drb|jjj j }|j||_n$|j|k7rtjd|j"|jjd|_ yd}|jj|jjdk7rd}|jj|jjdk7rd}|stjdy) aValidates and fixes update policy for patching stateful IGM. Updating stateful IGMs requires maxSurge=0 and replacementMethod=RECREATE. If the field has the value set, it is validated. If the field has the value not set, it is being set. Args: update_policy: Update policy to be validated igm_info: Full resource of IGM being validated client: The compute API client args: argparse namespace used to select used version NrzUFor performing this action on a stateful IGMs --replacement-method has to be RECREATEr)fixedTFzEFor performing this action on a stateful IGMs --max-surge has to be 0)r IsStatefulr7r)r* ReplacementMethodValueValuesEnumRECREATEr:r ErrorrFixedOrPercentrWpercent)rOrKrDrCrecreatemax_surge_is_zeros rSr;r;ss ' 1 1( ;  T'(88 ) )(( &&.(0m%  ( (H 4    4 55##__;;!;DM##/    % % *!%%1    ' '1 ,!     $ %% rU)N)__doc__ __future__rrrgooglecloudsdk.api_lib.computer"googlecloudsdk.command_lib.computerrr!2googlecloudsdk.command_lib.compute.instance_groupsr:googlecloudsdk.command_lib.compute.managed_instance_groupsr googlecloudsdk.corer googlecloudsdk.core.utilr r1rTr;rUrSris=D&'H4E]]** ! Q%h&%rU