Skip to content

Commit 0dae919

Browse files
CuriousDimadhermes
authored andcommitted
Add getters for Bigtable Row.row_key and Row.table (#3408)
1 parent 10b50f3 commit 0dae919

File tree

2 files changed

+37
-0
lines changed

2 files changed

+37
-0
lines changed

packages/google-cloud-bigtable/google/cloud/bigtable/row.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,24 @@ def __init__(self, row_key, table):
5555
self._row_key = _to_bytes(row_key)
5656
self._table = table
5757

58+
@property
59+
def row_key(self):
60+
"""Row key.
61+
62+
:rtype: bytes
63+
:returns: The key for the current row.
64+
"""
65+
return self._row_key
66+
67+
@property
68+
def table(self):
69+
"""Row table.
70+
71+
:rtype: table: :class:`Table <google.cloud.bigtable.table.Table>`
72+
:returns: table: The table that owns the row.
73+
"""
74+
return self._table
75+
5876

5977
class _SetDeleteRow(Row):
6078
"""Row helper for setting or deleting cell values.

packages/google-cloud-bigtable/tests/unit/test_row.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,25 @@
1616
import unittest
1717

1818

19+
class TestRow(unittest.TestCase):
20+
21+
@staticmethod
22+
def _get_target_class():
23+
from google.cloud.bigtable.row import Row
24+
return Row
25+
26+
def _make_one(self, *args, **kwargs):
27+
return self._get_target_class()(*args, **kwargs)
28+
29+
def test_row_key_getter(self):
30+
row = self._make_one(row_key=b'row_key', table='table')
31+
self.assertEqual(b'row_key', row.row_key)
32+
33+
def test_row_table_getter(self):
34+
row = self._make_one(row_key=b'row_key', table='table')
35+
self.assertEqual('table', row.table)
36+
37+
1938
class Test_SetDeleteRow(unittest.TestCase):
2039

2140
@staticmethod

0 commit comments

Comments
 (0)