Re-enable pipeline support for JSON and TimeSeries (#1674)

4 files changed, 110 insertions, 8 deletions

diff --git a/redis/commands/helpers.py b/redis/commands/helpers.py
index 2a4298c..22cb622 100644
--- a/redis/commands/helpers.py
+++ b/redis/commands/helpers.py
@@ -1,3 +1,7 @@
+import random
+import string
+
+
 def list_or_args(keys, args):
     # returns a single new list combining keys and args
     try:
@@ -42,3 +46,31 @@ def parse_to_list(response):
         except TypeError:
             res.append(None)
     return res
+
+
+def random_string(length=10):
+    """
+    Returns a random N character long string.
+    """
+    return "".join(  # nosec
+        random.choice(string.ascii_lowercase) for x in range(length)
+    )
+
+
+def quote_string(v):
+    """
+    RedisGraph strings must be quoted,
+    quote_string wraps given v with quotes incase
+    v is a string.
+    """
+
+    if isinstance(v, bytes):
+        v = v.decode()
+    elif not isinstance(v, str):
+        return v
+    if len(v) == 0:
+        return '""'
+
+    v = v.replace('"', '\\"')
+
+    return '"{}"'.format(v)
diff --git a/redis/commands/json/__init__.py b/redis/commands/json/__init__.py
index d00627e..d634dbd 100644
--- a/redis/commands/json/__init__.py
+++ b/redis/commands/json/__init__.py
@@ -6,6 +6,7 @@ from .decoders import (
 )
 from ..helpers import nativestr
 from .commands import JSONCommands
+import redis
 
 
 class JSON(JSONCommands):
@@ -91,3 +92,29 @@ class JSON(JSONCommands):
     def _encode(self, obj):
         """Get the encoder."""
         return self.__encoder__.encode(obj)
+
+    def pipeline(self, transaction=True, shard_hint=None):
+        """Creates a pipeline for the JSON module, that can be used for executing
+        JSON commands, as well as classic core commands.
+
+        Usage example:
+
+        r = redis.Redis()
+        pipe = r.json().pipeline()
+        pipe.jsonset('foo', '.', {'hello!': 'world'})
+        pipe.jsonget('foo')
+        pipe.jsonget('notakey')
+        """
+        p = Pipeline(
+            connection_pool=self.client.connection_pool,
+            response_callbacks=self.MODULE_CALLBACKS,
+            transaction=transaction,
+            shard_hint=shard_hint,
+        )
+        p._encode = self._encode
+        p._decode = self._decode
+        return p
+
+
+class Pipeline(JSONCommands, redis.client.Pipeline):
+    """Pipeline for the module."""
diff --git a/redis/commands/json/commands.py b/redis/commands/json/commands.py
index 716741c..4436f6a 100644
--- a/redis/commands/json/commands.py
+++ b/redis/commands/json/commands.py
@@ -154,6 +154,9 @@ class JSONCommands:
         ``xx`` if set to True, set ``value`` only if it exists.
         ``decode_keys`` If set to True, the keys of ``obj`` will be decoded
         with utf-8.
+
+        For the purpose of using this within a pipeline, this command is also
+        aliased to jsonset.
         """
         if decode_keys:
             obj = decode_dict_keys(obj)
@@ -212,3 +215,18 @@ class JSONCommands:
             pieces.append(key)
             pieces.append(str(path))
         return self.execute_command("JSON.DEBUG", *pieces)
+
+    @deprecated(version='4.0.0',
+                reason='redisjson-py supported this, call get directly.')
+    def jsonget(self, *args, **kwargs):
+        return self.get(*args, **kwargs)
+
+    @deprecated(version='4.0.0',
+                reason='redisjson-py supported this, call get directly.')
+    def jsonmget(self, *args, **kwargs):
+        return self.mget(*args, **kwargs)
+
+    @deprecated(version='4.0.0',
+                reason='redisjson-py supported this, call get directly.')
+    def jsonset(self, *args, **kwargs):
+        return self.set(*args, **kwargs)
diff --git a/redis/commands/timeseries/__init__.py b/redis/commands/timeseries/__init__.py
index db9c3a5..83fa170 100644
--- a/redis/commands/timeseries/__init__.py
+++ b/redis/commands/timeseries/__init__.py
@@ -1,4 +1,4 @@
-from redis.client import bool_ok
+import redis.client
 
 from .utils import (
     parse_range,
@@ -37,12 +37,12 @@ class TimeSeries(TimeSeriesCommands):
     def __init__(self, client=None, version=None, **kwargs):
         """Create a new RedisTimeSeries client."""
         # Set the module commands' callbacks
-        MODULE_CALLBACKS = {
-            CREATE_CMD: bool_ok,
-            ALTER_CMD: bool_ok,
-            CREATERULE_CMD: bool_ok,
+        self.MODULE_CALLBACKS = {
+            CREATE_CMD: redis.client.bool_ok,
+            ALTER_CMD: redis.client.bool_ok,
+            CREATERULE_CMD: redis.client.bool_ok,
             DEL_CMD: int,
-            DELETERULE_CMD: bool_ok,
+            DELETERULE_CMD: redis.client.bool_ok,
             RANGE_CMD: parse_range,
             REVRANGE_CMD: parse_range,
             MRANGE_CMD: parse_m_range,
@@ -57,5 +57,30 @@ class TimeSeries(TimeSeriesCommands):
         self.execute_command = client.execute_command
         self.MODULE_VERSION = version
 
-        for k in MODULE_CALLBACKS:
-            self.client.set_response_callback(k, MODULE_CALLBACKS[k])
+        for key, value in self.MODULE_CALLBACKS.items():
+            self.client.set_response_callback(key, value)
+
+    def pipeline(self, transaction=True, shard_hint=None):
+        """Creates a pipeline for the TimeSeries module, that can be used
+        for executing only TimeSeries commands and core commands.
+
+        Usage example:
+
+        r = redis.Redis()
+        pipe = r.ts().pipeline()
+        for i in range(100):
+            pipeline.add("with_pipeline", i, 1.1 * i)
+        pipeline.execute()
+
+        """
+        p = Pipeline(
+            connection_pool=self.client.connection_pool,
+            response_callbacks=self.MODULE_CALLBACKS,
+            transaction=transaction,
+            shard_hint=shard_hint,
+        )
+        return p
+
+
+class Pipeline(TimeSeriesCommands, redis.client.Pipeline):
+    """Pipeline for the module."""