dZddlmZddlmZddlmZddlmZdZdZ d d Z d d Z dd Z dd Z y)zBigtable clusters API helper.)absolute_import)division)unicode_literals)utilctj}tj}|j|j }|j j |y)z]Delete a cluster. Args: cluster_ref: A resource reference to the cluster to delete. )nameN)rGetAdminClientGetAdminMessages3BigtableadminProjectsInstancesClustersDeleteRequest RelativeNameprojects_instances_clustersDelete) cluster_refclientmsgsmsgs /lib/googlecloudsdk/api_lib/bigtable/clusters.pyrrsV    &   $ @@  # # % A '#$$++C0ctj}tj}|j||j |j j }|jj|S)zCreate a cluster. Args: cluster_ref: A resource reference to the cluster to create. cluster: A Cluster msg object to create. Returns: Long running operation. )cluster clusterIdparent) rr r 3BigtableadminProjectsInstancesClustersCreateRequestNameParentr r Create)rrrrrs rrr%ss    &   $ @@  "    ! . . 0 A 2#  + + 2 23 77rNctj}|j||}|j||}|j ||S)z'Build a ClusterAutoscalingConfig field.) minServeNodes maxServeNodes)cpuUtilizationPercentstorageUtilizationGibPerNode)autoscalingLimitsautoscalingTargets)rr AutoscalingLimitsAutoscalingTargetsClusterAutoscalingConfig) min_nodes max_nodes cpu_targetstorage_targetrlimitstargetss rBuildClusterAutoscalingConfigr-8sf    $  ! !Y " 8&  # # $ 3'  & &7 ' <    $  <##+3  5  66rc|j||}||||t|||||_|j|||S)z$Build a PartialUpdateClusterRequest.)r serveNodes)r1r2r3r4)rr updateMask)Clusterr5 clusterConfigABigtableadminProjectsInstancesClustersPartialUpdateClusterRequest) rrnodesr1r2r3r4 update_maskrs r BuildPartialUpdateClusterRequestr>Ush LLduL 5'!_%@( ,.''5#= ?G  O OD[ P ::rc @tj}tj}|rI|||| td|jj t ||j|dSg} || jd|| jd|| jd|| jd|| jdd j| } |jj t ||j|||||| S) aPartially update a cluster. Args: cluster_ref: A resource reference to the cluster to update. nodes: int, the number of nodes in this cluster. autoscaling_min: int, the minimum number of nodes for autoscaling. autoscaling_max: int, the maximum number of nodes for autoscaling. autoscaling_cpu_target: int, the target CPU utilization percent for autoscaling. autoscaling_storage_target: int, the target storage utilization gibibytes per node for autoscaling. disable_autoscaling: bool, True means disable autoscaling if it is currently enabled. False means change nothing whether it is currently enabled or not. Returns: Long running operation. zEautoscaling arguments cannot be set together with disable_autoscalingz5serve_nodes,cluster_config.cluster_autoscaling_config)rrr<r= serve_nodeszLcluster_config.cluster_autoscaling_config.autoscaling_limits.min_serve_nodeszLcluster_config.cluster_autoscaling_config.autoscaling_limits.max_serve_nodeszUcluster_config.cluster_autoscaling_config.autoscaling_targets.cpu_utilization_percentz^cluster_config.cluster_autoscaling_config.autoscaling_targets.storage_utilization_gib_per_node,)rrr<r1r2r3r4r=) rr r ValueErrorr PartialUpdateClusterr>r appendjoin) rr<r1r2r3r4disable_autoscalingrrchanged_fieldsr=s r PartialUpdaterHmsS2    &   $#'B*". - ..  - - B B )))+O    . -( V V'_ +h(+  + + @ @&'')))!7%?!# $ $r)NNNN)NNNNNNN)NNNNNF) __doc__ __future__rrrgooglecloudsdk.api_lib.bigtablerrrr-r5r>rHrrrMs|$&'0 18&-1,0-115 <(,'+.226 6+/+/5959<@@D15:2"&"&)--1&+ J$r