dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZej:j<fdZ d0dZ d1dZ! d2dZ"dZ#dZ$dZ%d Z&d!Z'd"Z(d#Z)d$Z*d%Z+d&Z, d3d'Z- d3d(Z.d)Z/d*Z0d+Z1d,Z2d-Z3ej:jhfd.Z5d/Z6y)4zKHelper functions for constructing and validating AlloyDB instance requests.)absolute_import)division)unicode_literals)api_util)baseDetailedArgumentError) labels_util) propertiesctt|||}|j||j|jS)axValidates command line input arguments and passes parent's resources for GA track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. project_ref: parent resource path of the resource being created args: Command line input arguments. Returns: Fully-constructed request to create an AlloyDB instance. instance instanceIdparent)_ConstructInstanceFromArgs6AlloydbProjectsLocationsClustersInstancesCreateRequestr RelativeNameclientalloydb_messages project_refargsinstance_resources 9lib/googlecloudsdk/command_lib/alloydb/instance_helper.py ConstructCreateRequestFromArgsGArsE19I4PMM$]]))+Nctt|||}|j||j|jS)a{Validates command line input arguments and passes parent's resources for beta tracks. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. project_ref: Parent resource path of the resource being created args: Command line input arguments. Returns: Fully-constructed request to create an AlloyDB instance. r )_ConstructInstanceFromArgsBetarrrrs r"ConstructCreateRequestFromArgsBetar5sH5  MM$]]))+Nrctt|||}|j||j|jS)a{Validates command line input arguments and passes parent's resources for alpha track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. project_ref: Parent resource path of the resource being created args: Command line input arguments. Returns: Fully-constructed request to create an AlloyDB instance. r )_ConstructInstanceFromArgsAlpharrrrs r#ConstructCreateRequestFromArgsAlphar"PsH6  MM$]]))+Nrc|js |jr'|j|j|jStd)zJValidates command line input arguments and creates a MachineConfig object.cpuCount machineTypez7Either --cpu-count or --machine-type must be specified.) cpu_count machine_type MachineConfigr )rrs r$ConstructCreateMachineConfigFromArgsr*ksJ ^^t((  ) )T->-> *  A rc|j}t||j|_t |||_|j jdtjjjj|j|j|j}|j!|_t%j&||jj(d|_t-||j.|_|j0|jj2j4k(r!|j7|j8|_t=||j>|j@|jB|jD|_#tI||jJ|jL|_'tQ||jR|jT|jV|jX|_-|j\s|j^ |j`2tc||j\|j^|j`|_2|jfrti||jf|jj|jl|jn|jp|jr|jt|jv|jx|jz|j||j~| |_@|S) aUValidates command line input arguments and passes parent's resources to create an AlloyDB instance. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. args: Command line input arguments. Returns: An AlloyDB instance to create with the specified command line arguments. -alloydb.projects.locations.clusters.instances projectsId locationsId clustersId instancesIddatabase_flags labels_dest nodeCount)#insights_config_query_string_length&insights_config_query_plans_per_minute'insights_config_record_application_tags%insights_config_record_client_addressrassign_inbound_public_ipauthorized_external_networksoutbound_public_ipallocated_ip_range_overriderallowed_psc_projectspsc_network_attachment_uripsc_auto_connectionsrenable_connection_poolingconnection_pooling_pool_mode connection_pooling_min_pool_size connection_pooling_max_pool_size"connection_pooling_max_client_conn&connection_pooling_server_idle_timeout%connection_pooling_query_wait_timeoutconnection_pooling_stats_users,connection_pooling_ignore_startup_parameters"connection_pooling_server_lifetime1connection_pooling_client_connection_idle_timeout*connection_pooling_max_prepared_statementsr)AInstanceParseAvailabilityTypeavailability_typeavailabilityTyper* machineConfigresource_parserCreater VALUEScoreproject GetOrFailregionclusterrrnamer ParseCreateArgsDatabaseFlagsValue databaseFlags_ParseInstanceType instance_type instanceTypeInstanceTypeValueValuesEnum READ_POOLReadPoolConfigread_pool_node_countreadPoolConfig_QueryInsightsConfigr7r8r9r:queryInsightsConfigClientConnectionConfigssl_moderequire_connectorsclientConnectionConfig NetworkConfigr<r=r>r? networkConfigrArBrCPscInstanceConfigpscInstanceConfigrE_ConnectionPoolConfigrFrGrH)connection_pooling_max_client_connectionsrJrKrLrMrNrOrPconnectionPoolConfigrrrr instance_refs rrrws'//1(=..($%I%!''..5""''//99++-- /,(446$/$?$? 22"%! $6**$ $$  " " > > H HI(8'F'F++(G($+?*.*R*R-1-X-X.2.Z.Z,0,V,V +'.D mm .* %2'#<<#'#D#D00"&"B"B %!  ( ( 4 " " .*;)!66#'#B#B!66 +' ##-B)"&"@"@%)%F%F)-)N)N)-)N)N+/+Y+Y/3/Z/Z.2.X.X'+'J'J595f5f+/+R+R:>:p:p373b3b .*" rc t|||}t||j|j|j|j |j |j|j|_ |S)adValidates command line input arguments and passes parent's resources to create an AlloyDB instance for beta track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. args: Command line input arguments. Returns: An AlloyDB instance to create with the specified command line arguments. )observability_config_enabled&observability_config_preserve_comments&observability_config_track_wait_events,observability_config_max_query_string_length,observability_config_record_application_tags+observability_config_query_plans_per_minute)observability_config_track_active_queries) r_ObservabilityConfigrzr{r|r}r~rrobservabilityConfigrrrrs rrrsl19I4P*>#'#D#D-1-X-X-1-X-X373d3d373d3d262b2b040^0^ +' rc t|||}|S)aeValidates command line input arguments and passes parent's resources to create an AlloyDB instance for alpha track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. args: Command line input arguments. Returns: An AlloyDB instance to create with the specified command line arguments. )rrs rr!r!s5  rc|j}|jjdtjj j j|j|j|j}|j|_ |jjj|_t!||j"|_t'||j(|j*|_t/j0||jj2d|_t7||j8|j:|j<|j>|_ |jBs|jD |jF2tI||jB|jD|jF|_%|jLrtO||jL|jP|jR|jT|jV|jX|jZ|j\|j^|j`|jb|jd||_3|S)zmValidates command line input arguments and passes parent's resources to create an AlloyDB secondary instance.r,r-r2r3r;r@rD)4rQrVrWr rXrYrZr[r\r]rrr^re SECONDARYrdrRrSrTrlrmrnror r_r`rarpr<r=r>r?rqrArBrCrrrsrErtrFrGrHrurJrKrLrMrNrOrPrvrws r#_ConstructSecondaryInstanceFromArgsrs'//1''..5""''//99++-- /,(446;;EE (=..($.D t'>'>.*%0$?$? 22"%! %2'#<<#'#D#D00"&"B"B %!  ( ( 4 " " .*;)!66#'#B#B!66 +' ##-B)"&"@"@%)%F%F)-)N)N)-)N)N+/+Y+Y/3/Z/Z.2.X.X'+'J'J595f5f+/+R+R:>:p:p373b3b .*" rct|||S)a|Validates command line input arguments and passes parent's resources to create an AlloyDB secondary instance for beta track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. args: Command line input arguments. Returns: An AlloyDB secondary instance to create with the specified command line arguments. )r)rrrs r'_ConstructSecondaryInstanceFromArgsBetarJs -  rc t|||}|S)a}Validates command line input arguments and passes parent's resources to create an AlloyDB secondary instance for alpha track. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. args: Command line input arguments. Returns: An AlloyDB secondary instance to create with the specified command line arguments. )rrs r(_ConstructSecondaryInstanceFromArgsAlphar\s>  rctt|||}|j||j|jS)zRValidates command line input arguments and passes parent's resources for GA track.r )r?AlloydbProjectsLocationsClustersInstancesCreatesecondaryRequestrrrr cluster_refrrs r)ConstructSecondaryCreateRequestFromArgsGArpsH :   Y Y   % % ' Z rctt|||}|j||j|jS)zTValidates command line input arguments and passes parent's resources for beta track.r )rrrrrs r+ConstructSecondaryCreateRequestFromArgsBetarsH>   Y Y   % % ' Z rctt|||}|j||j|jS)zUValidates command line input arguments and passes parent's resources for alpha track.r )rrrrrs r,ConstructSecondaryCreateRequestFromArgsAlpharsH?   Y Y   % % ' Z rct|||tjj\}}|rdj |nd}|j ||j |S)a1Constructs the request to update an AlloyDB instance. Args: alloydb_messages: Messages module for the API client. instance_ref: parent resource path of the resource being updated args: Command line input arguments. Returns: Fully-constructed request to update an AlloyDB instance.  release_track,Nrr^ updateMask)'ConstructInstanceAndUpdatePathsFromArgsr ReleaseTrackGAjoin5AlloydbProjectsLocationsClustersInstancesPatchRequestrrrxrrpathsmasks rConstructPatchRequestFromArgsrshE d$:K:K:N:NU"%t$  O O  $ $ & P rc d}d}d}d}d}d} d} d} d } d } |j}g}|j|_|jr"|j|_|j | t ||j}|r||_|j |tj||jjd }|r||_ |j ||js |jrf|j|j|j |_|jr|j ||jr|j ||j"r2|j%|j"|_|j ||j(r|j | |j*r|j | |j,|j | |j.|j | t1||j(|j*|j,|j.|_|j4d}|j ||j6rd}|j ||j4 |j6r&t9||j6|j4|_|j<s|j> |j@2tC||j<|j>|j@|_"|j@d}|j ||j<r9|jDjFs#|j d|j dn:|j<r|j d|j>|j d|jH0|jJ$|jLs|jN |jPrHtS||jH|jJ|jL|jN|jP|_*|jJ |jLr|j d|jH|j d|jN |jPr|j d|tVjXjZk(r8|j\|j^x|j`l|jb`|jdT|jfH|jh<|jj0|jl$|jn|jp |jr|j dtu||||j\|j^|j`|jb|jd|jf|jh|jj|jl|jn|jp|jr|_;||fS)asValidates command line arguments and creates the instance and update paths. Args: alloydb_messages: Messages module for the API client. instance_ref: parent resource path of the resource being updated args: Command line input arguments. release_track: The release track of the gcloud client. Returns: An AlloyDB instance and paths for update. rTrazmachineConfig.cpuCountzmachineConfig.machineTypezreadPoolConfig.nodeCountz%queryInsightsConfig.queryStringLengthz'queryInsightsConfig.queryPlansPerMinutez)queryInsightsConfig.recordApplicationTagsz'queryInsightsConfig.recordClientAddressactivationPolicyr2r3r$r5z(clientConnectionConfig.requireConnectorsz(clientConnectionConfig.sslConfig.sslMode)rr<r=r>z$networkConfig.enableOutboundPublicIpznetworkConfig.enablePublicIpz(networkConfig.authorizedExternalNetworks)rrArB clear_psc_network_attachment_urirCclear_psc_auto_connectionsz%pscInstanceConfig.pscInterfaceConfigsz)pscInstanceConfig.allowedConsumerProjectsz$pscInstanceConfig.pscAutoConnectionsrvrrrErFrGrHrIrJrKrLrMrNrOrP)rprqenablePublicIprArBrrCrrrrsrrrrErFrGrHrurJrKrLrMrNrOrP_UpdateConnectionPoolConfigrv)rrxrravailability_type_pathdatabase_flags_pathcpu_count_pathmachine_type_pathread_pool_node_count_path(insights_config_query_string_length_path+insights_config_query_plans_per_minute_path,insights_config_record_application_tags_path*insights_config_record_client_address_pathactivation_policy_pathrrrSr2require_connectors_path ssl_mode_pathoutbound_public_ip_paths rrrs.'+.18-+0.2/0-.&//1 %'446 )-)?)?& LL'(+..):& LL'(.. 22". &4# LL$% ^^t((&6&D&DT->->'E'# ~~ ll>"  ll$% '7'F'F++(G($ LL*+ -- LL9: 00 LL<= 11= LL=> //; LL;<*> .. 11 22 00 +' (H LL() ]]>M LL (DMM/E$--)@)@0,  ## * * 6 ,&3)!%!>!>%)%F%F22 '#  (D LL()  ##--<< LL/0 LL;< $$ ll12 ((4 ll=> + ( ( 4 . . " " . ( (*;)!66#'#B#B)-)N)N!66#'#B#B +' %%1 . . LL89 * LL<= *d.M.M LL78t((+++ $$0 * * 6 . . : . . : 7 7 C 4 4 @ 3 3 ? , , 8 : : F 0 0 < ? ? K 8 8 D LL'(-H#)"&"@"@%)%F%F)-)N)N)-)N)N+/+Y+Y/3/Z/Z.2.X.X'+'J'J595f5f+/+R+R:>:p:p373b3b.*$ E !!rNct|du|du|du|dug}|sy|j}|||_|||_|||_|||_|S)aWGenerates the insights config for the instance. Args: alloydb_messages: module, Message module for the API client. insights_config_query_string_length: number, length of the query string to be stored. insights_config_query_plans_per_minute: number, number of query plans to sample every minute. insights_config_record_application_tags: boolean, True if application tags should be recorded. insights_config_record_client_address: boolean, True if client address should be recorded. Returns: alloydb_messages.QueryInsightsInstanceConfig or None N)anyQueryInsightsInstanceConfigqueryStringLengthqueryPlansPerMinuterecordApplicationTagsrecordClientAddress)rr7r8r9r:should_generate_configinsights_configs rrjrj{s0)5,D8-T9+47   %@@B/(4(KO%+7*PO',8/)+6*OO' rc t|du|du|du|du|du|du|dug}|sy|j} ||| _||| _||| _||| _||| _||| _||| _| S)aGenerates the observability config for the instance. Args: alloydb_messages: module, Message module for the API client. observability_config_enabled: boolean, True if observability should be enabled. observability_config_preserve_comments: boolean, True if comments should be preserved in the query string. observability_config_track_wait_events: boolean, True if wait events should be tracked. observability_config_max_query_string_length: number, length of the query string to be stored. observability_config_record_application_tags: boolean, True if application tags should be recorded. observability_config_query_plans_per_minute: number, number of query plans to sample every minute. observability_config_track_active_queries: boolean, True if active queries should be tracked. Returns: alloydb_messages.ObservabilityInstanceConfig or None N) rObservabilityInstanceConfigenabledpreserveCommentstrackWaitEventsmaxQueryStringLengthrrtrackActiveQueries) rrzr{r|r}r~rrrobservability_configs rrrsB"$.,D8,D82$>2$>1=/t;  *EEG!-#? +7.),7.(2=4-2=4.1<3,/:1+ rct|du|dug}|sy|j}||_|j}t |||_||_|S)aGenerates the client connection config for the instance. Args: alloydb_messages: module, Message module for the API client. ssl_mode: string, SSL mode to use when connecting to the database. require_connectors: boolean, whether or not to enforce connections to the database to go through a connector (ex: Auth Proxy). Returns: alloydb_messages.ClientConnectionConfig N)rrlrequireConnectors SslConfig _ParseSSLModesslMode sslConfig)rrmrnrclient_connection_config ssl_configs rrlrlss"d$  .DDF/A,))+*$%5x@*'1$ !!rcn|r3|jjj|jSyN)rQAvailabilityTypeValueValuesEnumlookup_by_nameupper)rrSs rrRrRs5  $ $ D D S S!  rcn|r3|jjj|jSyr)rQrerr)rrcs rrbrb$s5  $ $ @ @ O O  rcn|r3|jjj|jSyr) UpdatePolicyModeValueValuesEnumrr)r update_modes r_ParseUpdateModer,s5  ( ( < < K K  rc|dk(r |jjjS|dk(r |jjjSy)NENCRYPTED_ONLYALLOW_UNENCRYPTED_AND_ENCRYPTED)rSslModeValueValuesEnumrr)rrms rrr4sK !!  % % < < K KK44""99YY rc|dk(r |jjjS|dk(r |jjjSy)N TRANSACTIONSESSION)ConnectionPoolConfigPoolModeValueValuesEnumPOOL_MODE_TRANSACTIONPOOL_MODE_SESSION)r pool_modes r_ParsePoolModer>sM---EE[[I--EEWW rc |jd}|jd}|jd}|jd}|jd}t||du|du|dug}|sy|j}|rt||_|||_|5||js t dt|||j|_|||_ |S)z.Generates the network config for the instance.r<r=rr>r?NzCannot update an instance's authorized networks and disable Public-IP. You must do one or the other. Note, that disabling Public-IP will clear the list of authorized networks.) getrInstanceNetworkConfig_ParseAssignInboundPublicIprenableOutboundPublicIpr _ParseAuthorizedExternalNetworksauthorizedExternalNetworksallocatedIpRangeOverride)kwargsr<r=rr>r?rinstance_network_configs rrprpJs#ZZ(BC!',J!KZZ 23zz"67 & +H I$"$.!-   -BBD-H .*#5G2!- ,'66 !   )  ( # 2 2 6!,#4 ! rc |jd}|sy|jd}|jd}|jd}|jd}|jd}|jd}|jd }|jd } |jd } |jd } |jd } |jd} | j}||_i}||j|d<|||d<|||d<|||d<|||d<|||d<|dj ||d<| dj | |d<| | |d<| | |d<| t | |d<| jj |jDcgc],\}}| jj j||.c}}|_ |Scc}}w)z9Generates the connection pooling config for the instance.rENrFrGrHrIrJrKrLrMrNrOrPrr min_pool_size max_pool_sizemax_client_connectionsserver_connection_idle_timeoutquery_wait_timeoutr stats_usersignore_startup_parametersserver_lifetimeclient_connection_idle_timeoutmax_prepared_statementskeyvalueadditionalProperties) rrrlowerrstr FlagsValueitemsAdditionalPropertyflags)rrErrdefault_pool_sizemax_client_connserver_idle_timeoutrrrrrrrconfigr rrs rrtrtsP$jj)DE " jj78)**?@-jj!CDJJCD/ #KLzz"IJ ;<+$jj4JJ*/$*::9$ #JJ2ZZ 23  0 0 2&,&. %"*E+*E/".E/ &5E "#$.AE *+#"4E 88K0E-*),2K)LE %& .E #/& *+('*+B'CE #$!66AA "KKM *jc5  / / : : M MU N * B&, -s$1G% c |jd}|jd}|jd}|jd}|jd}|jd}|jd} |jd} |jd } |jd } |jd } |jd }|jd }t|du|du|du|du|du|du| du| du| du| du| du|dug }|sy|j}i}|s | d|_|St j |}|j }|j|j}|jj|}|jdu}|r8|jj"gd}|D]}t||\}}|s|||<|||_n|r|jj|_||j|d<|||d<|||d<|||d<|||d<| | |d<| dj| |d<| dj| |d<| | |d<| | |d<|t!||d<|jj#|j%Dcgc],\}}|jj"j'||.c}}|_ |Scc}}w) aSUpdates the connection pooling config for the instance. Args: instance_ref: A reference to the instance to be updated. release_track: The release track of the gcloud client. **kwargs: A map of the managed connection pooling flags and their values to be updated. Returns: alloydb_messages.ConnectionPoolConfig rErFrGrHrIrJrKrLrMrNrOrPrNFr^) rrrrrrrrrrrrrrrrrrrrrrrrr)rrrrr AlloyDBClientalloydb_client3AlloydbProjectsLocationsClustersInstancesGetRequestr%projects_locations_clusters_instancesGetrvr &_CheckIfConnectionPoolConfigFlagExistsrrrrrr )rxrrrErrr r r rrrrrrrshould_update_configrr rrreqexisting_instancehas_existing_config flag_namesfexistsrrs rrrs%jj)DEjj78)**?@-jj!CDJJCD/ #KLzz"IJ ;<+$jj4JJ*/$*::9$ #JJ2ZZ 23t+t4t#T!%$t+T!$D0T)      0 0 2& % #'@'LFN M  ! !- 0&((.LL  $ $ & M #%JJNN *>>dJ  0 0 6 6 B J< Qmfe a *.FN&;;CCFN"*E+*E/".E/ &5E "#$.AE *+#"4E 88K0E-*),2K)LE %& .E #/.LE *+('*+B'CE #$!66AA "KKM *jc5  / / : : M MU N * B&, -s$1K% c|jjjy|jjjD]!}|j|k(sd|jfcSy)a#Checks if a flag exists in the instance's connection pool config. Args: instance: The existing instance to check against. flag_name: The name of the flag to check if it exists in the instance. Returns: True and the value of the flag if the flag exists, False otherwise. )FNT)rvr rrr)r flag_nameprops rrrIs\""((==E ++11FFd xx9 4:: G rc |jd}|jd}|jd}|jd}|jd}|jd}|j}|r||_|rg|_n(|&|jj t |||r g|_|S|t|||_|S) z3Generates the PSC instance config for the instance.rrArBrrCr)rrB)rrC)rrrallowedConsumerProjectspscInterfaceConfigsr_PscInterfaceConfigpscAutoConnections_PscAutoConnections)rrrArBrrCrpsc_instance_configs rrrrr\sZZ 23$:;%zz*FG%+ZZ(&" $:;%zz*FG(::<2F/%.0+!-++22-'A   -/*  '-@)1.* rc4|j}||_|S)z4Generates the PSC interface config for the instance.)PscInterfaceConfignetworkAttachmentResource)rrBpsc_interface_configs rr$r$~s" *<<>3M0 rcg}|D]r}|j}|jd|_|jd|_|jr|jr|j |it d|S)z4Generates the PSC auto connections for the instance.rZnetworkzaInvalid PSC auto connection. Please provide both project and network for the PSC auto connection.)PscAutoConnectionConfigrconsumerProjectconsumerNetworkrr )rrCout_psc_auto_connections connectionrs rr&r&s}  (j  5 5 7F'^^I6F'^^I6F &"8"8%%f- ! * ) "!rc0|dk(ry|dk(rytd)aCParses the assign_inbound_public_ip flag. Args: assign_inbound_public_ip: string, the Public-IP mode to use. Returns: boolean, whether or not Public-IP is enabled. Raises: ValueError if try to use any other value besides NO_PUBLIC_IP during instance creation, or if use an unrecognized argument. NO_PUBLIC_IPF ASSIGN_IPV4Tz>Unrecognized argument. Please use NO_PUBLIC_IP or ASSIGN_IPV4.r)r<s rrrs*/ . F rc|g}||s|S|D].}|jt|}|j|0|S)a8Parses the authorized_external_networks flag. Args: alloydb_messages: Messages module for the API client. authorized_external_networks: list, list of authorized networks. public_ip_enabled: boolean, whether or not Public-IP is enabled. Returns: list of alloydb_messages.AuthorizedNetwork ) cidrRange)AuthorizedNetworkrr)rr=public_ip_enabled auth_networksr-s rrrsO-"+< -g003w<0HG!. rct|||tjj\}}|rdj |nd}|j ||j |Sz5Constructs the request to update an AlloyDB instance.rrNr)+ConstructInstanceAndUpdatePathsFromArgsBetarrBETArrrrs r!ConstructPatchRequestFromArgsBetar?sfH d%%**U"%t$  O O  $ $ & P rct|||tjj\}}|rdj |nd}|j ||j |Sr<),ConstructInstanceAndUpdatePathsFromArgsAlpharrALPHArrrrs r"ConstructPatchRequestFromArgsAlpharCsfI d%%++U "%t$  O O  $ $ & P rcd}d}d}d}d}d} d} t||||\} } |jr>|jt||j| _d } | j | |j | j ||j| j ||j| j ||j| j ||j| j ||j| j | |j| j | t||j |j|j|j|j|j|j| _|tj j"k(r8|j$|j&x|j(l|j*`|j,T|j.H|j0<|j20|j4$|j6|j8 |j:| j d t=||||j$|j&|j(|j*|j,|j.|j0|j2|j4|j6|j8|j: | _| | fS) aValidates command line arguments and creates the instance and update paths for beta track. Args: alloydb_messages: Messages module for the API client. instance_ref: parent resource path of the resource being updated args: Command line input arguments. release_track: The release track of the gcloud client. Returns: An AlloyDB instance and paths for update. zobservabilityConfig.enabledz$observabilityConfig.preserveCommentsz#observabilityConfig.trackWaitEventsz(observabilityConfig.maxQueryStringLengthz)observabilityConfig.recordApplicationTagsz'observabilityConfig.queryPlansPerMinutez&observabilityConfig.trackActiveQueries)modezupdatePolicy.modervr) rrrr updatePolicyrrzr{r|r}r~rrrrrrr>rErFrGrHrurJrKrLrMrNrOrPrrv)rrxrr!observability_config_enabled_path+observability_config_preserve_comments_path+observability_config_track_wait_events_path1observability_config_max_query_string_length_path1observability_config_record_application_tags_path0observability_config_query_plans_per_minute_path.observability_config_track_active_queries_pathrrupdate_mode_paths rr=r=s'D#,.,.142403/1E dMU %5%B%B .0@0@ A&C&"+ LL!" &&2 LL23 00< LL<= 00< LL<= 66B LLBC 66B LLBC 55A LLAB 33? LL?@*> '' 11 11 77 77 66 44 +'&t((--- $$0 * * 6 . . : . . : 7 7 C 4 4 @ 3 3 ? , , 8 : : F 0 0 < ? ? K 8 8 D LL'(-H#)"&"@"@%)%F%F)-)N)N)-)N)N+/+Y+Y/3/Z/Z.2.X.X'+'J'J595f5f+/+R+R:>:p:p373b3b.*$ E !!rct||||\}}|j|jx|jl|j`|j T|j H|j<|j0|j$|j|j |j|jdt||||j|j|j|j|j |j |j|j|j|j|j|j|_||fS)aValidates command line arguments and creates the instance and update paths for alpha track. Args: alloydb_messages: Messages module for the API client. instance_ref: parent resource path of the resource being updated args: Command line input arguments. release_track: The release track of the gcloud client. Returns: An AlloyDB instance and paths for update. rvr)r=rErFrGrHrurJrKrLrMrNrOrPrrrv)rrxrrrrs rrArAYsMI dMU $$0 * * 6 . . : . . : 7 7 C 4 4 @ 3 3 ? , , 8 : : F 0 0 < ? ? K 8 8 D LL'(-H#)"&"@"@%)%F%F)-)N)N)-)N)N+/+Y+Y/3/Z/Z.2.X.X'+'J'J595f5f+/+R+R:>:p:p373b3b.*" E !!rc|j|j}|jr#|j|j}||_|S)a2Constructs the request to restart an AlloyDB instance. Args: alloydb_messages: Messages module for the API client. project_ref: parent resource path of the resource being updated args: Command line input arguments. Returns: Fully-constructed request to restart an AlloyDB instance. r)nodeIds)7AlloydbProjectsLocationsClustersInstancesRestartRequestrnode_idsRestartInstanceRequestrestartInstanceRequest)rrrrrestart_requests rConstructRestartRequestFromArgsrWs]NN'')O  ]]&== >O"1C *r)NNNN)NNNNNNN)NNr)7__doc__ __future__rrrgooglecloudsdk.api_lib.alloydbrgooglecloudsdk.callioper%googlecloudsdk.calliope.parser_errorsr $googlecloudsdk.command_lib.util.argsr googlecloudsdk.corer rrr"r*rrr!rrrrrrrrrrrjrrlrRrbrrrrprtrrrrr$r&rrr?rCr=rBrArWrrr`sFR&'3(G<*266 bJ4$?D$(   09=8I8I8L8LE"T)-+/,0*. .f"&+/+/151504.2J^ "F 2!j?DEP&H $"*,,   n"d9=8I8I8O8O5"p r