Merge pull request #114 from bbonenfant/implement-hash-setnx

Setnx method on the Hash object.
coleifer · Jan 8, 2020 · 870cb4a · 870cb4a
2 parents b4ab286 + 6fdffc4
commit 870cb4a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/walrus/containers.py b/walrus/containers.py
@@ -182,6 +182,14 @@ def items(self, lazy=False):
         else:
             return list(self)
 
+    def setnx(self, key, value):
+        """
+        Set ``key`` to ``value`` if ``key`` does not exist.
+
+        :returns: True if successfully set or False if the key already existed.
+        """
+        return bool(self.database.hsetnx(self.key, key, value))
+
     @chainable_method
     def update(self, *args, **kwargs):
         """

diff --git a/walrus/tests/containers.py b/walrus/tests/containers.py
@@ -67,6 +67,11 @@ def test_from_dict(self):
         hsh = Hash.from_dict(db, 'test', data)
         self.assertEqual(hsh.as_dict(True), data)
 
+    def test_setnx(self):
+        key, value = "key_setnx", "value_setnx"
+        self.assertTrue(self.hsh.setnx(key, value))
+        self.assertFalse(self.hsh.setnx(key, value))
+
 
 class TestSet(WalrusTestCase):
     def setUp(self):