feat: implement row and cell model classes

google/cloud/bigtable/__init__.py

@@ -22,9 +22,9 @@
 from google.cloud.bigtable.client import Table
 
 from google.cloud.bigtable.read_rows_query import ReadRowsQuery
+from google.cloud.bigtable.row import Row
+from google.cloud.bigtable.row import Cell
 from google.cloud.bigtable.read_rows_query import RowRange
-from google.cloud.bigtable.row_response import RowResponse
-from google.cloud.bigtable.row_response import CellResponse
 
 from google.cloud.bigtable.mutations_batcher import MutationsBatcher
 from google.cloud.bigtable.mutations import Mutation
@@ -52,6 +52,6 @@
     "DeleteRangeFromColumn",
     "DeleteAllFromFamily",
     "DeleteAllFromRow",
-    "RowResponse",
-    "CellResponse",
+    "Row",
+    "Cell",
 )

google/cloud/bigtable/client.py

@@ -25,7 +25,7 @@
 if TYPE_CHECKING:
     from google.cloud.bigtable.mutations import Mutation, BulkMutationsEntry
     from google.cloud.bigtable.mutations_batcher import MutationsBatcher
-    from google.cloud.bigtable.row_response import RowResponse
+    from google.cloud.bigtable.row import Row
     from google.cloud.bigtable.read_rows_query import ReadRowsQuery
     from google.cloud.bigtable import RowKeySamples
     from google.cloud.bigtable.row_filters import RowFilter
@@ -109,7 +109,7 @@ async def read_rows_stream(
         idle_timeout: int | float | None = 300,
         per_request_timeout: int | float | None = None,
         metadata: list[tuple[str, str]] | None = None,
-    ) -> AsyncIterable[RowResponse]:
+    ) -> AsyncIterable[Row]:
         """
         Returns a generator to asynchronously stream back row data.
 
@@ -166,7 +166,7 @@ async def read_rows(
         per_row_timeout: int | float | None = 10,
         per_request_timeout: int | float | None = None,
         metadata: list[tuple[str, str]] | None = None,
-    ) -> list[RowResponse]:
+    ) -> list[Row]:
         """
         Helper function that returns a full list instead of a generator
 
@@ -184,7 +184,7 @@ async def read_row(
         operation_timeout: int | float | None = 60,
         per_request_timeout: int | float | None = None,
         metadata: list[tuple[str, str]] | None = None,
-    ) -> RowResponse:
+    ) -> Row:
         """
         Helper function to return a single row
 
@@ -206,7 +206,7 @@ async def read_rows_sharded(
         idle_timeout: int | float | None = 300,
         per_request_timeout: int | float | None = None,
         metadata: list[tuple[str, str]] | None = None,
-    ) -> AsyncIterable[RowResponse]:
+    ) -> AsyncIterable[Row]:
         """
         Runs a sharded query in parallel
 
@@ -410,7 +410,7 @@ async def read_modify_write_row(
         *,
         operation_timeout: int | float | None = 60,
         metadata: list[tuple[str, str]] | None = None,
-    ) -> RowResponse:
+    ) -> Row:
         """
         Reads and modifies a row atomically according to input ReadModifyWriteRules,
         and returns the contents of all modified cells
@@ -429,7 +429,7 @@ async def read_modify_write_row(
                 Failed requests will not be retried.
             - metadata: Strings which should be sent along with the request as metadata headers.
         Returns:
-            - RowResponse: containing cell data that was modified as part of the
+            - Row: containing cell data that was modified as part of the
                 operation
         Raises:
             - GoogleAPIError exceptions from grpc call

google/cloud/bigtable/mutations.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from google.cloud.bigtable.row_response import family_id, qualifier, row_key
+from google.cloud.bigtable.row import family_id, qualifier, row_key
 
 
 class Mutation:

google/cloud/bigtable/mutations_batcher.py

@@ -18,7 +18,7 @@
 from typing import TYPE_CHECKING
 
 from google.cloud.bigtable.mutations import Mutation
-from google.cloud.bigtable.row_response import row_key
+from google.cloud.bigtable.row import row_key
 from google.cloud.bigtable.row_filters import RowFilter
 
 if TYPE_CHECKING:

google/cloud/bigtable/read_modify_write_rules.py

@@ -16,7 +16,7 @@
 
 from dataclasses import dataclass
 
-from google.cloud.bigtable.row_response import family_id, qualifier
+from google.cloud.bigtable.row import family_id, qualifier
 
 
 class ReadModifyWriteRule:

google/cloud/bigtable/read_rows_query.py

@@ -14,7 +14,7 @@
 #
 from __future__ import annotations
 from typing import TYPE_CHECKING, Any
-from .row_response import row_key
+from .row import row_key
 from dataclasses import dataclass
 from google.cloud.bigtable.row_filters import RowFilter