
                         \    S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  S rS	 rg
)z.Utils for cluster maintenance window commands.    )absolute_import)division)unicode_literals)util)
exceptions)flagsc                    A [         R                  " US5      (       d  U$ UR                  (       d  [        R                  " SS5      eUR
                  cC  UR                  R                  5       n[        R                  " U5      R                  5       Ul        UR
                  R                  b  SUR
                  l        [        SU5        U$ )aC  Clears max_unavailable_worker_nodes in the request.

If --clear-max-unavailable-worker-nodes flag is specified,
cluster.upgrade_settings.max_unavailable_worker_nodes is cleared.

Args:
  ref: reference to the cluster object.
  args: command line arguments.
  request: API request to be issued.

Returns:
  modified request
"clear_max_unavailable_worker_nodesz'--no-clear-max-unavailable-worker-nodeszThe flag is not supportedNz-upgrade_settings.max_unavailable_worker_nodes)r   FlagIsExplicitlySetr
   r   BadArgumentExceptionclustercalliope_commandReleaseTrackr   GetMessagesModuleClusterupgradeSettings_AddFieldToUpdateMask)refargsrequestrelease_tracks       Flib/googlecloudsdk/command_lib/edge_cloud/container/max_unavailable.pyClearMaxUnavailabler      s     
		"	"4)M	N	NN		0	0

)
)13N  __))668M,,];CCEGO__$$0&*GOO#5w 
.    c                     UR                   (       d  Xl         U$ XR                   ;  a  UR                   S-   U -   Ul         U$ )N,)
updateMask)fieldr   s     r   r   r   ?   s?    			N
$$$ ++c1E9G	.r   N)__doc__
__future__r   r   r   +googlecloudsdk.api_lib.edge_cloud.containerr   googlecloudsdk.callioper   googlecloudsdk.command_lib.runr   r   r    r   r   <module>r%      s'    5 &  ' < . 0#Lr   