ndZddlmZddlmZddlmZddlmZddlmZddl m Z dZ d Z d Z d Zd Zy )zOUtils for cluster maintenance window and maintenance exclusion window commands.)absolute_import)division)unicode_literals)util) exceptions)flagsc~tj|ds|S|jj}|j(t j |j|_|jj4t j |}|j|j_|j|jj_ td||S)zUpdates the cluster.control_plane_encryption if --control-plane-kms-key flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request control_plane_kms_keycontrolPlaneEncryption) rFlagIsExplicitlySetcalliope_command ReleaseTrackclusterrGetMessagesModuleClusterr ControlPlaneEncryptionr kmsKey_AddFieldToUpdateMaskrefargsrequest release_trackmessagess :lib/googlecloudsdk/command_lib/edge_cloud/container/kms.py UpdateKmsKeyrs  " "4)@ A N''446- __,,];CCEGO __++3%%m4H-5-L-L-NGOO*262L2L'//((/0': .cR~tj|ds|S|jstjdd|j B|j j}tj|j|_d|j _ td||S)aClears cluster.control_plane_encryption in the request if --use-google-managed-key flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request use_google_managed_keyz--no-use-google-managed-keyThe flag is not supportedNr ) rr rrBadArgumentExceptionrr rrrrr rrrrrs rUseGoogleManagedKeyr#9s  " "4)A B N $ $  ) )%'B  __))668M,,];CCEGO+/'//(0': .rc~tj|ds|S|jj}|j(t j |j|_|jj4t j |}|j|j_|j|jj_ td||S)zUpdates the cluster.zone_storage_encryption if --zone-storage-kms-key flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request zone_storage_kms_keyzoneStorageEncryption) rr r rrrrrr&ZoneStorageEncryptionr%rrrs rUpdateZoneKmsKeyr(Ys  " "4)? @ N''446- __,,];CCEGO __**2%%m4H,4,J,J,LGOO)151J1J'//''./9 .rcR~tj|ds|S|jstjdd|j B|j j}tj|j|_d|j _ td||S)aClears cluster.zone_storage_encryption in the request if --use-google-managed-zone-key flag is specified. Args: ref: reference to the cluster object. args: command line arguments. request: API request to be issued Returns: modified request use_google_managed_zone_keyz --no-use-google-managed-zone-keyr Nr&) rr r*rr!rr rrrrr&rr"s rUseGoogleManagedZoneKeyr+ys  " "4)F G N ) )  ) )*,G  __))668M,,];CCEGO*.'//'/9 .rcz|js ||_|S||jvr|jdz|z|_|S)N,) updateMask)fieldrs rrrsC   G N '$$$ ++c1E9G .rN)__doc__ __future__rrr+googlecloudsdk.api_lib.edge_cloud.containerrgooglecloudsdk.calliopergooglecloudsdk.command_lib.runrrr#r(r+rrrr6s:V&'<.0@@@@r