(tdZddlmZddlmZddlmZddlmZddlmZddl m Z dZ d Z d Z d Zd Zd Zy)zOUtils for cluster maintenance window and maintenance exclusion window commands.)absolute_import)division)unicode_literals)util) exceptions)flagsc |jj|j|j_|jjjg|jj_|jjjj |j |j |j|j|j|S)a!Returns an update request with a new maintenance exclusion window with id, start time, and end time specified from args. Args: req: API request to be issued. messages: message module of edgecontainer cluster. args: command line arguments. Returns: modified request ) startTimeendTime)idwindow) clustermaintenancePolicyMaintenancePolicymaintenanceExclusionsappendMaintenanceExclusionWindowadd_maintenance_exclusion_name TimeWindowadd_maintenance_exclusion_startadd_maintenance_exclusion_end)reqmessagesargss Ilib/googlecloudsdk/command_lib/edge_cloud/container/maintenance_policy.py"RequestWithNewMaintenanceExclusionrs [[""*$,$>$>$@CKK![[""88@:   - -  ) )+D   + +  ) ))C  ,rc>~tj|ds|S|jB|jj }t j |j|_|jjtd||St|jjjD]"\}}|j|jk(s |}ntjdd|jjj|=td||S)aRemoves the cluster.maintenance_policy.maintenance_exclusion_window if --remove-maintenance-exclusion-window flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request #remove_maintenance_exclusion_windowrz%--remove-maintenance-exclusion-windowz@Cannot remove a maintenance exclusion window that doesn't exist.)rr rr"r#rr$r%rr' enumeraterr r/rr&)r(rr)r*idxr+is r RemoveMaintenanceExclusionWindowr3zs  " "4)N O N __))668M,,];CCEGO __&&.-w7 N oo''==hc3 vv999 a   ) )/J  oo''==a@+W5 .rc~tj|ds|S|jstjdd|j B|j j}tj|j|_|j jr6|j jjrtjddd|j _ td||S)aClears cluster.maintenance_policy.window in the request if --clear-maintenance-window flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request clear_maintenance_windowz--no-clear-maintenance-windowzThe flag is not supportedNz--clear-maintenance-windowzFCannot clear a maintenance window if there are maintenance exclusions.r)rr r5rr&rr"r#rr$r%rrr')r(rr)r*s rClearMaintenanceWindowr6s  " "4)C D N & &  ) )')D  __))668M,,];CCEGO __&&((>>  + + &   '+'//#+W5 .rcz|js ||_|S||jvr|jdz|z|_|S)N,) updateMask)fieldr)s rr'r'sC   G N '$$$ ++c1E9G .rN)__doc__ __future__rrr+googlecloudsdk.api_lib.edge_cloud.containerrgooglecloudsdk.calliopergooglecloudsdk.command_lib.runrrr,r!r3r6r'rrrAs>V&'<.0 8(V4(V"Jr