2dZddlZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZdd l mZdd lmZdd lmZddlmZddlmZdZGdde j0ZGdde j0ZGdde j0ZdZGddeZdZdZ dZ!dZ"dZ#dZ$d Z%d!Z&d"Z'd#Z(d$Z)d%Z*d&Z+d'Z,d(e jZfd)Z.d*Z/d+Z0d;d,Z1dd:Z?y)>z0Module for wrangling bigtable command arguments.N)util)actions) arg_parsers)base) exceptions)concepts) completers) arg_utils) yaml_data) resource_args)concept_parsers)presentation_specs)textctjjjj Dcgc]}d|vrt j |c}Scc}w)N UNSPECIFIED)rGetAdminMessagesCluster NodeScalingFactorValueValuesEnumnamesr EnumNameToChoicexs 4lib/googlecloudsdk/command_lib/bigtable/arguments.pyGetValidNodeScalingFactorsr#sZ$$&..OOUUW W! a   #W  s AceZdZfdZxZS)ClusterCompleterc 2tt| dddd|y)N)bigtableadmin.projects.instances.clustersz!beta bigtable clusters list --uri collection list_command)superr__init__selfkwargs __class__s rr$zClusterCompleter.__init__-s( D*>8 __name__ __module__ __qualname__r$ __classcell__r(s@rrr+ r)rceZdZfdZxZS)InstanceCompleterc 2tt| dddd|y)N bigtableadmin.projects.instancesz"beta bigtable instances list --urirr")r#r2r$r%s rr$zInstanceCompleter.__init__7s( T+59 r)r*r/s@rr2r25r0r)r2ceZdZfdZxZS)TableCompleterc 2tt| dddd|y)N'bigtableadmin.projects.instances.tablesz)beta bigtable instances tables list --urirr")r#r6r$r%s rr$zTableCompleter.__init__As' .$(<@ r)r*r/s@rr6r6?r0r)r6ctj}|j}|jj |j }|j ds'||jjjk(rd}|S||jjjk(rtjdd|dkrtjdd|S)a5Ensure that --instance-type and --num-nodes are consistent. If --instance-type is DEVELOPMENT, then no --cluster-num-nodes can be specified. If --instance-type is PRODUCTION, then --cluster-num-nodes defaults to 3 if not specified, but can be any positive value. Args: args: an argparse namespace. Raises: exceptions.InvalidArgumentException: If --cluster-num-nodes is specified when --instance-type is DEVELOPMENT, or --cluster-num-nodes is not positive. Returns: Number of nodes or None if DEVELOPMENT instance-type. cluster_num_nodes--cluster-num-nodesz9Cannot set --cluster-num-nodes for DEVELOPMENT instances.z;Clusters of PRODUCTION instances must have at least 1 node.) rrr:InstanceTypeValueValuesEnum instance_type IsSpecified PRODUCTION DEVELOPMENTrInvalidArgumentException)argsmsgs num_nodesr@s rProcessInstanceTypeAndNodesrHIs$    $$$)--33D4F4FG-  - . 99DDDi  99EEE  / /  E  Q  / /  G  r)ceZdZdZdZdZdZdZdZdZ dZ d Z dd Z d Z dd ZdZ ddZddZdZddefdZddefdZdZddZdZdZ d dZdZdZddZy )!ArgAdderz3A class for adding Bigtable command-line arguments.c||_yN)parser)r&rMs rr$zArgAdder.__init__rs DKr)ctj|tjjj t S)N) valid_choices)r ChoiceToEnumrrrrr)r&node_scaling_factors r_GetNodeScalingFactorChoicez$ArgAdder._GetNodeScalingFactorChoiceus6  ! ! ''HH02 r)cXtjj|j|SrL)r ASYNC_FLAG AddToParserrMr&s rAddAsynczArgAdder.AddAsync|sOO , Kr)cL|jjdtdd|S)zAdd cluster argument. --clusterzID of the cluster.T completerhelprequired)rM add_argumentrrVs r AddClusterzArgAdder.AddClusters,KK" !  Kr)c ||jjdtddtjdddd|S)z Add deprecated cluster argument.rYzID of the clusterFEThe {flag_name} argument is deprecated; use --cluster-config instead.storewarnremovedaction)r[r\r]rf)rMr^rrDeprecationActionrVs rAddDeprecatedClusterzArgAdder.AddDeprecatedClustersJKK" ((   Kr)c ||jjdddttjdddd|S)z&Add deprecated cluster nodes argument.r<Number of nodes to serve.Frarbrc)r\r]typerf)rMr^intrrgrVs rAddDeprecatedClusterNodesz"ArgAdder.AddDeprecatedClusterNodessJKK ( (( !  Kr)crtjdddgdd}|j|j|S)Nz--cluster-storage-typehddssdzStorage class for the cluster.)choicesdefaulthelp_str)rChoiceArgumentrUrM)r&storage_arguments rAddClusterStoragezArgAdder.AddClusterStorages<** 1    - Kr)c~tjddtd}|j|j|S)Nz--node-scaling-factorz$Node scaling factor for the cluster.znode-scaling-factor-1x)rsrqrr)rrtrrUrM)r&node_scaling_factor_arguments rAddClusterNodeScalingFactorz$ArgAdder.AddClusterNodeScalingFactors;#'#6#67*,( $ !,,T[[9 Kr)cJ|jj|rdnddd|S)N--cluster-zonez--zone|ID of the zone where the cluster is located. Supported zones are listed at https://cloud.google.com/bigtable/docs/locations.Tr\r]rMr^)r& in_instances rAddClusterZonezArgAdder.AddClusterZones3KK'X N  Kr)c r|jjdddtjdddd|S)z%Add deprecated cluster zone argument.r{r|Frarbrc)r\r]rfrMr^rrgrVs rAddDeprecatedClusterZonez!ArgAdder.AddDeprecatedClusterZonesLKK N((  " Kr)Nc8djtj|rdndd}|rdj||g}|rdnd}t|d}|r&|rd|d <nd }t j |d <d |d <|s||d<|jj|fi||S)z'Add argument for instance ID to parser.z ID of the {}.r=instance --instance)r[r\+nargs --instancesrkINSTANCEmetavarr]) formatr Pluralizejoinr2rArgListrMr^)r& positionalr]multipleadditional_help help_textnamerEs r AddInstancezArgAdder.AddInstances && Hq!Z8I((I78i#:D*I >D W "**,V $Y !d:DKKT*T* Kr)cL|jjdtdd|S)zAdd table argument.--tablezID of the table.TrZ)rMr^r6rVs rAddTablezArgAdder.AddTables,KK   Kr)c|jj|}|jd}|jddddd|jd t j d d |jd ddd|jd}|jdt ddd}|jddd||S)z1Adds arguments for app_profile routing to parser.r]zMulti Cluster Routing Policyz --route-any store_trueTFz!Use Multi Cluster Routing policy.)rfr]rrr\z --restrict-toztCluster IDs to route to using the Multi Cluster Routing Policy. If unset, all clusters in the instance are eligible. RESTRICT_TO)rkr\rz--row-affinityNz.Use row-affinity routing for this app profile.rfrrr\zSingle Cluster Routing Policyz --route-toz;Cluster ID to route to using Single Cluster Routing policy.)r[r]r\z@Allow transactional writes with a Single Cluster Routing policy.z--transactional-writes)rMadd_mutually_exclusive_group add_groupr^rrr)r&r] routing_group any_grouproute_to_grouptransactional_write_helps rAddAppProfileRoutingzArgAdder.AddAppProfileRoutings KK<z'ArgAdder.AddIsolation..ksqyyc*002r)NaSpecify the request priority under standard provisioned node compute capabilities. Passing this option implies standard provisioned node compute, e.g. the `--standard` option. If not specified, the app profile uses standard provisioned node compute with PRIORITY_HIGH by default. Specifying request priority on an app profile that has Data Boost serverless compute enabled changes the compute option to standard and uses the specified priority, which might cause unexpected behavior for running applications.T)rkrqrrr\r]z --standardrFz|Use standard provisioned node compute option, rather than Data Boost compute option. If specified, `--priority` is required.rzData Boost Read-only Isolationz --data-boosta*Use Data Boost serverless compute option, rather than standard provisioned node compute. If specified, --data-boost-compute-billing-owner is required. Specifying Data Boost compute on an app profile that uses standard provisioned node compute may cause unexpected behavior for running applications.)rfrrr\r] HOST_PAYSzwCompute billing should be accounted towards the host Cloud project (containing the targeted Bigtable instance / table).z"--data-boost-compute-billing-ownerc"|jSrLrrs rrz'ArgAdder.AddIsolation.. qwwyr)zQSpecify the Data Boost compute billing owner, required if --data-boost is passed.)rMrrr^)r&isolation_groupstandard_isolation_grouprqdata_boost_isolation_groupcompute_billing_choicess r AddIsolationzArgAdder.AddIsolation]s kk>>@O.88  BGCG )) 2 !*&)) M *"1!:!:("++ ,   L ++, ' ',  Kr)cB|jjdd||S)z.Add argument group for display-name to parser.z--display-namezFriendly name of the instance.r}r~rs rAddInstanceDisplayNamezArgAdder.AddInstanceDisplayNames)KK - Kr)cddd}|jjddd|ddtjdd dd  |S) z&Add deprecated instance type argument.zProduction instances provide high availability and are suitable for applications in production. Production instances created with the --instance-type argument have 3 nodes if a value is not provided for --cluster-num-nodes.zDevelopment instances are low-cost instances meant for development and testing only. They do not provide high availability and no service level agreement applies.)rBrCz--instance-typerBc"|jSrLrrs rrz4ArgAdder.AddDeprecatedInstanceType..rr)zThe type of instance to create.FzzThe {flag_name} argument is deprecated. DEVELOPMENT instances are no longer offered. All instances are of type PRODUCTION.rbrc)rrrkrqr\r]rfr)r&rqs rAddDeprecatedInstanceTypez"ArgAdder.AddDeprecatedInstanceTypesd 7  ! G KK  .(( P " Kr)c|jjddtjttt |j tt t t t d ddgddtjd  |S) z5Add the cluster-config argument as repeated kv dicts.z--cluster-configappend) idzonenodesznode-scaling-factorkms-keyzautoscaling-min-nodeszautoscaling-max-nodeszautoscaling-cpu-targetzautoscaling-storage-targetrr )spec required_keys max_lengthaid=ID,zone=ZONE,[nodes=NODES],[node-scaling-factor=NODE_SCALING_FACTOR],[kms-key=KMS_KEY],[autoscaling-min-nodes=AUTOSCALING_MIN_NODES,autoscaling-max-nodes=AUTOSCALING_MAX_NODES,autoscaling-cpu-target=AUTOSCALING_CPU_TARGET,autoscaling-storage-target=AUTOSCALING_STORAGE_TARGET]a *Repeatable*. Specify cluster config as a key-value dictionary. This is the recommended argument for specifying cluster configurations. Keys can be: *id*: Required. The ID of the cluster. *zone*: Required. ID of the zone where the cluster is located. Supported zones are listed at https://cloud.google.com/bigtable/docs/locations. *nodes*: The number of nodes in the cluster. Default=1. *node-scaling-factor*: The node scaling factor for the cluster. Default=node-scaling-factor-1x. NODE_SCALING_FACTOR must be one of: node-scaling-factor-1x, node-scaling-factor-2x. *kms-key*: The Cloud KMS (Key Management Service) cryptokey that will be used to protect the cluster. *autoscaling-min-nodes*: The minimum number of nodes for autoscaling. *autoscaling-max-nodes*: The maximum number of nodes for autoscaling. *autoscaling-cpu-target*: The target CPU utilization percentage for autoscaling. Accepted values are from 10 to 80. *autoscaling-storage-target*: The target storage utilization gibibytes per node for autoscaling. Accepted values are from 2560 to 5120 for SSD clusters and 8192 to 16384 for HDD clusters. If this argument is specified, the deprecated arguments for configuring a single cluster will be ignored, including *--cluster*, *--cluster-zone*, *--cluster-num-nodes*. See *EXAMPLES* section. )rfrkrr\) rMr^rArgDictstrrlrRtextwrapdedentrVs rAddClusterConfigzArgAdder.AddClusterConfigsKK  '+'G'G),),*-.1  .  E__  78t Kr)c|jj|}|jd}|jdd||td|r|jddd d d d |jd d }|jddd|td|jddd|td|jddd|td|jdddd td|S)Add scaling related arguments.rzManual Scalingz --num-nodesrj NUM_NODES)r\rrr]rkrz--disable-autoscalingz~Set this flag and --num-nodes to disable autoscaling. If autoscaling is currently not enabled, setting this flag does nothing.rF)r\rfrrr]hidden Autoscaling)rz--autoscaling-min-nodesz,The minimum number of nodes for autoscaling.NAUTOSCALING_MIN_NODESz--autoscaling-max-nodesz,The maximum number of nodes for autoscaling.AUTOSCALING_MAX_NODESz--autoscaling-cpu-targetzYThe target CPU utilization percentage for autoscaling. Accepted values are from 10 to 80.AUTOSCALING_CPU_TARGETz--autoscaling-storage-targetzThe target storage utilization gibibytes per node for autoscaling. Accepted values are from 2560 to 5120 for SSD clusters and 8192 to 16384 for HDD clusters.AUTOSCALING_STORAGE_TARGET)rMrrr^rl) r&r]num_nodes_requirednum_nodes_defaultadd_disable_autoscaling&require_all_essential_autoscaling_args scaling_groupmanual_scaling_groupautoscaling_groups rAddScalingArgszArgAdder.AddScalingArgssAKK<=` or namespace format `=`. Example: `123/environment=production,123/costCenter=marketing` T)rr\r]rkrfr)rMr^rrrr UpdateActionrs rAddTagszArgAdder.AddTagstsTKK __    "'' Kr)F)TTFN)T)FFNFF)r+r,r-__doc__r$rRrWr_rhrmrvryrrrrrrrboolrrrrrrrrrrr"r)rrJrJos;&& ,MQ0.`4 D Tl!F<@#-2 IV  r)rJc0tjddS)Nrz%Bigtable instance for the {resource}.rrr ResourceParameterAttributeConfigr"r)rInstanceAttributeConfigrs  2 2 !H r)c0tjddS)Ntablez"Bigtable table for the {resource}.rrr"r)rTableAttributeConfigrs  2 2 B r)c0tjddS)Nclusterz$Bigtable cluster for the {resource}.rrr"r)rClusterAttributeConfigrs  2 2  F r)c0tjddS)N app profilez0Bigtable application profile for the {resource}.rrr"r)rAppProfileAttributeConfigrs  2 2 B r)c0tjddS)Nbackupz#Bigtable backup for the {resource}.rrr"r)rBackupAttributeConfigrs  2 2 D r)c.tjdS)Nr)rrr"r)rKmsKeyAttributeConfigr s  2 2 BBr)c0tjddS)N kms-keyringz%The KMS keyring id of the {resource}.rrr"r)rKmsKeyringAttributeConfigr s  2 2 $K r)c0tjddS)N kms-locationz&The Cloud location for the {resource}.rrr"r)rKmsLocationAttributeConfigrs  2 2 %M r)c0tjddS)N kms-projectz(The Cloud project id for the {resource}.rrr"r)rKmsProjectAttributeConfigrs  2 2 $N r)cbtjddttjdS)z:Return the resource specification for a Bigtable instance.r4rF resource_name instancesId projectsIddisable_auto_completersr ResourceSpecr DEFAULT_PROJECT_ATTRIBUTE_CONFIGr"r)rGetInstanceResourceSpecrs-   ()+::#  r)cttjddtttjdS)z7Return the resource specification for a Bigtable table.r8rF)rtablesIdrrr)rrrrrr"r)rGetTableResourceSpecrs3   /#%)+::#  r)cttjddtttjdS)z9Return the resource specification for a Bigtable cluster.rrF)r clustersIdrrr)rrrrrr"r)rGetClusterResourceSpecr"s3   1'))+::#  r)cbtjddttjdS)z=Return the resource specification for a Bigtable app profile.z,bigtableadmin.projects.instances.appProfilesrFrrr"r)rGetAppProfileResourceSpecr$s-   4!)+::#  r)cbtjddttjdS)z>Return the resource specification for a Bigtable logical view.z-bigtableadmin.projects.instances.logicalViewsz logical viewFrrr"r)rGetLogicalViewResourceSpecr&s-   5")+::#  r)returncbtjddttjdS)zCReturn the resource specification for a Bigtable materialized view.z2bigtableadmin.projects.instances.materializedViewszmaterialized viewFrrr"r)rGetMaterializedViewResourceSpecr)s-   :')+::#  r)c ztjddttt t dS)Nz/cloudkms.projects.locations.keyRings.cryptoKeyskeyF)r cryptoKeysId keyRingsId locationsIdrr)rrr r rrr"r)rGetKmsKeyResourceSpecr/s6   7(**,,.*,# r)c tjddttt tj dS)Nz1bigtableadmin.projects.instances.clusters.backupsrF)r backupsIdr!rrr)rrrrrrr"r)rGetBackupResourceSpecr2s9   9%''))+::# r)ctjj|rdndtdj ||dj |y)z0Add --instances resource argument to the parser.rrzThe instances {}.Tr]pluralNr ConceptParser ForResourcerrrUrMrrs rAddInstancesResourceArgr:sH++jM  &  , Kr)ctjj|rdndtdj ||dj |y)z/Add --instance resource argument to the parser.rrThe instance {}.Fr4Nr6)rMrrr]s rAddInstanceResourceArgr=&sH++jL%  , Kr)ctjj|rdndtdj |ddj |y)z,Add --table resource argument to the parser.rrz The table {}.TFr4N)r r7r8rrrUr9s rAddTableResourceArgr?1sH++gT"  , Kr)ctjjdtdj |dj |y)z7Add cluster positional resource argument to the parser.rThe cluster {}.TrN)r r7r8r"rrUrMrs rAddClusterResourceArgrC<sA++t$ , Kr)ctjjdtdj |dj |y)z;Add app profile positional resource argument to the parser. app_profilezThe app profile {}.TrN)r r7r8r$rrUrBs rAddAppProfileResourceArgrFFsA++!""4( , Kr)ctjjdtd|ddj |y)zrs>70+/(.56::>DG)z66 77Z44#LWvWt    C      )>)>  &M@.$  "Mr)