Cluster schema support

redis/client.py

@@ -105,15 +105,16 @@ class Redis(RedisModuleCommands, CoreCommands, SentinelCommands):
     """
 
     @classmethod
-    def from_url(cls, url: str, **kwargs) -> "Redis":
+    def from_url(cls, url: str, **kwargs) -> Union["Redis", "RedisCluster"]:
         """
-        Return a Redis client object configured from the given URL
+        Return a Redis or RedisCluster client object configured from the given URL
 
         For example::
 
             redis://[[username]:[password]]@localhost:6379/0
             rediss://[[username]:[password]]@localhost:6379/0
             unix://[username@]/path/to/socket.sock?db=0[&password=password]
+            rediscluster://[[username]:[password]]@localhost:6379/0
 
         Three URL schemes are supported:
 
@@ -146,6 +147,9 @@ class initializer. In the case of conflicting arguments, querystring
         arguments always win.
 
         """
+        if url.startswith("rediscluster"):
+            from .cluster import RedisCluster
+            return RedisCluster.from_url(url, **kwargs)
         single_connection_client = kwargs.pop("single_connection_client", False)
         connection_pool = ConnectionPool.from_url(url, **kwargs)
         client = cls(

redis/connection.py

@@ -957,6 +957,8 @@ def parse_url(url):
     if not (
         url.startswith("redis://")
         or url.startswith("rediss://")
+        or url.startswith("rediscluster://")
+        or url.startswith("redisclusters://")
         or url.startswith("unix://")
     ):
         raise ValueError(
@@ -1004,7 +1006,7 @@ def parse_url(url):
             except (AttributeError, ValueError):
                 pass
 
-        if url.scheme == "rediss":
+        if url.scheme in ["rediss", "redisclusters"]:
             kwargs["connection_class"] = SSLConnection
 
     return kwargs

tests/test_rediscluster_schema.py

@@ -0,0 +1,16 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Xiang Wang <ramwin@qq.com>
+
+
+import unittest
+
+from redis import Redis
+from redis.cluster import RedisCluster
+
+
+class Test(unittest.TestCase):
+
+    def test(self):
+        res = Redis.from_url("rediscluster://localhost:7000")
+        self.assertIsInstance(res, RedisCluster)