counter_incs by incrementRows

happybase/table.py

@@ -3,11 +3,12 @@
 """
 
 import logging
+from itertools import izip
 from numbers import Integral
 from operator import attrgetter
 from struct import Struct
 
-from .hbase.ttypes import TScan
+from .hbase.ttypes import TScan, TIncrement
 from .util import thrift_type_to_dict, str_increment, OrderedDict
 from .batch import Batch
 
@@ -568,3 +569,24 @@ def counter_dec(self, row, column, value=1):
         :rtype: int
         """
         return self.counter_inc(row, column, -value)
+
+    def counters_inc(self, row, data):
+        """
+
+        This method increments (or decrements) the counter columns in the row
+        specified by `row`. The `data` argument is iteratble data type that
+        contains tuple of column and value, e.g.
+        [(`cf:col`, 1), (`cf:col2`, 2)].
+
+        Note that unlike `counter_inc`, does not return value after
+        incrementing.
+
+        :param str row: the row key
+        :param list data: list of tuple for columns and values
+        """
+        if data is not None and not isinstance(data, (list, tuple, izip)):
+            raise TypeError("'data' must be a iterable data types of tuple")
+
+        self.connection.client.incrementRows(
+            [TIncrement(table=self.name, row=row, column=column, ammount=value)
+             for column, value in data])

tests/test_api.py

@@ -183,6 +183,34 @@ def test_atomic_counters():
     assert_equal(10, table.counter_dec(row, column, -7))
 
 
+def test_multiple_counters():
+    row = 'row-with-counters'
+    columns = ['cf1:counter1', 'cf1:counter2', 'cf1:counter3']
+
+    table.counters_inc(row, zip(columns, [0, 0, 0]))
+    assert_equal([0, 0, 0], [table.counter_get(row, c) for c in columns])
+
+    table.counters_inc(row, zip(columns, [1, 3, 5]))
+    assert_equal([1, 3, 5], [table.counter_get(row, c) for c in columns])
+
+    table.counters_inc(row, zip(columns, [-1, -3, -5]))
+    assert_equal([0, 0, 0], [table.counter_get(row, c) for c in columns])
+
+    for col, delta in zip(columns, [1, 3, 5]):
+        table.counters_inc(row, [(col, delta)])
+        assert_equal(delta, table.counter_get(row, col))
+        table.counter_dec(row, [(col, -delta)])
+        assert_equal(0, table.counter_get(row, col))
+
+    table.counters_inc(row, zip(columns, [1, 3, 5]))
+    table.counters_inc(row, zip(columns, [1, 3, 5]))
+    assert_equal([2, 6, 10], [table.counter_get(row, c) for c in columns])
+
+    table.counters_inc(row, zip(columns, [-1, -3, -5]))
+    table.counters_inc(row, zip(columns, [-1, -3, -5]))
+    assert_equal([0, 0, 0], [table.counter_get(row, c) for c in columns])
+
+
 def test_batch():
     with assert_raises(TypeError):
         table.batch(timestamp='invalid')