V#dZddlmZddlmZddlmZddlmZddlmZ ddl m Z ddl mZ ddlmZdd lmZdd lZdd lmZdd lmZGd de j.ZdZdZdZ dde fdZ ddZdZdZdZ dZ!dZ"dZ#dZ$y )z!Cloud Spanner backups API helper.)absolute_import)division)unicode_literals)apis) exceptions)CloudKmsKeyName)http)timesN) http_client)urllibceZdZdZy)HttpRequestFailedErrorz3Indicates that the http request failed in some way.N)__name__ __module__ __qualname____doc__-lib/googlecloudsdk/api_lib/spanner/backups.pyrr s;rrctj|jtjtj}tj |dtj}|S)z5Parse flag value into Datetime format for expireTime.z%Y-%m-%dT%H:%M:%S.%6f%Ez)tzinfo)r ParseDurationGetRelativeDateTimeNowUTCFormatDateTime)expiration_valuedatetimeparsed_datetimes rParseExpireTimer &sW *+?? ))EII   ((*599>/ rc*|jdr|jds"|jds'|jdstjdd|jr |jS|jrt |jSy)zGCheck if fields for expireTime are correctly specified and parse value.expiration_dateretention_periodz'--expiration-date or --retention-periodz'>v'F,3,H(M h  GJJ. !B!."/ 00{+,  " "< ??rctjdd}tjdd}|j|j |j }t ||_|r.|j||j|j|_ n|r|j||_ |j|jj |}|jj|S)zCopy a backup.r*r+)r< sourceBackup)encryptionType kmsKeyName kmsKeyNames)rg)r`copyBackupRequest)rr-r.CopyBackupRequestNamer0r(r@CopyBackupEncryptionConfigrJrKencryptionConfig*SpannerProjectsInstancesBackupsCopyRequestrLr1Copy) source_backup_refdestination_backup_refr'r^r8r4r5copy_backup_requestr6s r CopyBackuprtzs  ! !)T 2&   4 0$..%**,$113/5$9#>  +/+J+J&'')),K,( +/+J+J&,K,( 77 # * * , 9 9 ;+ 8 -#  * * / / 44rc|jjdz|jz|j_t ||j_d|_|S)z4Parse arguments and construct update backup request.z /backups/ expire_time)rLr0rIr,r(r@ updateMask)r3r'r6s rModifyUpdateMetadataRequestrxsR%%'44KK #**//5#** #. *rc|j|_|jr6|jdz|jz}dj||_|S)z3Parse arguments and construct list backups request.r>z database="{}")r0r`r?rMfilter) instance_refr'r6r?s rModifyListRequestr|sL((*#* ]]((*]:T]]JH ''1CJ *rct||S)z0Checks if backup exists, if so, returns request.)r7)r3_r6s rCheckBackupExistsrs J *rc>|Dcgc] }t|c}Scc}w)zFormats existing fields for displaying them in the list response. Args: backup_refs: A list of backups. Returns: The list of backups with the new formatting. ) _FormatBackup) backup_refsr~r3s rFormatListBackupsrs"7B Bk - #k BB Bsc|}|jDcgc] }t|c}|_|jDcgc]}t|jc}|_|Scc}wcc}w)zFormats a single backup for displaying it in the list response. This function makes in-place modifications. Args: backup_ref: The backup to format. Returns: The backup with the new formatting. )backupSchedules#_ExtractScheduleNameFromScheduleUriinstancePartitions5_ExtractInstancePartitionNameFromInstancePartitionUriinstancePartition)r3formatted_backup_ref schedule_uriinstance_partitions rrrs$%44*4,*,74*&!+ = = -!> <  . .!> -) *-s AA$c*|jddS)a9Converts a schedule URI to an schedule name. Args: schedule_uri: The URI of the schedule, e.g., "projects/test-project/instances/test-instance/databases/test-database/backupSchedules/test-backup-schedule". Returns: The name of the schedule ("test-backup-schedule" in the example above). /split)rs rrrs   C  $$rc.d|jddiS)adConverts an instance partition URI to an instance partition name. Args: instance_partition_uri: The URI of an instance partition, e.g., "projects/test-project/instances/test-instance/instancePartitions/test-instance-partition". Returns: The name of the instance partition ("test-instance-partition" in the example above). rrrr)instance_partition_uris rrrs  5;;C@D EEr)NN)%r __future__rrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioperr%0googlecloudsdk.command_lib.spanner.resource_argsrgooglecloudsdk.corecore_exceptionsgooglecloudsdk.core.credentialsr googlecloudsdk.core.utilr rS six.movesr rZr Errorrr r(r7rdrtrxr|rrrrrrrrrs(&',>L=0* ,_22 2"4HL,@5D,@d $ 5:     C2 % Fr