5ndZddlmZddlmZddlmZddlmZdZdZ dd Z d Z d Z dd Z dd Z y)zGUseful commands for interacting with the Cloud Firestore Databases API.)absolute_import)division)unicode_literals) api_utilsc>tjjS)zEReturns the service for interacting with the Firestore admin service.)r GetClientprojects_databases1lib/googlecloudsdk/api_lib/firestore/databases.py_GetDatabaseServicer s     1 11r ctj}tj|j dj ||S)zPerforms a Firestore Admin v1 Database Get. Args: project: the project id to get, a string. database: the database id to get, a string. Returns: a database. projects/{}/databases/{})name)r GetMessagesr Get$FirestoreProjectsDatabasesGetRequestformat)projectdatabasemessagess r GetDatabasersK " " $(   " "33)00(C4 r Nc tj} tj|| jj} t j | jdj||| j||||||| S)aqPerforms a Firestore Admin v1 Database Creation. Args: project: the project id to create, a string. location: the database location to create, a string. database: the database id to create, a string. database_type: the database type, an Enum. database_edition: the database edition, an Enum. delete_protection_state: the value for deleteProtectionState, an Enum. pitr_state: the value for PitrState, an Enum. cmek_config: the CMEK config used to encrypt the database, an object. tags: the tags to attach to the database, a key-value dictionary, or None. Returns: an Operation. projects/{})typedatabaseEdition locationIddeleteProtectionStatepointInTimeRecoveryEnablement cmekConfigtags)parent databaseIdgoogleFirestoreAdminV1Database) rrParseTagsForTagsValueGoogleFirestoreAdminV1Database TagsValuer Create'FirestoreProjectsDatabasesCreateRequestr) rlocationr database_typedatabase_editiondelete_protection_state pitr_state cmek_configr!r tags_values r CreateDatabaser1/s6 " " $(.. H 3 3 = =*   % %66%%g.)1)P)P .!$;,6$*Q* 7  r ctj}tj|j dj |||S)aPerforms a Firestore Admin v1 Database Deletion. Args: project: the project of the database to delete, a string. database: the database id to delete, a string. etag: the current etag of the Database, a string. Returns: an Operation. r)retag)rrr Delete'FirestoreProjectsDatabasesDeleteRequestr)rrr3rs r DeleteDatabaser6_sN " " $(   % %66)00(C7 r c tj}ttj |j dj ||rdndjS)zLists all Firestore databases under the project. Args: project: the project ID to list databases, a string. show_deleted: if true, also returns deleted resources, a boolean. Returns: a List of Databases. rTN)r" showDeleted)rrlistr List%FirestoreProjectsDatabasesListRequestr databases)r show_deletedrs r ListDatabasesr>ss^ " " $(  t  8 8"))'2".$D 9  y  r ctj}tj||jj}|j||||}t j |jdj||S)aRestores a Firestore database from a backup. Args: project: the project ID to list databases, a string. source_backup: the backup to restore from, a string. destination_database: the database to restore to, a string. encryption_config: the encryption config to use for the restored database, an optional object. tags: the tags to attach to the database, a key-value dictionary. Returns: an Operation. )backupr#encryptionConfigr!r)r",googleFirestoreAdminV1RestoreDatabaseRequest) rrr%,GoogleFirestoreAdminV1RestoreDatabaseRequestr'r Restore(FirestoreProjectsDatabasesRestoreRequestr)r source_backupdestination_databaseencryption_configr!rr0restore_requests r RestoreDatabaserJs( " " $(.. H A A K K*II %(  J/   & &77%%g.7F8 r c<tj}tj||jj}|j|j |||||}t j|jdj||S)aClones one Firestore database from another. Args: project: the project ID containing the source database, a string. source_database: the resource name of the database to clone, a string. snapshot_time: the timestamp at which to clone, a DateTime. destination_database: the database to clone to, a string. encryption_config: the encryption config to use for the cloned database, an optional object. tags: the tags to attach to the database, a key-value dictionary, or None. Returns: an Operation. )r snapshotTime) pitrSnapshotr#rAr!r)r"*googleFirestoreAdminV1CloneDatabaseRequest) rrr%*GoogleFirestoreAdminV1CloneDatabaseRequestr'"GoogleFirestoreAdminV1PitrSnapshotr Clone&FirestoreProjectsDatabasesCloneRequestr) rsource_database snapshot_timerGrHr!rr0 clone_requests r CloneDatabaserVs, " " $(.. H ? ? I I*EE>>"$?&( F-   $ $55%%g.5B6 r )N)__doc__ __future__rrr googlecloudsdk.api_lib.firestorerr rr1r6r>rJrVr r r rZsKN&'62 6 -`(8  $Z  )r