ConnectionPool's get_connection() now can take optional kwargs

Author: andymccurdy

redis/client.py

@@ -216,8 +216,9 @@ def pubsub(self, shard_hint=None):
     #### COMMAND EXECUTION AND PROTOCOL PARSING ####
     def execute_command(self, *args, **options):
         "Execute a command and return a parsed response"
+        pool = self.connection_pool
         command_name = args[0]
-        connection = self.connection_pool.get_connection(command_name)
+        connection = pool.get_connection(command_name, **options)
         try:
             connection.send_command(*args)
             return self.parse_response(connection, command_name, **options)
@@ -226,7 +227,7 @@ def execute_command(self, *args, **options):
             connection.send_command(*args)
             return self.parse_response(connection, command_name, **options)
         finally:
-            self.connection_pool.release(connection)
+            pool.release(connection)
 
     def parse_response(self, connection, command_name, **options):
         "Parses a response from the Redis server"

redis/connection.py

@@ -250,7 +250,7 @@ def __init__(self, connection_class=Connection, max_connections=None,
         self._available_connections = []
         self._in_use_connections = set()
 
-    def get_connection(self, command_name, *keys):
+    def get_connection(self, command_name, *keys, **options):
         "Get a connection from the pool"
         try:
             connection = self._available_connections.pop()