Adding support for CLIENT LIST with ID

Part of redis#1434

Author: chayim

Diff for: redis/client.py

@@ -1243,21 +1243,26 @@ def client_kill_filter(self, _id=None, _type=None, addr=None, skipme=None):
                             "<value> must specify at least one filter")
         return self.execute_command('CLIENT KILL', *args)
 
-    def client_list(self, _type=None):
+    def client_list(self, _type=None, client_id=None):
         """
         Returns a list of currently connected clients.
         If type of client specified, only that type will be returned.
         :param _type: optional. one of the client types (normal, master,
          replica, pubsub)
         """
         "Returns a list of currently connected clients"
+        args = []
         if _type is not None:
             client_types = ('normal', 'master', 'replica', 'pubsub')
             if str(_type).lower() not in client_types:
                 raise DataError("CLIENT LIST _type must be one of %r" % (
                                 client_types,))
-            return self.execute_command('CLIENT LIST', b'TYPE', _type)
-        return self.execute_command('CLIENT LIST')
+            args.append(b'TYPE')
+            args.append(_type)
+        if client_id is not None:
+            args.append(b"ID")
+            args.append(client_id)
+        return self.execute_command('CLIENT LIST', *args)
 
     def client_getname(self):
         "Returns the current connection name"

Diff for: tests/test_commands.py

@@ -290,6 +290,14 @@ def test_client_list_type(self, r):
             clients = r.client_list(_type=client_type)
             assert isinstance(clients, list)
 
+    @skip_if_server_version_lt('6.2.0')
+    def test_client_list_client_id(self, r):
+        clients = r.client_list()
+        client_id = clients[0]['id']
+        clients = r.client_list(client_id=client_id)
+        assert len(clients) == 1
+        assert 'addr' in clients[0]
+
     @skip_if_server_version_lt('5.0.0')
     def test_client_id(self, r):
         assert r.client_id() > 0