Resolve a redis-py bug with decode_responses=True

cf redis/redis-py#694

limpyd/database.py

@@ -129,3 +129,22 @@ def has_scripting(self):
             except:
                 self._has_scripting = False
         return self._has_scripting
+
+
+class Lock(redis.client.Lock):
+    """
+    Override the default Lock class to manage the fact that we use ``decode_responses=True``
+    but it is not taken into account in python 3 with redis 2.10, as the lock token in
+    ``threading.local`` is saved as ``bytes`` but the value retrieved from redis is decoded.
+    So the equal check doesn't work.
+    So we decode the value in ``threading.local`` to make this check work.
+    See https://github.com/andymccurdy/redis-py/issues/694
+    """
+
+    def do_release(self, expected_token):
+
+        if isinstance(expected_token, bytes) and \
+                self.redis.connection_pool.connection_kwargs.get('decode_responses', False):
+            expected_token = expected_token.decode()
+
+        super(Lock, self).do_release(expected_token)

limpyd/fields.py

@@ -9,8 +9,8 @@
 from copy import copy
 
 from redis.exceptions import RedisError
-from redis.client import Lock
 
+from limpyd.database import Lock
 from limpyd.utils import make_key, normalize
 from limpyd.exceptions import *