feat: Expose incr/decr default and time protocol arguments in client class (#243)

The protocol implements `default` and `time` arguments, but those
are not exposed in the client class implementation.

This adds those fields.

Author: bisho

bmemcached/client/distributed.py

@@ -204,21 +204,25 @@ def cas(self, key, value, cas, time=0, compress_level=-1):
         server = self._get_server(key)
         return server.cas(key, value, cas, time, compress_level)
 
-    def incr(self, key, value):
+    def incr(self, key, value, default=0, time=1000000):
         """
         Increment a key, if it exists, returns it's actual value, if it don't, return 0.
 
         :param key: Key's name
         :type key: six.string_types
         :param value: Number to be incremented
         :type value: int
+        :param default: If key not set, initialize to this value
+        :type default: int
+        :param time: Time in seconds that your key will expire.
+        :type time: int
         :return: Actual value of the key on server
         :rtype: int
         """
         server = self._get_server(key)
-        return server.incr(key, value)
+        return server.incr(key, value, default=default, time=time)
 
-    def decr(self, key, value):
+    def decr(self, key, value, default=0, time=1000000):
         """
         Decrement a key, if it exists, returns it's actual value, if it don't, return 0.
         Minimum value of decrement return is 0.
@@ -227,8 +231,12 @@ def decr(self, key, value):
         :type key: six.string_types
         :param value: Number to be decremented
         :type value: int
+        :param default: If key not set, initialize to this value
+        :type default: int
+        :param time: Time in seconds that your key will expire.
+        :type time: int
         :return: Actual value of the key on server
         :rtype: int
         """
         server = self._get_server(key)
-        return server.decr(key, value)
+        return server.decr(key, value, default=default, time=time)

bmemcached/client/replicating.py

@@ -232,24 +232,28 @@ def delete_multi(self, keys):
 
         return all(returns)
 
-    def incr(self, key, value):
+    def incr(self, key, value, default=0, time=1000000):
         """
         Increment a key, if it exists, returns it's actual value, if it don't, return 0.
 
         :param key: Key's name
         :type key: six.string_types
         :param value: Number to be incremented
         :type value: int
+        :param default: If key not set, initialize to this value
+        :type default: int
+        :param time: Time in seconds that your key will expire.
+        :type time: int
         :return: Actual value of the key on server
         :rtype: int
         """
         returns = []
         for server in self.servers:
-            returns.append(server.incr(key, value))
+            returns.append(server.incr(key, value, default=default, time=time))
 
         return returns[0]
 
-    def decr(self, key, value):
+    def decr(self, key, value, default=0, time=1000000):
         """
         Decrement a key, if it exists, returns it's actual value, if it don't, return 0.
         Minimum value of decrement return is 0.
@@ -258,11 +262,15 @@ def decr(self, key, value):
         :type key: six.string_types
         :param value: Number to be decremented
         :type value: int
+        :param default: If key not set, initialize to this value
+        :type default: int
+        :param time: Time in seconds that your key will expire.
+        :type time: int
         :return: Actual value of the key on server
         :rtype: int
         """
         returns = []
         for server in self.servers:
-            returns.append(server.decr(key, value))
+            returns.append(server.decr(key, value, default=default, time=time))
 
         return returns[0]

test/test_simple_functions.py

@@ -208,10 +208,18 @@ def testIncrement(self):
         self.assertEqual(0, self.client.incr('test_key', 1))
         self.assertEqual(1, self.client.incr('test_key', 1))
 
+    def testIncrementInitialize(self):
+        self.assertEqual(10, self.client.incr('test_key', 1, default=10))
+        self.assertEqual(11, self.client.incr('test_key', 1, default=10))
+
     def testDecrement(self):
         self.assertEqual(0, self.client.decr('test_key', 1))
         self.assertEqual(0, self.client.decr('test_key', 1))
 
+    def testDecrementInitialize(self):
+        self.assertEqual(10, self.client.decr('test_key', 1, default=10))
+        self.assertEqual(9, self.client.decr('test_key', 1, default=10))
+
     def testFlush(self):
         self.client.set('test_key', 'test')
         self.assertTrue(self.client.flush_all())