d dZddlmZddlmZddlmZddlZddlmZddlmZddl m Z dd l m Z dd l mZGd d e jZy) z%Command for spanner restore database.)absolute_import)division)unicode_literalsN)database_operations) databases)base) resource_args)logcpeZdZdZej dej ddZedZdZ y)Restorez!Restore a Cloud Spanner database.zB Restores from a backup to a new Cloud Spanner database.a To restore a backup, run: $ {command} --source-backup=BACKUP_ID --source-instance=SOURCE_INSTANCE --destination-database=DATABASE --destination-instance=INSTANCE_NAME To restore a backup using relative names, run: $ {command} --source-backup=projects/PROJECT_ID/instances/SOURCE_INSTANCE_ID/backups/BACKUP_ID --destination-database=projects/PROJECT_ID/instances/SOURCE_INSTANCE_ID/databases/DATABASE_ID ) DESCRIPTIONEXAMPLESctj|tjj ||j }tj |tj|dy)z Register flags for this command.z%to restore the Cloud Spanner databaseN)r AddRestoreResourceArgsr ASYNC_FLAG AddToParseradd_argument_groupAddRestoreDbEncryptionTypeArgAddKmsKeyResourceArg)parserencryption_group_parsers (lib/surface/spanner/databases/restore.pyArgsz Restore.Args1sW((0OO'$779//0GH&&'>'NPc"|jjj}|jjj}t j |}t j |}tj||||}|jr8tjjdj|jStj |dj|j#|j#S)aThis is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: A message indicating database is restoring or when async, the operation. z/Restore database in progress. Operation name={}z$Restoring backup {0} to database {1})CONCEPTSsourceParse destinationr GetRestoreDbEncryptionTypeGetAndValidateKmsKeyNamerr async_r statusPrintformatnamerAwaitName)selfargs backup_ref database_refencryption_typekms_keyops rRunz Restore.Run;s%%++-J==,,224L#>>tDO44T:G   <_g NB {{ ZZ   ; B B277 KMM  $ $ .55joo6G6B6G6G6I K LLrN) __name__ __module__ __qualname____doc__textwrapdedent detailed_help staticmethodrr0rrr r sT) (//E F (//  - PPLrr )r4 __future__rrrr5googlecloudsdk.api_lib.spannerrrgooglecloudsdk.callioper"googlecloudsdk.command_lib.spannerr googlecloudsdk.corer RestoreCommandr r9rrr@s9,&'>4(<#3Ld!!3Lr