feat: Expose tags field in Database and RestoreDatabaseRequest public protos (#1074)

Author: gcf-owl-bot[bot]

google/cloud/firestore_admin_v1/types/database.py

@@ -119,6 +119,13 @@ class Database(proto.Message):
         source_info (google.cloud.firestore_admin_v1.types.Database.SourceInfo):
             Output only. Information about the provenance
             of this database.
+        tags (MutableMapping[str, str]):
+            Optional. Input only. Immutable. Tag
+            keys/values directly bound to this resource. For
+            example:
+
+              "123/environment": "production",
+              "123/costCenter": "marketing".
         free_tier (bool):
             Output only. Background: Free tier is the
             ability of a Firestore database to use a small
@@ -524,6 +531,11 @@ class CustomerManagedEncryptionOptions(proto.Message):
         number=26,
         message=SourceInfo,
     )
+    tags: MutableMapping[str, str] = proto.MapField(
+        proto.STRING,
+        proto.STRING,
+        number=29,
+    )
     free_tier: bool = proto.Field(
         proto.BOOL,
         number=30,

google/cloud/firestore_admin_v1/types/firestore_admin.py

@@ -1047,6 +1047,12 @@ class RestoreDatabaseRequest(proto.Message):
             If this field is not specified, the restored database will
             use the same encryption configuration as the backup, namely
             [use_source_encryption][google.firestore.admin.v1.Database.EncryptionConfig.use_source_encryption].
+        tags (MutableMapping[str, str]):
+            Optional. Immutable. Tags to be bound to the restored
+            database.
+
+            The tags should be provided in the format of
+            ``tagKeys/{tag_key_id} -> tagValues/{tag_value_id}``.
     """
 
     parent: str = proto.Field(
@@ -1066,6 +1072,11 @@ class RestoreDatabaseRequest(proto.Message):
         number=9,
         message=gfa_database.Database.EncryptionConfig,
     )
+    tags: MutableMapping[str, str] = proto.MapField(
+        proto.STRING,
+        proto.STRING,
+        number=10,
+    )
 
 
 __all__ = tuple(sorted(__protobuf__.manifest))

scripts/fixup_firestore_admin_v1_keywords.py

@@ -72,7 +72,7 @@ class firestore_adminCallTransformer(cst.CSTTransformer):
         'list_indexes': ('parent', 'filter', 'page_size', 'page_token', ),
         'list_user_creds': ('parent', ),
         'reset_user_password': ('name', ),
-        'restore_database': ('parent', 'database_id', 'backup', 'encryption_config', ),
+        'restore_database': ('parent', 'database_id', 'backup', 'encryption_config', 'tags', ),
         'update_backup_schedule': ('backup_schedule', 'update_mask', ),
         'update_database': ('database', 'update_mask', ),
         'update_field': ('field', 'update_mask', ),

tests/unit/gapic/firestore_admin_v1/test_firestore_admin.py

@@ -20792,6 +20792,7 @@ def test_create_database_rest_call_success(request_type):
             "backup": {"backup": "backup_value"},
             "operation": "operation_value",
         },
+        "tags": {},
         "free_tier": True,
         "etag": "etag_value",
         "database_edition": 1,
@@ -21303,6 +21304,7 @@ def test_update_database_rest_call_success(request_type):
             "backup": {"backup": "backup_value"},
             "operation": "operation_value",
         },
+        "tags": {},
         "free_tier": True,
         "etag": "etag_value",
         "database_edition": 1,