moved json encoding into a utils file

Author: dantownsend

piccolo/columns/column_types.py

@@ -17,6 +17,7 @@
 )
 from piccolo.columns.defaults.uuid import UUIDArg, UUID4
 from piccolo.querystring import Unquoted, QueryString
+from piccolo.utils.encoding import dump_json
 
 if t.TYPE_CHECKING:
     from piccolo.table import Table
@@ -1064,15 +1065,7 @@ def __init__(
         self._validate_default(default, (str, list, dict, None))
 
         if isinstance(default, (list, dict)):
-            # TODO - move to utils
-            try:
-                import orjson
-            except ImportError:
-                import json
-
-                default = json.dumps(default, default=str)
-            else:
-                default = orjson.dumps(default, default=str).decode("utf8")
+            default = dump_json(default)
 
         self.default = default
         kwargs.update({"default": default})

piccolo/query/base.py

@@ -5,6 +5,7 @@
 
 from piccolo.querystring import QueryString
 from piccolo.utils.sync import run_sync
+from piccolo.utils.encoding import dump_json
 
 if t.TYPE_CHECKING:
     from piccolo.table import Table  # noqa
@@ -68,14 +69,7 @@ async def _process_results(self, results):
                     else:
                         raw = list(itertools.chain(*[j.values() for j in raw]))
                 if output._output.as_json:
-                    try:
-                        import orjson
-                    except ImportError:
-                        import json
-
-                        raw = json.dumps(raw, default=str)
-                    else:
-                        raw = orjson.dumps(raw, default=str).decode("utf8")
+                    raw = dump_json(raw)
 
         return raw
 

piccolo/utils/encoding.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+import typing as t
+
+try:
+    import orjson
+
+    ORJSON = True
+except ImportError:
+    import json
+
+    ORJSON = False
+
+
+def dump_json(data: t.Any) -> str:
+    if ORJSON:
+        return orjson.dumps(data, default=str).decode("utf8")
+    else:
+        return json.dumps(data, default=str)