HKEYS HVALS and HGETALL implemented

Author: andymccurdy

redis/client.py

@@ -153,6 +153,10 @@ def get_value(value):
             info[key] = get_value(value)
     return info
 
+def pairs_to_dict(response):
+    "Create a dict given a list of key/value pairs"
+    return dict(zip(response[::2], map(float, response[1::2])))
+    
 def zset_score_pairs(response, **options):
     """
     If ``withscores`` is specified in the options, return the response as
@@ -202,6 +206,7 @@ class Redis(threading.local):
         string_keys_to_dict('ZRANGE ZRANGEBYSCORE ZREVRANGE', zset_score_pairs),
         {
             'BGSAVE': lambda r: r == 'Background saving started',
+            'HGETALL': lambda r: r and pairs_to_dict(r) or None,
             'INFO': parse_info,
             'LASTSAVE': timestamp_to_datetime,
             'PING': lambda r: r == 'PONG',
@@ -920,13 +925,24 @@ def hget(self, name, key):
         "Return the value of ``key`` within the hash ``name``"
         return self.format_bulk('HGET', name, key)
 
+    def hgetall(self, name):
+        "Return a Python dict of the hash's name/value pairs"
+        return self.format_inline('HGETALL', name)
+        
+    def hkeys(self, name):
+        "Return the list of keys within hash ``name``"
+        return self.format_inline('HKEYS', name)
+        
     def hset(self, name, key, value):
         """
         Set ``key`` to ``value`` within hash ``name``
         Returns 1 if HSET created a new field, otherwise 0
         """
         return self.format_multi_bulk('HSET', name, key, value)
 
+    def hvals(self, name):
+        "Return the list of values within hash ``name``"
+        return self.format_inline('HVALS', name)
 
 class Pipeline(Redis):
     """

tests/server_commands.py

@@ -728,6 +728,51 @@ def test_hdel(self):
         self.assert_(self.client.hdel('a', 'a2'))
         self.assertEquals(self.client.hget('a', 'a2'), None)
 
+    def test_hgetall(self):
+        # key is not a hash
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.hgetall, 'a')
+        del self.client['a']
+        # no key
+        self.assertEquals(self.client.hgetall('a'), None)
+        # real logic
+        h = {'a1': 1, 'a2': 2, 'a3': 3}
+        self.make_hash('a', h)
+        remote_hash = self.client.hgetall('a')
+        self.assertEquals(h, remote_hash)
+        
+        
+    def test_hkeys(self):
+        # key is not a hash
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.hkeys, 'a')
+        del self.client['a']
+        # no key
+        self.assertEquals(self.client.hkeys('a'), None)
+        # real logic
+        h = {'a1': '1', 'a2': '2', 'a3': '3'}
+        self.make_hash('a', h)
+        keys = h.keys()
+        keys.sort()
+        remote_keys = self.client.hkeys('a')
+        remote_keys.sort()
+        self.assertEquals(keys, remote_keys)
+        
+    def test_hvals(self):
+        # key is not a hash
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.hvals, 'a')
+        del self.client['a']
+        # no key
+        self.assertEquals(self.client.hvals('a'), None)
+        # real logic
+        h = {'a1': '1', 'a2': '2', 'a3': '3'}
+        self.make_hash('a', h)
+        vals = h.values()
+        vals.sort()
+        remote_vals = self.client.hvals('a')
+        remote_vals.sort()
+        self.assertEquals(vals, remote_vals)
 
     # SORT
     def test_sort_bad_key(self):