always get the same proxy node if the master node is the same

Author: kefu-reddit

rediscluster/pipeline.py

@@ -187,7 +187,7 @@ def _send_cluster_commands(self, stack, raise_on_error=True, allow_redirections=
 
         # build a list of node objects based on node names we need to
         nodes = {}
-        node_by_slot = {}
+        proxy_node_by_master = {}
         connection_by_node = {}
 
         # as we move through each command that still needs to be processed,
@@ -197,11 +197,13 @@ def _send_cluster_commands(self, stack, raise_on_error=True, allow_redirections=
             # command should route to.
             slot = self._determine_slot(*c.args)
 
-            if slot in node_by_slot:
-                node = node_by_slot[slot]
+            master_node = self.connection_pool.get_node_by_slot(slot)
+
+            if master_node['name'] in proxy_node_by_master:
+                node = proxy_node_by_master[master_node['name']]
             else:
                 node = self.connection_pool.get_node_by_slot(slot, self.read_from_replicas)
-                node_by_slot[slot] = node
+                proxy_node_by_master[master_node['name']] = node
 
                 # little hack to make sure the node name is populated. probably could clean this up.
                 self.connection_pool.nodes.set_node_name(node)

setup.py

@@ -20,7 +20,7 @@
 
 setup(
     name="redis-py-cluster-patched",
-    version="2.1.0.999.5",
+    version="2.1.0.999.6",
     description="Library for communicating with Redis Clusters. Built on top of redis-py lib",
     long_description=readme + '\n\n' + history,
     long_description_content_type="text/markdown",
@@ -30,7 +30,7 @@
     maintainer_email='Grokzen@gmail.com',
     packages=["rediscluster"],
     url='http://github.com/grokzen/redis-py-cluster',
-    download_url="https://github.com/duke-cliff/redis-py-cluster/archive/2.1.0.999.5.tar.gz",
+    download_url="https://github.com/duke-cliff/redis-py-cluster/archive/2.1.0.999.6.tar.gz",
     license='MIT',
     install_requires=[
         'redis>=3.0.0,<4.0.0'