U dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd lmZdd lmZdd lmZej&j(j*Zej&j.j0Ze j4d ej6ddj8j:ddddddZe j4d ej6ddj>j@ddddddZ!e j4d ej6ddjDj@ddddddZ#e j4d ej6ddjHj@ddddddZ%e j4dej6ddjLjNdddd d!Z(e j4d"ej6ddjLjRd#dd$d%d&d'(Z*e j4d)ej6ddjVjXd*dd+d,d-Z-d.Z.d/Z/d0Z0d1Z1d2Z2d3Z3d4Z4d5Z5d6Z6d7Z7d8Z8d9Z9d:Z:d;Z;d<ZZ>d[d?Z?d[d@Z@d[dAZAd\dBZBd[dCZCd[dDZDd[dEZEdFZFdGZGdHZHdIZIdJZJdKZKdLZLdMZMdNZNdOZOGdPdQZPdRePfdSZQdTZRdUZSdVZTdWZUdXZVdYZWyZ)]z1Shared resource flags for Cloud Spanner commands.)absolute_import)division)unicode_literals)apis) exceptions)concepts)deps) arg_utils)concept_parsers)presentation_specs) propertiesz--encryption-typespannerv1z"The encryption type of the backup.F)zuse-database-encryptionz6Use the same encryption configuration as the database.)zgoogle-default-encryptionzUse Google default encryption.)customer-managed-encryptionz^Use the provided Cloud KMS key for encryption.If this option is selected, kms-key must be set.)USE_DATABASE_ENCRYPTIONGOOGLE_DEFAULT_ENCRYPTIONCUSTOMER_MANAGED_ENCRYPTION)help_strrequiredcustom_mappings)rz_Use the provided Cloud KMS key for encryption. If this option is selected, kms-key must be set.z)The encryption type of the copied backup.)'use-config-default-or-backup-encryptionz}Use the default encryption configuration if one exists. otherwise use the same encryption configuration as the source backup.)'USE_CONFIG_DEFAULT_OR_BACKUP_ENCRYPTIONrrz-The encryption type of the restored database.)rzvUse the default encryption configuration if one exists, otherwise use the same encryption configuration as the backup.z--instance-typez%Specifies the type for this instance.) provisionedzSProvisioned instances have dedicated resources, standard usage limits, and support.)z free-instancezFree trial instances provide no guarantees for dedicated resources, both node_count and processing_units should be 0. They come with stricter usage limits and limited support.) PROVISIONED FREE_INSTANCEz--default-storage-typez5Specifies the default storage type for this instance.T)ssdz1Use ssd as default storage type for this instance)hddz1Use hdd as default storage type for this instance)SSDHDD)rrhiddenrz--expire-behaviorz-The expire behavior of a free trial instance.)zfree-to-provisionedzUWhen the free trial instance expires, upgrade the instance to a provisioned instance.)zremove-after-grace-periodzWhen the free trial instance expires, disable the instance, and delete it after the grace period passes if it has not been upgraded to a provisioned instance.)FREE_TO_PROVISIONEDREMOVE_AFTER_GRACE_PERIODcbtjddtjtgS)z3Get instance resource attribute with default value.instancez.The Cloud Spanner instance for the {resource}.)name help_text fallthroughs)r ResourceParameterAttributeConfigr PropertyFallthrough _INSTANCE7lib/googlecloudsdk/command_lib/spanner/resource_args.pyInstanceAttributeConfigr.s,  2 2 @,,Y78 ::r,c0tjddS)z=Get instance partition resource attribute with default value.instance partitionz2The Spanner instance partition for the {resource}.r%r&rr(r+r,r- InstancePartitionAttributeConfigr3s  2 2 D r,c0tjddS)z Get database resource attribute.databasez.The Cloud Spanner database for the {resource}.r1r2r+r,r-DatabaseAttributeConfigr6s  2 2 @ BBr,c0tjddS)zGet backup resource attribute.backupz,The Cloud Spanner backup for the {resource}.r1r2r+r,r-BackupAttributeConfigr9s  2 2 > @@r,c0tjddS)z'Get backup schedule resource attribute.backup-schedulez5The Cloud Spanner backup schedule for the {resource}.r1r2r+r,r-BackupScheduleAttributeConfigr<s  2 2 G IIr,c0tjddS)zGet session resource attribute.sessionz-The Cloud Spanner session for the {resource}.r1r2r+r,r-SessionAttributeConfigr?s  2 2  O QQr,c.tjdS)Nkms-key)r%r2r+r,r-KmsKeyAttributeConfigrBs  2 2 BBr,c0tjddS)N kms-keyringz!KMS keyring id of the {resource}.r1r2r+r,r-KmsKeyringAttributeConfigrEs  2 2 $G IIr,c0tjddS)N kms-locationz"Cloud location for the {resource}.r1r2r+r,r-KmsLocationAttributeConfigrHs  2 2 %I KKr,c0tjddS)N kms-projectz$Cloud project id for the {resource}.r1r2r+r,r-KmsProjectAttributeConfigrKs  2 2 $J LLr,c`tjddttjS)Nzspanner.projects.instancesr$) resource_name instancesId projectsId)r ResourceSpecr. DEFAULT_PROJECT_ATTRIBUTE_CONFIGr+r,r-GetInstanceResourceSpecrR s*   ")+::  <)rM sessionsIdrWrNrO)rrPr?r6r.rQr+r,r-GetSessionResourceSpecrgBs6   5'))+)+::  <z --sessionzThe Cloud Spanner session {}.TriN)r rjrkrgrlrmrns r-AddSessionResourceArgrsL!k$++ %,,T2 ,![(r,c|rdnd}tjj|tdj |dj |y)abAdd a resource argument for a Cloud Spanner backup. NOTE: Must be used only if it's the only resource arg in the command. Args: parser: the argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. positional: bool, if True, means that the backup ID is a positional rather than a flag. r8z--backupzThe Cloud Spanner backup {}.TriN)r rjrkrarlrmrns r-AddBackupResourceArgrsL Z$++ $++D1 ,![(r,c|rdnd}tjj|tdj |dj |y)auAdd a resource argument for a Cloud Spanner backup schedule. NOTE: Must be used only if it's the only resource arg in the command. Args: parser: the argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. positional: bool, if True, means that the backup schedules ID is a positional rather than a flag. backup_schedulez--backup-schedulez%The Cloud Spanner backup schedule {}.TriN)r rjrkrdrlrmrns r-AddBackupScheduleResourceArgrsN) .A$++ #%-44T: ,![(r,c@tjj|SN)%_CREATE_BACKUP_ENCRYPTION_TYPE_MAPPER choice_argrmros r- AddCreateBackupEncryptionTypeArgrs . 9 9 E Ef MMr,c@tj|jSr)rGetEnumForChoiceencryption_typeargss r-GetCreateBackupEncryptionTypers . ? ?  r,c@tjj|Sr),_CREATE_BACKUP_ENCRYPTION_CONFIG_TYPE_MAPPERrrmrs r-&AddCreateBackupEncryptionConfigTypeArgrs 5 @ @ L L  r,c@tj|jSr)rrrrs r-#GetCreateBackupEncryptionConfigTypers 5 F F  r,c tjdtdddddtjdtddd d dg}tj|j |y ) z@Add backup resource args (source, destination) for copy command.--sourceTEXTT--source-instance--source-backupr$r8rr --destination--destination-instancez--destination-backupN)r rrar rjrmro arg_specss r-AddCopyBackupResourceArgsrsx11   ! -)  11   ! 2.  )* *66v>r,c@tjj|Sr)#_COPY_BACKUP_ENCRYPTION_TYPE_MAPPERrrmrs r-AddCopyBackupEncryptionTypeArgrs , 7 7 C CF KKr,c@tj|jSr)rrrrs r-GetCopyBackupEncryptionTypers , = =  r,c tjdtdddddtjdtddd d d g}t j |j |y ) zCAdd backup resource args (source, destination) for restore command.rrTrrrrrrz--destination-database)r$r5N)r rrarXr rjrmrs r-AddRestoreResourceArgsrsx11   ! -)  11  ! # 22  )* *66v>r,c@tjj|Sr)"_RESTORE_DB_ENCRYPTION_TYPE_MAPPERrrmrs r-AddRestoreDbEncryptionTypeArgr8s + 6 6 B B6 JJr,c@tj|jSr)rrrrs r-GetRestoreDbEncryptionTyper<s + < <  r,ceZdZdZddZy)CloudKmsKeyNamezCloudKmsKeyName to encapsulate `kmsKeyName` and `kmsKeyNames` fields. Single `kmsKeyName` and repeated `kmsKeyNames` fields are extracted from user input, which are later used in `EncryptionConfig` to pass to Spanner backend. Nc4||_|g|_y||_yr)rr)selfrrs r-__init__zCloudKmsKeyName.__init__Hs $Dd(dr,)NN)__name__ __module__ __qualname____doc__rr+r,r-rrAs  )r,rreturnc|jjj}|jjj}t }|r|j |_|S|r%|Dcgc]}|j c}|_|SdD]5}t||jddds!tjddycc}w)zParse the KMS key resource arg, make sure the key format is correct. Args: args: calliope framework gcloud args Returns: CloudKmsKeyName: if CMEK. None: if non-CMEK. )rArDrGrJrz-_NzB--kms-project --kms-location --kms-keyring --kms-key or --kms-keyszFor a single KMS key, specify fully qualified KMS key ID with --kms-key, or use combination of --kms-project, --kms-location, --kms-keyring and --kms-key to specify the key ID in pieces. Or specify fully qualified KMS key ID with --kms-keys.) CONCEPTSkms_keyParsekms_keysr RelativeNamerrgetattrreplacerInvalidArgumentException)rrrcloud_kms_key_namekeywords r-GetAndValidateKmsKeyNamerPs&&,,.,--((..0-&(&2&?&?&A#2 18E(8E !!# ($. % wsC0$ 711  5  " -(s2Cc@tjj|Sr)_INSTANCE_TYPE_MAPPERrrmrs r-AddInstanceTypeArgrzs  ) ) 5 5f ==r,c@tj|jSr)rr instance_typers r-GetInstanceTyper~s  / /0B0B CCr,c@tjj|Sr)_DEFAULT_STORAGE_TYPE_MAPPERrrmrs r-AddDefaultStorageTypeArgrs % 0 0 < rs~8&',.51:DG*    ! ! ) )    % % . . (B (B(BD4223`3` 1 $ & ( )%20Jy/I/ID4""#@#@ 1 $ & ( 0,4'Ai&@&@D4  !>!> 8 4 & ( '#<&@Y%?%?D4%%&C&C < 4 & ( &":3 22D4h** 4   4 :y99D4h00 D  KK 5)44D4--.K.K <  ( 6  &:B@IQ C I K L <<.<<<)&()&"J)&)()(N   ?4L ?4K ) )'o'T>DE @Hr,