>dZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd l m Z dd l m Zdd lmZdd lmZddlm Zdd lmZddlmZddlmZdZdZdZdZdZdZe j>e j@e jBjDe jBjFe jBjHGdde jJZ&y)z*Restores a backup of a Cloud SQL instance.)absolute_import)division)unicode_literals)api_util) exceptions) operations)validate)base) resource_args)flags) instances)log) properties) console_ioi)@activation_policyactive_directory_dns_serversactive_directory_domainactive_directory_mode$active_directory_organizational_unit#active_directory_secret_manager_key assign_ipauthorized_networksavailability_typebackupbackup_start_timebackup_location"clear_active_directory_dns_serverscpu collationenable_bin_logretained_backups_countretained_transaction_log_daysfailover_replica_namemaintenance_release_channelmaintenance_window_daymaintenance_window_hour"deny_maintenance_period_start_date deny_maintenance_period_end_datedeny_maintenance_period_time&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_minutememory require_sslstorage_auto_increase storage_sizestorage_provisioned_iopsstorage_provisioned_throughput storage_typetiereditionenable_point_in_time_recoverynetwork clear_networkaudit_bucket_pathdeletion_protection time_zoneconnector_enforcementtimeoutenable_google_private_pathenable_data_cacheenable_private_service_connectallowed_psc_projectsssl_modeenable_google_ml_integrationdatabase_versiondisk_encryption_keydisk_encryption_key_keyringdisk_encryption_key_locationdisk_encryption_key_projectpsc_auto_connectionsserver_ca_moderegionretain_backups_on_delete final_backupfinal_backup_retention_daysc> |jjtjtj|tj |tj |tj|tj|tj|tj|tj|dtj|tj|tj|tj|tj |tj"|tj$|tj&|tj(|tj*|tj,|tj.|tj0|tj2|tj4|tj6|tj8|tj:|tj<|tj>|tj@|tjB|tjD|tjF|tjH|ddddd}tKjL|d| tjN|tjP|tjR|tjT|tjV|tjX|tjZ|tj\|t^tj`|d tjb|d |je}tjf|tjh|tjj|d tjl|tjn|d tjp|d tjr|dd tjt|ddddtjv|d tjx|d tjz|d tj||tj~|tj|d tj|d tj|d tj|d tj|d tj|d y)z#Declare flag for instance settings.F) allow_emptyz--disk-encryption-keyz--disk-encryption-key-keyringz--disk-encryption-key-locationz--disk-encryption-key-project)zkms-keyz kms-keyringz kms-locationz kms-projectinstance)flag_overrides)show_negated_in_helpT)hidden)rVspecify_default_regionzt Note for restore to new instance major version upgrades are not supported. Only minor version upgrades are allowed.)restrict_choicesrVsupport_default_versionadditional_help_textN)F display_info AddFormatr GetInstanceListFormatAddActivationPolicyAddActiveDirectoryDomain AddAssignIpAddAuthorizedNetworksAddAvailabilityType AddBackupAddBackupStartTimeAddBackupLocationAddCPUAddInstanceCollationAddEnableBinLogAddRetainedBackupsCountAddRetainedTransactionLogDaysAddMaintenanceReleaseChannelAddMaintenanceWindowDayAddMaintenanceWindowHour!AddDenyMaintenancePeriodStartDateAddDenyMaintenancePeriodEndDateAddDenyMaintenancePeriodTime%AddInsightsConfigQueryInsightsEnabled"AddInsightsConfigQueryStringLength&AddInsightsConfigRecordApplicationTags$AddInsightsConfigRecordClientAddress$AddInsightsConfigQueryPlansPerMinute AddMemory AddRequireSslAddStorageAutoIncreaseAddStorageSizeAddStorageProvisionedIopsAddStorageProvisionedThroughputAddStorageTypeAddTier AddEditionkms_resource_argsAddKmsKeyResourceArgAddEnablePointInTimeRecovery AddNetworkAddClearNetworkAddSqlServerAuditAddDeletionProtectionAddSqlServerTimeZoneAddConnectorEnforcement AddTimeout"_INSTANCE_CREATION_TIMEOUT_SECONDSAddEnableGooglePrivatePathAddEnableDataCache add_groupAddEnablePrivateServiceConnectAddAllowedPscProjectsAddPscAutoConnections AddSslModeAddEnableGoogleMLIntegrationAddEnableDataplexIntegrationAddLocationGroupAddDatabaseVersionAddServerCaModeAddTagsAddRetainBackupsOnDeleteAddFinalBackupAddFinalbackupRetentionDaysAddActiveDirectoryMode"AddActiveDirectorySecretManagerKey$AddActiveDirectoryOrganizationalUnitAddActiveDirectoryDNSServersClearActiveDirectoryDNSServersAddClearActiveDirectory)parserkms_flag_overridespsc_setup_groups "lib/surface/sql/backups/restore.pyAddInstanceSettingsArgsrjs ; ; =>F#  (Ff%F#//&6"&e4,,vV$'%%f-$$V,'  ())&1''/$$V,--f5**62..v6,,V4,,V4//&fv&v!!&)''/v--6(464  (( j);$$V,6&!f%V$'6=>""6F6$/$$&/&&7o.fT26$$VD9$$VD9eL  # A t,--t$  5v##F+vd3**6$?,,VDA$$VD9&&vd;t4c>tj|jtj|j s|rC|j rtjd|jrtjdy|rtjd|y)aValidates the backup request. Args: is_project_backup: bool, Whether the backup request is for a project level backup. args: argparse.Namespace, The arguments that this command was invoked with. overrides: list[str], The list of flags that were overridden. zP --backup-project is not supported when using backup name based restore command.zQ --backup-instance is not supported when using backup name based restore command.z_Overrides are only supported for backup name based restore to new instance. Unsupported flags: N) r ValidateInstanceNamerestore_instancecommand_validateIsBackupDrBackupRequestidbackup_projectr ArgumentErrorbackup_instance)is_project_backupargs overridess r_ValidateBackupRequestrs  5 56--dgg6:K   $ $      $ $     $ $ ++4+ 7 rcHtj|jrB|j|j|j |j |jS|j|j|j |j |jS)aEGet the restore backup request. Args: args: argparse.Namespace, The arguments that this command was invoked with. sql_messages: sql_v1beta4_messages.SqlMessagesV1Beta4, The SQL API messages. instance_ref: base.ResourceParse, The parsed instance reference. Returns: A SqlInstancesRestoreBackupRequest. )backupdrBackupprojectrSinstancesRestoreBackupRequest)r)rrr SqlInstancesRestoreBackupRequestrrSInstancesRestoreBackupRequest)r sql_messages instance_refs r_GetRestoreBackupRequestrs--dgg6  8 8$$&&&2&P&P77'Q'  9   8 8$$&&&2&P&P77'Q'  9 rcg}|jdr|jd|jr|jd|jdr|jd|jr|jd|S)zGet the database instance clear overrides. Args: args: argparse.Namespace, The arguments that this command was invoked with. Returns: An array with the database instance fields that should be cleared. r:z)settings.ip_configuration.private_networkz,settings.active_directory_config.dns_serversclear_disk_encryptiondisk_encryption_configz settings.active_directory_config)IsKnownAndSpecifiedappendrclear_active_directory)rcleared_fieldss r!_GetRestoreInstanceClearOverridesrsz. o.EF ,,HI 5623   <= rc&eZdZdZedZdZy) RestoreBackupamRestores a backup of a Cloud SQL instance. The command lets you restore to an existing instance using ID. To restore using a backupDR backup, use the backupDR ID. When backup Name is used to restore it lets you restore to an existing instance or a new instance. When restoring to new instance, optional flags can be used to customize the new instance. c^tj|d|jddtjd|jdtjd |jd d |jd ddtj j |t|y)zArgs is called by calliope to gather arguments for this command. Args: parser: An argparse parser that you can use to add arguments that go on the command line after this command. Positional arguments are allowed. zThe ID of the backup run to restore from or the backup NAME for restore to existing/new instance. To find the NAME, run the following command: $ gcloud sql backups list --filter=instance:{instance}) help_textz--restore-instanceTzGThe ID of the target Cloud SQL instance that the backup is restored to.)required completerhelpz--backup-instancea}The ID of the instance that the backup was taken from. This argument must be specified when the backup instance is different from the restore instance. If it is not specified, the backup instance is considered the same as the restore instance. This flag is not supported when restore happens from backup name, only supported when restore happens from backup ID in timestamp format.)rrz--backup-projectaThe project of the instance to which the backup belongs. If it isn't specified, backup and restore instances are in the same project. This flag is not supported when restore happens from backup name, only supported when restore happens from backup ID in timestamp format.)rz--clear-disk-encryptionFz'Disables CMEK in the restored instance.)rrN)r AddBackupId add_argumentInstanceCompleterr ASYNC_FLAG AddToParserr)rs rArgszRestoreBackup.Args#s  F ))   ))     $  ! 5  OO'F#rc tjtj}|j}|j}t |dd}|Dcgc] }|t vs |}}tj|j}t||||jj|jdtjj j"j$id} t'j(dsy|rxt+||| } |rMt,j.j1||| } | | j2_t7|| j2_|j:j=| } n|j>s|j|_tA|j} |j:j=|jC| j"| jD|jG|jI| |j>|jJ  } |jjMd | jN| j" }|jPr@|jRjU|jW|j"|jX StRjZj]||dt^j`jcdje| ycc}w)a/Restores a backup of a Cloud SQL instance. Args: args: argparse.Namespace, The arguments that this command was invoked with. Returns: A dict object representing the operations resource describing the restoreBackup operation if the restoreBackup was successful. _specified_argsNrz sql.instances)params collectionz1. All current data on the instance will be lost when the backup is restored to an existing instance. 2. If restoring to a new instance, settings will be applied from the backups unless they are overriden.)r) backupRunId instanceIdr)restoreBackupContextrzsql.operations) operationr)rrzRestoring Cloud SQL instancezRestored [{instance}]. )rS)3r SqlClientAPI_VERSION_DEFAULT sql_clientrgetattrOVERRIDE_FLAGS_SETrIsProjectLevelBackupRequestrrresource_parserParserrVALUEScorer GetOrFailrPromptContinuer command_utilInstancesV1Beta4ConstructCreateInstanceFromArgsrrestoreInstanceSettingsr'restoreInstanceClearOverridesFieldNamesr rrintrrSrRestoreBackupContextrCreatenameasync_rGetSqlOperationsGetRequestrOperationsV1Beta4WaitForOperationrstatuswriteformat)selfrclientrrspecified_args_dictkeyrrrrestore_backup_requestinstance_resourceresult_operation backup_run_id operation_refs rRunzRestoreBackup.Run^s   < < =F""J&&L!$(94@**c5G.G*)DD ,dI>))// :,,1199CCD"0L  $ $ P  7  l    ) ) I Id J    <<T .d 3 <<d$--;;  ! !#44$''lm#--;;  7 7"**#,,<<+7+L+L&3%)%9%9"&"5"5,M,= 8  **11"''$$2&M  {{  " " & &  . .#++%// / 122   11M#ACJJ/66 6MN Ys K+K+N)__name__ __module__ __qualname____doc__ staticmethodrrrrrrs" 8$8$t]rrN)'r __future__rrrgooglecloudsdk.api_lib.sqlrrrr googlecloudsdk.callioper googlecloudsdk.command_lib.kmsr rgooglecloudsdk.command_lib.sqlr r rrgooglecloudsdk.corerrgooglecloudsdk.core.consolerrrrrrrDefaultUniverseOnly ReleaseTracks ReleaseTrackGABETAALPHARestoreCommandrrrrrs1&'/11/(M0DG#*2&*"AHS5l<:4$++00$2C2C2I2IbD''bbr