dZddlmZddlmZddlmZddlZddlZddlmZ ddlm Z ddlm Z dd lm Zdd lmZddlm Z dd lmZdd lmZddlZ d%d Z d&dZ d'dZ d(dZd)dZ d&dZ d*dZdZdZ d%dZd)dZ d+dZ d,dZ!dZ"d+dZ#dZ$dZ% d&dZ&d)dZ'd Z( d-d!Z) d)d"Z*d.d#Z+ d&d$Z,y)/zHReducer functions to generate instance props from prior state and flags.)absolute_import)division)unicode_literalsN)api_util) constants) exceptions) instances) arg_parsers) properties)filesct|du|du|du|du|dug}|sy|j}|||_|||_|8|jjj |j |_|r||_|||_ |S)aGenerates the Active Directory configuration for the instance. Args: sql_messages: module, The messages module that should be used. domain: string, the Active Directory domain value. mode: string, the Active Directory mode value. dns_servers: list of strings, the list of dns servers. admin_credential_secret_key: string, the name of the admin credential secret manager key. organizational_unit: string, the organizational unit value. Returns: sql_messages.SqlActiveDirectoryConfig object. N) anySqlActiveDirectoryConfigdomainadminCredentialSecretNameModeValueValuesEnumlookup_by_nameuppermode dnsServersorganizationalUnit) sql_messagesrr dns_serversadmin_credential_secret_keyorganizational_unitshould_generate_configconfigs 8lib/googlecloudsdk/api_lib/sql/instance_prop_reducers.pyActiveDirectoryConfigr$s. D $!-%     0 0 2& FM ,'BF$ 77KKZZ FK#F$ 3F -c|||y|j}|||_|tj|dz|_|tj|dz|_|S)ayGenerates the Audit configuration for the instance. Args: sql_messages: module, The messages module that should be used. bucket: string, the GCS bucket name. retention_interval: duration, how long to keep generated audit files. upload_interval: duration, how often to upload generated audit files. Returns: sql_messages.SqlServerAuditConfig object. Ns)SqlServerAuditConfigbucketsix text_typeretentionIntervaluploadInterval)rr$retention_intervalupload_intervalrs rr#r#Usp  ^*27N   , , .& FM#"}}-?@3FF MM/:S@F -r c `t|du||du|du|du|du| g} | sy|r|jjs|jdd|} n|jj} | j|jj j k(} | r| rtjd||| _ d| _ |r|| _ d| _ |X| jxs|j} |jjj| _|| _| | _ d| _ ||| _d| _ |s$||s||tjdd| _ ||| _||| _| j*|| _|r-| j&s!| j(stjd | S) aGenerates the backup configuration for the instance. Args: sql_messages: module, The messages module that should be used. instance: sql_messages.DatabaseInstance, the original instance, if the previous state is needed. backup_enabled: boolean, True if backup should be enabled. backup_location: string, location where to store backups by default. backup_start_time: string, start time of backup specified in 24-hour format. enable_bin_log: boolean, True if binary logging should be enabled. enable_point_in_time_recovery: boolean, True if point-in-time recovery (using write-ahead log archiving) should be enabled. retained_backups_count: int, how many backups to keep stored. retained_transaction_log_days: int, how many days of transaction logs to keep stored. patch_request: boolean, True if this is a patch request. Returns: sql_messages.BackupConfiguration object, or None Raises: ToolException: Bad combination of arguments. Nzsql#backupConfigurationz00:00)kind startTimeenabledzZBackup configuration cannot be changed for instances with a BackupDR backup plan attached.TzArgument --no-backup not allowed with --backup-location, --backup-start-time, --retained-backups-count, or --retained-transaction-log-daysFzArgument --retained-transaction-log-days only valid when transaction logs are enabled. To enable transaction logs, use --enable-bin-log for MySQL, and use --enable-point-in-time-recovery for Postgres and SQL Server.)rsettingsbackupConfigurationBackupConfiguration backupTierBackupTierValueValuesEnumENHANCEDsql_exceptions ArgumentErrorlocationr.r-backupRetentionSettingsBackupRetentionSettingsRetentionUnitValueValuesEnumCOUNT retentionUnitretainedBackupstransactionLogRetentionDaysbinaryLogEnabledpointInTimeRecoveryEnabledreplicationLogArchivingEnabled)rinstancebackup_enabledbackup_locationbackup_start_timeenable_bin_logenable_point_in_time_recoveryretained_backups_countretained_transaction_log_days patch_requestr backup_config gcbdr_managedbackup_retention_settingss rr1r1ssFT!D #4/D(#4/    **>> 44 &5 M %%99M  ) ) C C L LM }  & & !   ,M M/M M'-- /,,./;.R.R.o.o.u.u+0F-,EM) M".0MM- M #'8*%1  ( ( , --"M%3M""./LM,33? '2$,,66  & & '  r cg}|rEt|jD]'\}}|j|j||)|S|rg}|S|r |j}|S)aGenerates the database flags for the instance. Args: sql_messages: module, The messages module that should be used. settings: sql_messages.Settings, the original settings, if the previous state is needed. database_flags: dict of flags. clear_database_flags: boolean, True if flags should be cleared. Returns: list of sql_messages.DatabaseFlags objects namevalue)sorteditemsappend DatabaseFlags databaseFlags)rr/database_flagsclear_database_flags updated_flagsrPrQs rrUrUsv - 4 4 67 u<554u5MN8  M **M r c^g}|r(|D]#}|j|j|%|S)zGenerates the tags for the instance. Args: sql_messages: module, The messages module that should be used. tags: list of tags. Returns: list of sql_messages.Tags objects )tag)rTTags)rtags updated_tagsr[s rr\r\ s:, ,+++45 r c4|}|}|}t|||gsy|jd}|r"|jr|jjs||||t j dd|rj|jj j|jj j|jj jd} | ||_ |r7tjjj|} | dk(rd} | |_|||_|S)aUGenerates the maintenance window for the instance. Args: sql_messages: module, The messages module that should be used. instance: sql_messages.DatabaseInstance, The original instance, if it might be needed to generate the maintenance window. maintenance_release_channel: string, which channel's updates to apply. maintenance_window_day: string, maintenance window day of week. maintenance_window_hour: int, maintenance window hour of day. Returns: sql_messages.MaintenanceWindow or None Raises: argparse.ArgumentError: no maintenance window specified. Nzsql#maintenanceWindow)r,zlThere is currently no maintenance window on the instance. To add one, specify values for both day, and hour.)week5 productionpreviewr)rMaintenanceWindowr/maintenanceWindowargparser6UpdateTrackValueValuesEnumr`stablecanary updateTrackr DayOfWeekDAYSindexdayhour) rrBmaintenance_release_channelmaintenance_window_daymaintenance_window_hourchannelrnromaintenance_windownamesday_nums rrdrds/* ('# $ gsD ! " #55 "6$ (++    - - ) #/  " " ? @@   * * E E K K  * * E E L L  * * E E L L  E&+7^"##((..s3G!|g$ " r cd}|rT|jrH|jjr2|jjdr|jjd}|j}|r@|}|rt|||_|rt|||_|rt |||_|S|r|stjddt|||_t|||_|rt |||_|S)aGenerates the deny maintenance period for the instance. Args: sql_messages: module, The messages module that should be used. instance: sql_messages.DatabaseInstance, The original instance, if it might be needed to generate the deny maintenance period. deny_maintenance_period_start_date: date, Date when the deny maintenance period begins, i.e., 2020-11-01. deny_maintenance_period_end_date: date, Date when the deny maintenance period ends, i.e., 2021-01-10. deny_maintenance_period_time: Time when the deny maintenance period starts/ends, i.e., 05:00:00. Returns: sql_messages.DenyMaintenancePeriod or None Raises: argparse.ArgumentError: invalid deny maintenance period specified. NrzqThere is no deny maintenance period on the instance. To add one, specify values for both start date and end date.) r/denyMaintenancePeriodsDenyMaintenancePeriod ValidateDate startDateendDate ValidTimetimerfr6)rrB"deny_maintenance_period_start_date deny_maintenance_period_end_datedeny_maintenance_period_timeold_deny_maintenance_perioddeny_maintenance_periods rryry\s0!%8$$....q1"*"3"3"J"J1"M(>>@ 9)56*L''34(H%#,-%A"" !  / ,  " " J KK34(J%12&F##,-%A"  r c tjj|dy#t$rtjddwxYw)Nz%H:%M:%Sz2Invalid time value. The format should be HH:mm:SS.datetimestrptime ValueErrorrfr6r"s rr}r}sJD q*- D   B DDDs # Ac tjj|dS#t$rG tjj|dcYS#t$rtjddwxYwwxYw)Nz%Y-%m-%dz%m-%dz=Invalid date value. The format should be yyyy-mm-dd or mm-dd.rrs rrzrzs}Q    % %a 44 QQ    ' '7 33 Q  " " O QQQQs " A2A A2 A..A2ct|du|du|du|du|dug}|sy|j}|||_|||_|||_|||_|||_|S)aGenerates the insights config for the instance. Args: sql_messages: module, The messages module that should be used. insights_config_query_insights_enabled: boolean, True if query insights should be enabled. insights_config_query_string_length: number, length of the query string to be stored. 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. insights_config_query_plans_per_minute: number, number of query plans to sample every minute. Returns: sql_messages.InsightsConfig or None N)rInsightsConfigqueryInsightsEnabledqueryStringLengthrecordApplicationTagsrecordClientAddressqueryPlansPerMinute)r&insights_config_query_insights_enabled#insights_config_query_string_length'insights_config_record_application_tags%insights_config_record_client_address&insights_config_query_plans_per_minuterinsights_configs rrrs2,D8)5-T9+47,D8   !//1/+7.((4(KO%,8/)*6*OO'+7*PO' r c,|y|j|S)z?Generates the db aligned atomic writes Config for the instance.N)dbAlignedAtomicWrites)DbAlignedAtomicWritesConfig)rdb_aligned_atomic_writess rrrs&%   1 14 2 r ct|du|du|dugry|xs|j}|||_|Ng}t|j D]'\}}|j |j ||)||_|S|rg|_|S)z9Generates the connection pooling config for the instance.NrO)allConnectionPoolConfigconnectionPoolingEnabledrRrSrTConnectionPoolFlagsflags) renable_connection_poolingconnection_pool_flagsclear_connection_pool_flagscurrent_configconnection_pool_configrYrPrQs rrrs 4't#!T)  )P\-N-N-P*6O3&M399;< e  * *E * B=$1  ##% r c t|du|du|du|du|du|du|dugry|xs|j} ||rd| _n|jdS||| _||| _|Kg| _|j D]1\} } | j j|j| | 3||| _ ||| _ ||| _ | S)z;Generates the read pool auto-scale config for the instance.NTF)r.)metric targetValue) rReadPoolAutoScaleConfigr. minNodeCount maxNodeCount targetMetricsrSrT TargetMetricdisableScaleInscaleInCooldownSecondsscaleOutCooldownSeconds) rauto_scale_enabledauto_scale_min_node_countauto_scale_max_node_countauto_scale_target_metricsauto_scale_disable_scale_inauto_scale_in_cooldown_secondsauto_scale_out_cooldown_secondsrread_pool_auto_scale_configrrQs rrrsC D 4'4'4'!T)$,%-  > <<>#,0!) 1 1% 1 @@*/H,*/H,*02- # ( ( *  !//66  # # $  +!,1L.#/& 6%0' 7 %$r c*dj||}|S)a/Creates a custom machine type from the CPU and memory specs. Args: cpu: the number of cpu desired for the custom machine type memory_mib: the amount of ram desired in MiB for the custom machine type instance Returns: The custom machine type name for the 'instance create' call zdb-custom-{0}-{1})format)cpu memory_mib machine_types r_CustomMachineTypeStringrLs%++C<, r cd}|r|}|s|rY|stjdd|stjdd|rtjddt||dz}|}|s|stj }|S)aGenerates the machine type for the instance. Adapted from compute. Args: instance: sql_messages.DatabaseInstance, The original instance, if it might be needed to generate the machine type. tier: string, the v1 or v2 tier. memory: string, the amount of memory. cpu: int, the number of CPUs. Returns: A string representing the URL naming a machine-type. Raises: exceptions.RequiredArgumentException when only one of the two custom machine type flags are used, or when none of the flags are used. exceptions.InvalidArgumentException when both the tier and custom machine type flags are used to generate a new instance. Nz--cpuzQBoth [--cpu] and [--memory] must be set to create a custom machine type instance.z--memoryz--tierzFCannot set both [--tier] and [--cpu]/[--memory] for the same instance.i)rRequiredArgumentExceptionInvalidArgumentExceptionrrDEFAULT_MACHINE_TYPE)rBtiermemoryrrcustom_type_strings r MachineTyper[s., L F   0 0 : ;;   0 0 = >>  / / 6 772 5 &L h11L r cH|jddj||S)a[Generates the external primary configuration for the instance. Args: sql_messages: module, The messages module that should be used. source_ip_address: string, the IP address of the external data source. source_port: number, the port number of the external data source. Returns: sql_messages.OnPremisesConfiguration object. zsql#onPremisesConfigurationz{0}:{1})r,hostPort)OnPremisesConfigurationr)rsource_ip_address source_ports rrrs2  - - ( 1;? . AAr ctjtj}|jj |dt j jjjid}|jS)zGenerates the self-link of the instance's private network. Args: network: The ID of the network. Returns: string, the URL of the network. projectcompute.networks)params collection) common_api_util SqlClientAPI_VERSION_DEFAULTresource_parserParser VALUEScorer GetOrFailSelfLink)networkclient network_refs rPrivateNetworkUrlrsp  $ $_%H%H I&&&,, Z&&++33==$ -%+    r c|jd|||j}|rtj||_|rtj||_|rtj||_|jd|S)aGenerates the config for an external primary replica. Args: sql_messages: module, The messages module that should be used. primary_username: The username for connecting to the external instance. primary_password: The password for connecting to the external instance. primary_dump_file_path: ObjectReference, a wrapper for the URI of the Cloud Storage path containing the dumpfile to seed the replica with. primary_ca_certificate_path: The path to the CA certificate PEM file. client_certificate_path: The path to the client certificate PEM file. client_key_path: The path to the client private key PEM file. Returns: sql_messages.MySqlReplicaConfiguration object. zsql#mysqlReplicaConfiguration)r,usernamepassword dumpFilePathz)sql#demoteMasterMysqlReplicaConfiguration)r,mysqlReplicaConfiguration)MySqlReplicaConfigurationToUrlr ReadFileContents caCertificateclientCertificate clientKeyReplicaConfiguration)rprimary_usernameprimary_passwordprimary_dump_file_pathprimary_ca_certificate_pathclient_certificate_pathclient_key_pathmysql_replica_configurations rrrs,!- F F *)//1 !G!3 !050F0F#1%-494J4J5!1,1,B,B-)  * * 6 ; + ==r c|rF|rDtj|}tj|}||k7rtjd|rtj|}|S|S)aIGenerates the region string for the instance. Args: specified_region: string, the GCE region to create the instance in. gce_zone: string, the GCE zone to create the instance in. secondary_zone: string, the GCE zone to create the standby instance in. Returns: string, the region to create the instance in. zKZones in arguments --zone and --secondary-zone belong to different regions.)rGetRegionFromZonerConflictingArgumentsException)specified_regiongce_zonesecondary_zoneregion_from_zoneregion_from_secondary_zonederived_regions rRegionrsm.11(;!)!;!;N!K55  4 4 ) **//9N  r cn|r3|jjj|jSyN)PasswordValidationPolicyComplexityValueValuesEnumrr)r complexitys r_ParseComplexityrs3  0 0 J J Y Y  r c t|du|du|du|du|du|dug}|r|ry|j} ||| _|t||| _||| _||| _|t|dz| _||| _ | S)aDGenerates or clears password policy for the instance. Args: sql_messages: module, The messages module that should be used. password_policy_min_length: int, Minimum number of characters allowed. password_policy_complexity: string, The complexity of the password. password_policy_reuse_interval: int, Number of previous passwords that cannot be reused. password_policy_disallow_username_substring: boolean, True if disallow username as a part of the password. password_policy_password_change_interval: duration, Minimum interval at which password can be changed. enable_password_policy: boolean, True if password validation policy is enabled. clear_password_policy: boolean, True if clear existing password policy. Returns: sql_messages.PasswordValidationPolicy or None Nr") rr minLengthrr reuseIntervaldisallowUsernameSubstringstrpasswordChangeIntervalenablePasswordPolicy) rpassword_policy_min_lengthpassword_policy_complexitypassword_policy_reuse_interval+password_policy_disallow_username_substring(password_policy_password_change_intervalenable_password_policyclear_password_policyshould_generate_policypassword_policys rPasswordPolicyrs: , ,$D01=.d:D(   #8 !99;/+ :O+!1,2L"NO#/$BO!0<0[O--9-00.247.8O*'+AO( r cg}|D]}|j}|jd|_|jdx}r||_n[t j tj }|jj|jd}|j|_|jr|jr|j|tjdd|S)a~Generates PSC auto connections for the instance. Args: sql_messages: module, The messages module that should be used. psc_auto_connections: dict of the allowed consumer projects and networks. Returns: list of sql_messages.PscAutoConnectionConfig objects Raises: exceptions.InvalidArgumentException when there is no valid network or project specified. rrr)rz--psc-auto-connectionsz0PSC auto connection must have network specified.) PscAutoConnectionConfiggetconsumerNetworkconsumerProjectrrrrParseRelativeNamerrTrr)rpsc_auto_connectionsupdated_psc_auto_connections connectioncurrent_psc_auto_connectionrrrs rPscAutoConnectionsr;s""$(j"."F"F"H2<..2K/..++w+4;!1(()L)LMf**<< % 5 5'=k5@4G4G!1 $33 ' 7 7"))*EF  / / "% ))0 &%r c g}|r|D]} |jdd\}}|jdd\}}|j|j|||jjj |j |S|rg}|S#t$r} td| d} ~ wt$r} td| d} ~ wwxYw)a2Generates the database flags for the instance. Args: sql_messages: module, The messages module that should be used. unc_mappings: input unc mappings from the user. clear_unc_mappings: boolean, True if unc mappings should be cleared. Returns: list of sql_messages.UncMapping objects :=)uncPathgcsPathrz=Invalid UNC mapping input. Expected 'unc-path=gcs-path:mode'.Nz;Invalid mode. Expected 'snapshot_read' or 'snapshot_write'.) rsplitsplitrT UncMappingrrrrKeyError) r unc_mappingsclear_unc_mappingsupdated_mappings mapping_strrestmode_strunc_pathgcs_pathes r UncMappingsr1hs# $++C3h!ZZQ/(  # #  !,,@@OONN$ $  $0   K   I  s$A>B C B(( C4 CCc*t|du|dug}|sy||s|tjd|r|jjs|j }n|jj}|||_|||_d|_| |sd|_|S)a Generates the Final Backup configuration for the instance. Args: sql_messages: module, The messages module that should be used. instance: sql_messages.DatabaseInstance, the original instance, if the previous state is needed. final_backup_enabled: boolean, True if final backup should be enabled. final_backup_retention_days: int, how many days to retain the final backup. Returns: sql_messages.FinalBackupConfiguration object, or None Raises: sql_exceptions.ArgumentError: Bad combination of arguments. NzPYou cannot set final-backup-retention-days while final-backup field is disabled.T)rr5r6r/finalBackupConfigFinalBackupConfigr. retentionDays)rrBfinal_backup_enabledfinal_backup_retention_daysrfinal_backup_configs rFinalBackupConfigurationr9s*$&!-  %.B".  ( ( \  **<<&88:"++==%"6 ,(C%"&%.B(,% r )NNNNN)NNN) NNNNNNNNF)NNFr)NNz00:00:00)NNNN)NNNNNNNN)NNNNNNN)NF)-__doc__ __future__rrrrfrgooglecloudsdk.api_lib.sqlrrrrr5r googlecloudsdk.callioper googlecloudsdk.corer googlecloudsdk.core.utilr r%rr#r1rUr\rdryr}rzrrrrrrrrrrrrrr1r9r rrAsO''B0C</.**   $ .d!%,0)-@"&"&z| !%',8(37-1.2 ;@>B;?7A >!BDQ;?7;;?9=:> 2j# $  H""" $#'$(9%x 5p A  .6:15)- &=R0 $##'04-1:~*&Z%T $ 6r