Add translate_func

Amertz08 · Aug 17, 2020 · ede0d09 · ede0d09
1 parent d3fc8d4
commit ede0d09
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,11 @@
 - h3 heading
 - X.X.X - YYYY-MM-DD
 
+### 1.7.0.alpha1 - 2020-08-17
+
+- Added `UJSONRenderer.translate_func` attribute for translating data into a `ujson`
+serializable object.
+
 ### 1.6.0 - 2020-03-10
 
 - Add `ujson>=2` compatibility

diff --git a/drf_ujson/__init__.py b/drf_ujson/__init__.py
@@ -1,2 +1,2 @@
 __author__ = "y.gavenchuk aka murminathor"
-__version__ = (1, 6, 1)
+__version__ = (1, 7, 0)
diff --git a/drf_ujson/renderers.py b/drf_ujson/renderers.py
@@ -20,6 +20,9 @@ class UJSONRenderer(JSONRenderer):
     # Unicode sequences.
     encode_html_chars: bool = False
 
+    # TODO: typing
+    translate_func = None
+
     def render(
         self,
         data: Union[dict, None],
@@ -30,6 +33,9 @@ def render(
         if data is None:
             return bytes()
 
+        if self.translate_func is not None:
+            data = self.translate_func(data)
+
         accepted_media_type = accepted_media_type or ""
         renderer_context = renderer_context or {}
         indent = self.get_indent(accepted_media_type, renderer_context)

diff --git a/setup.py b/setup.py
@@ -7,9 +7,10 @@
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
+# TODO: remove alpha
 setup(
     name="drf_ujson2",
-    version=".".join([str(i) for i in __version__]),
+    version=".".join([str(i) for i in __version__]) + ".alpha1",
     python_requires=">=3.6",
     description="Django Rest Framework UJSON Renderer",
     keywords="django,djangorestframework,ujson",