7dZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z dd l m Z dd l m Z dd l mZdd lmZdd lmZdZdZdZdZdZdZdZdZdZdZdZdZdZy)z3Helpers for the container cluster related commands.)absolute_import)division)unicode_literalsN)util)base) admin_users)fleet) resource_args)robin)flags) resourcescBtj|}t|}|j|j |j |j j}t||||tjjk(r t|||S)zGet cluster create request message. Args: args: comand line arguments. release_track: release track of the command. Returns: message obj, cluster create request message. )cluster clusterIdparent) rGetMessagesModuleGetClusterReference3EdgecontainerProjectsLocationsClustersCreateRequestCluster clustersIdParent RelativeNamePopulateClusterMessager ReleaseTrackALPHAPopulateClusterAlphaMessage)args release_trackmessages cluster_refreqs :lib/googlecloudsdk/api_lib/edge_cloud/container/cluster.pyGetClusterCreateRequestr#!s # #M 2(#D)+DD &&    ! . . 0 E # h-d''---T* *ctj|}t|}|j}|j|_|j jdk7rtd|j z|jj|j j|_|j}|j|_ ||_ |S)zGet cluster upgrade request message. Args: args: comand line arguments. release_track: release track of the command. Returns: message obj, cluster upgrade request message. IMMEDIATELYzUnsupported --schedule value: )rrrUpgradeClusterRequestversion targetVersionscheduleupper ValueErrorScheduleValueValuesEnum4EdgecontainerProjectsLocationsClustersUpgradeRequestrnameupgradeClusterRequest)rrrr upgrade_cluster_reqr!s r"GetClusterUpgradeRequestr28s # #M 2(#D)+ 668&*ll# ]]m+ 5 E FF$$<< --    EEG#  % % '#(1# *r$c|j|j_|jg|jj_|j g|jj_tj|dr$t|j|j_ tj|dr|jj|j_g|jj_|jj!D]f\}}|jjj#}||_||_|jjjj)|htj|ds-tj|dstj|dr|j+|j_|j/|jj,_|j3|jj,j0_tj|dr9|j6|jj,j0j4_|j;|jj,j0j4_tj|drC|j<|jj,j0j4j0_tj|drC|j@|jj,j0j4j0_!tj|drD|jE|j_#|jH|jjF_%tj|drD|jM|j_'|jP|jjN_%tSjT|||tWjXt[|||tj|dr|j\|j_/tj|d r|j`|j_1tj|d rB|jje|jfji|j_5tj|d s-tj|d stj|d r|jm|j_7|jq|jjn_9tj|d r/|jt|jjnjr_;tj|d r8t|jx|jjnjr_=tj|d r/|j||jjnjr_?tj|drV|jpj|jji|jjnjr_Btj|drp|jjs|j|j_Ctj|jdz|jj_Htj|dr/|j|jjnjr_Jt|||t|||y)zFill the cluster message from command arguments. Args: req: create cluster request message. messages: message module of edgecontainer cluster. args: command line arguments. default_max_pods_per_nodelabelsmaintenance_window_recurrencemaintenance_window_startmaintenance_window_endcontrol_plane_kms_keyzone_storage_kms_keyexternal_lb_ipv4_address_poolsr(release_channelcontrol_plane_node_locationcontrol_plane_node_countcontrol_plane_machine_filter&control_plane_shared_deployment_policyoffline_reboot_ttls!control_plane_node_storage_schemaN)MClusterNetworkingr networkingcluster_ipv4_cidrclusterIpv4CidrBlocksservices_ipv4_cidrservicesIpv4CidrBlocksr FlagIsExplicitlySetintr4defaultMaxPodsPerNoder LabelsValuer5additionalPropertiesitemsAdditionalPropertykeyvalueappendMaintenancePolicymaintenancePolicyMaintenanceWindowwindowRecurringTimeWindowrecurringWindowr6 recurrence TimeWindowr7 startTimer8endTimeControlPlaneEncryptioncontrolPlaneEncryptionr9kmsKeyZoneStorageEncryptionzoneStorageEncryptionr:r SetAdminUsersr SetFleetProjectPathrr;$externalLoadBalancerIpv4AddressPoolsr(r)ReleaseChannelValueValuesEnumr<r+releaseChannel ControlPlane controlPlaneLocallocalr= nodeLocationr> nodeCountr? machineFilter%SharedDeploymentPolicyValueValuesEnumr@sharedDeploymentPolicysurvivabilityConfigSurvivabilityConfigjsondumpsrAofflineRebootTtlrCcontrolPlaneNodeStorageSchemaSetContainerRuntimeConfigEnableGoogleGroupAuthentication)r!rrrQrRvs r"rrSs$557#++262H2H1I#++.373J3J2K#++/ t%@A(+D,J,J(KCKK% tX.!))557CKK.0CKK+kk'') U    & & 9 9 ;aaeag kk--44Q7 *  &EF  " "4)C D  " "4)A B$,$>$>$@CKK!+3+E+E+GCKK!!($$&KK!!((8   'FG  , , kk##**::E KK!!((88?   'AB  ' ' kk##**::AAK   '?@  % % kk##**::AAI t%<=)1)H)H)JCKK&040J0JCKK&&- t%;<(0(F(F(HCKK%/3/H/HCKK%%, HdC0/5tSA t%EF ++KK4 tY/ $ CKK t%67!)!1!1!O!O ""$"CKK &CD  " "4)C D  " "4)G H (446CKK%-^^%5CKK"   'DE  * * kk$$1   'AB14  ' '2ckk$$.   'EF  + + kk$$2    6 .. > >99??A  kk$$;  t%9: ;; * *(0(D(D(Fckk% 4**+c1KK##4 t%HI ..KK""@Cx0!#tX6r$ctj|dr&|jg|jj_tj|dr&|j g|jj_tj|dr|j|j_ tj||tj||t||t||tj|drtj ||t#j$t&j(j*}tj|drt-|||yy)zFilled the Alpha cluster message from command arguments. Args: req: create cluster request message. args: command line arguments. cluster_ipv6_cidrservices_ipv6_cidrexternal_lb_ipv6_address_poolsenable_robin_cns,control_plane_node_system_partition_size_gibN)r rJr{rrEclusterIpv6CidrBlocksr|servicesIpv6CidrBlocksr}$externalLoadBalancerIpv6AddressPoolsr SetSystemAddonsConfig)SetExternalLoadBalancerAddressPoolsConfigEnableClusterIsolationConfigEnableRemoteBackupConfigr EnableRobinCNSInRequestrrrrr&SetControlPlaneNodeSystemPartitionSizer!rrs r"rrs t%89484J4J3KCKK0 t%9:595L5L4MCKK1 t%EF ++KK4%%dC099$DsD)3% t%78 !!#t,  # #D$5$5$;$; <(  :+3h?r$ctj|drCtj|dr-tj|dstj|dryy)zIdentify if the command is creating LCP cluster. Args: args: command line arguments. Returns: Boolean, indication of LCP cluster. r=r>r;external_lb_address_poolsTFr rJrs r" IsLCPClusterrsP &CD  # #D*D E  # #D*J K  & &t-H I  r$c2tj|dryy)zIdentify if the command is requesting an offline credential for LCP cluster. Args: args: command line arguments. Returns: Boolean, indication of requesting offline credential. offline_credentialTFrrs r"IsOfflineCredentialrs t%9:  r$ctjj|jjj j dS)zGet edgecontainer cluster resources. Args: args: command line arguments. Returns: edgecontainer cluster resources. z)edgecontainer.projects.locations.clusters) collection)r REGISTRYParseRelativeNameCONCEPTSrParserrs r"rrsD    - - mm!!#002< . r$ctj|}|jj|jj j k(r|jj yy)zValidate cluster create request message. Args: req: Create cluster request message. release_track: Release track of the command. Returns: Single string of error message. NzSInvalid Argument: REGULAR release channel does not support specification of version)rrrrgrrfREGULARr))r!rrs r"ValidateClusterCreateRequestrs[ # #M 2( kk     7 7 ? ?@ ++ # # / $ r$ctj|dr|j|j_|j j dk(r:|jjj|jj_ y|j j dk(r:|jjj|jj_ ytd|j zy)zSet container runtime config in the cluster request message. Args: req: Create cluster request message. args: Command line arguments. messages: Message module of edgecontainer cluster. container_default_runtime_classGVISORRUNCz5Unsupported --container_default_runtime_class value: N) r rJContainerRuntimeConfigrcontainerRuntimeConfigrr+&DefaultContainerRuntimeValueValuesEnumrdefaultContainerRuntimerr,rs r"rwrw%s t%FG)1)H)H)JCKK& ++113x?  ) ) P P W W kk((@  - - 3 3 5 ?  ) ) P P U U kk((@  A  0 0 1 Hr$ctj|drv|jjdk(rd|j_y|jjdk(rd|j_yt d|jzy)zSet secure cluster isolation config in the cluster request message. Args: req: Create cluster request message. args: Command line arguments. enable_cluster_isolationTRUETFALSEFz.Unsupported --enable_cluster_isolation value: N)r rJrr+renableClusterIsolationr,r!rs r"rr>s t%?@ $$**,6+/ckk( & & , , .' 9+0ckk(  :  ) ) *  Ar$ctj|drE|j|j_|j |jj_yy)zSet Google Group authentication config in the cluster request message. Args: req: Create cluster request message. args: Command line arguments. messages: Message module of edgecontainer cluster. "enable_google_group_authenticationN)r rJGoogleGroupAuthenticationConfigrgoogleGroupAuthenticationrenablers r"rxrxRsP t%IJ002KK) //KK))0 Kr$chtj|dr|j|j_yy)zSet remote backup config in the cluster request message. Args: req: Create cluster request message. args: Command line arguments. enable_remote_backupN)r rJrrenableRemoteBackuprs r"rrcs, t%;<%)%>%>CKK"=r$c|jdk(rD|jjj|jj j _y|jdk(rD|jjj|jj j _ytd|jzdz)zSet control plane node system partition size in the cluster request message. Args: req: Create cluster request message. args: Command line arguments. messages: Message module of edgecontainer cluster. di,zBUnsupported --control_plane_node_system_partition_size_gib value: z; valid values are 100 and 300.N) rrj2ControlPlaneNodeSystemPartitionSizeValueValuesEnumSYSTEM_PARTITION_GIB_SIZE100rrirk#controlPlaneNodeSystemPartitionSizeSYSTEM_PARTITION_GIB_SIZE300r,rs r"rros 66#=IIffKK""F 88C?IIffKK""F L  ; ; < + , r$) __doc__ __future__rrrrs+googlecloudsdk.api_lib.edge_cloud.containerrgooglecloudsdk.callioper/googlecloudsdk.command_lib.edge_cloud.containerrr r r googlecloudsdk.command_lib.runr googlecloudsdk.corer r#r2rrrrrrrwrrxrrr$r"rsr:&' <(GAIA0) . 6i7X@:*  .2(1" ?r$