redisjson support (#1636)

1 files changed, 84 insertions, 0 deletions

diff --git a/redis/commands/json/__init__.py b/redis/commands/json/__init__.py
new file mode 100644
index 0000000..92f1199
--- /dev/null
+++ b/redis/commands/json/__init__.py
@@ -0,0 +1,84 @@
+# from typing import Optional
+from json import JSONDecoder, JSONEncoder
+
+# # from redis.client import Redis
+
+from .helpers import bulk_of_jsons
+from ..helpers import nativestr, delist
+from .commands import JSONCommands
+# from ..feature import AbstractFeature
+
+
+class JSON(JSONCommands):
+    """
+    Create a client for talking to json.
+
+    :param decoder:
+    :type json.JSONDecoder: An instance of json.JSONDecoder
+
+    :param encoder:
+    :type json.JSONEncoder: An instance of json.JSONEncoder
+    """
+
+    def __init__(
+        self,
+        client,
+        decoder=JSONDecoder(),
+        encoder=JSONEncoder(),
+    ):
+        """
+        Create a client for talking to json.
+
+        :param decoder:
+        :type json.JSONDecoder: An instance of json.JSONDecoder
+
+        :param encoder:
+        :type json.JSONEncoder: An instance of json.JSONEncoder
+        """
+        # Set the module commands' callbacks
+        self.MODULE_CALLBACKS = {
+            "JSON.CLEAR": int,
+            "JSON.DEL": int,
+            "JSON.FORGET": int,
+            "JSON.GET": self._decode,
+            "JSON.MGET": bulk_of_jsons(self._decode),
+            "JSON.SET": lambda r: r and nativestr(r) == "OK",
+            "JSON.NUMINCRBY": self._decode,
+            "JSON.NUMMULTBY": self._decode,
+            "JSON.TOGGLE": lambda b: b == b"true",
+            "JSON.STRAPPEND": int,
+            "JSON.STRLEN": int,
+            "JSON.ARRAPPEND": int,
+            "JSON.ARRINDEX": int,
+            "JSON.ARRINSERT": int,
+            "JSON.ARRLEN": int,
+            "JSON.ARRPOP": self._decode,
+            "JSON.ARRTRIM": int,
+            "JSON.OBJLEN": int,
+            "JSON.OBJKEYS": delist,
+            # "JSON.RESP": delist,
+            "JSON.DEBUG": int,
+        }
+
+        self.client = client
+        self.execute_command = client.execute_command
+
+        for key, value in self.MODULE_CALLBACKS.items():
+            self.client.set_response_callback(key, value)
+
+        self.__encoder__ = encoder
+        self.__decoder__ = decoder
+
+    def _decode(self, obj):
+        """Get the decoder."""
+        if obj is None:
+            return obj
+
+        try:
+            return self.__decoder__.decode(obj)
+        except TypeError:
+            return self.__decoder__.decode(obj.decode())
+
+    def _encode(self, obj):
+        """Get the encoder."""
+        return self.__encoder__.encode(obj)