Add support for CLIENT SETINFO

redis/_parsers/helpers.py

@@ -629,7 +629,7 @@ def parse_client_info(value):
     "key1=value1 key2=value2 key3=value3"
     """
     client_info = {}
-    infos = str_if_bytes(value).split(" ")
+    infos = str_if_bytes(value[:-1]).split(" ")
     for info in infos:
         key, value = info.split("=")
         client_info[key] = value
@@ -700,6 +700,7 @@ def string_keys_to_dict(key_string, callback):
     "CLIENT KILL": parse_client_kill,
     "CLIENT LIST": parse_client_list,
     "CLIENT PAUSE": bool_ok,
+    "CLIENT SETINFO": bool_ok,
     "CLIENT SETNAME": bool_ok,
     "CLIENT UNBLOCK": bool,
     "CLUSTER ADDSLOTS": bool_ok,

redis/asyncio/connection.py

@@ -49,7 +49,7 @@
     TimeoutError,
 )
 from redis.typing import EncodableT
-from redis.utils import HIREDIS_AVAILABLE, str_if_bytes
+from redis.utils import HIREDIS_AVAILABLE, get_lib_version, str_if_bytes
 
 from .._parsers import (
     BaseParser,
@@ -347,6 +347,15 @@ async def on_connect(self) -> None:
             if str_if_bytes(await self.read_response()) != "OK":
                 raise ConnectionError("Error setting client name")
 
+        try:
+            # set the library name and version
+            await self.send_command("CLIENT", "SETINFO", "LIB-NAME", "redis-py")
+            await self.read_response()
+            await self.send_command("CLIENT", "SETINFO", "LIB-VER", get_lib_version())
+            await self.read_response()
+        except ResponseError:
+            pass
+
         # if a database is specified, switch to it
         if self.db:
             await self.send_command("SELECT", self.db)

redis/cluster.py

@@ -223,6 +223,7 @@ class AbstractRedisCluster:
                 "ACL WHOAMI",
                 "AUTH",
                 "CLIENT LIST",
+                "CLIENT SETINFO",
                 "CLIENT SETNAME",
                 "CLIENT GETNAME",
                 "CONFIG SET",

redis/commands/core.py

@@ -706,6 +706,13 @@ def client_setname(self, name: str, **kwargs) -> ResponseT:
         """
         return self.execute_command("CLIENT SETNAME", name, **kwargs)
 
+    def client_setinfo(self, attr: str, value: str, **kwargs) -> ResponseT:
+        """
+        Sets the current connection library name or version
+        For mor information see https://redis.io/commands/client-setinfo
+        """
+        return self.execute_command("CLIENT SETINFO", attr, value, **kwargs)
+
     def client_unblock(
         self, client_id: int, error: bool = False, **kwargs
     ) -> ResponseT:

redis/connection.py

@@ -31,6 +31,7 @@
     HIREDIS_AVAILABLE,
     HIREDIS_PACK_AVAILABLE,
     SSL_AVAILABLE,
+    get_lib_version,
     str_if_bytes,
 )
 
@@ -360,6 +361,15 @@ def on_connect(self):
             if str_if_bytes(self.read_response()) != "OK":
                 raise ConnectionError("Error setting client name")
 
+        try:
+            # set the library name and version
+            self.send_command("CLIENT", "SETINFO", "LIB-NAME", "redis-py")
+            self.read_response()
+            self.send_command("CLIENT", "SETINFO", "LIB-VER", get_lib_version())
+            self.read_response()
+        except ResponseError:
+            pass
+
         # if a database is specified, switch to it
         if self.db:
             self.send_command("SELECT", self.db)

redis/utils.py

@@ -1,4 +1,5 @@
 import logging
+import sys
 from contextlib import contextmanager
 from functools import wraps
 from typing import Any, Dict, Mapping, Union
@@ -27,6 +28,11 @@
 except ImportError:
     CRYPTOGRAPHY_AVAILABLE = False
 
+if sys.version_info >= (3, 8):
+    from importlib import metadata
+else:
+    import importlib_metadata as metadata
+
 
 def from_url(url, **kwargs):
     """
@@ -131,3 +137,11 @@ def _set_info_logger():
         handler = logging.StreamHandler()
         handler.setLevel(logging.INFO)
         logger.addHandler(handler)
+
+
+def get_lib_version():
+    try:
+        libver = metadata.version("redis")
+    except metadata.PackageNotFoundError:
+        libver = "99.99.99"
+    return libver

tests/test_asyncio/test_commands.py

@@ -355,6 +355,14 @@ async def test_client_setname(self, r: redis.Redis):
             r, await r.client_getname(), "redis_py_test", b"redis_py_test"
         )
 
+    @skip_if_server_version_lt("7.2.0")
+    async def test_client_setinfo(self, r: redis.Redis):
+        assert await r.client_setinfo("lib-name", "test")
+        assert await r.client_setinfo("lib-ver", "123")
+        info = await r.client_info()
+        assert info["lib-name"] == "test"
+        assert info["lib-ver"] == "123"
+
     @skip_if_server_version_lt("2.6.9")
     @pytest.mark.onlynoncluster
     async def test_client_kill(self, r: redis.Redis, r2):

tests/test_commands.py

@@ -554,6 +554,14 @@ def test_client_setname(self, r):
         assert r.client_setname("redis_py_test")
         assert_resp_response(r, r.client_getname(), "redis_py_test", b"redis_py_test")
 
+    @skip_if_server_version_lt("7.2.0")
+    def test_client_setinfo(self, r: redis.Redis):
+        assert r.client_setinfo("lib-name", "test")
+        assert r.client_setinfo("lib-ver", "123")
+        info = r.client_info()
+        assert info["lib-name"] == "test"
+        assert info["lib-ver"] == "123"
+
     @pytest.mark.onlynoncluster
     @skip_if_server_version_lt("2.6.9")
     def test_client_kill(self, r, r2):