using orjson for JSON serialisation when using output(as_json=True) clause

Author: dantownsend

docs/src/piccolo/query_types/select.rst

@@ -143,6 +143,8 @@ To return the data as a JSON string:
     >>> b.select().output(as_json=True).run_sync()
     '[{"name":"Pythonistas","manager":1,"popularity":1000,"id":1},{"name":"Rustaceans","manager":2,"popularity":500,"id":2}]'
 
+Piccolo uses `orjson <https://github.com/ijl/orjson>`_ for JSON serialisation, which is blazing fast, and can handle most Python types, including dates, datetimes, and UUIDs.
+
 where
 ~~~~~
 

piccolo/query/base.py

@@ -3,7 +3,8 @@
 from time import time
 import typing as t
 
-import ujson as json
+from asyncpg.pgproto.pgproto import UUID
+import orjson as json
 
 from piccolo.querystring import QueryString
 from piccolo.utils.sync import run_sync
@@ -21,6 +22,15 @@ def __exit__(self, exception_type, exception, traceback):
         print(f"Duration: {self.end - self.start}s")
 
 
+def default(obj):
+    """
+    Used for handling edge cases which orjson can't serialise out of the box.
+    """
+    if isinstance(obj, UUID):
+        return str(obj)
+    raise TypeError
+
+
 class Query:
 
     __slots__ = ("table",)
@@ -70,7 +80,7 @@ async def _process_results(self, results):
                     else:
                         raw = list(itertools.chain(*[j.values() for j in raw]))
                 if output._output.as_json:
-                    raw = json.dumps(raw)
+                    raw = json.dumps(raw, default=default)
 
         return raw
 

requirements.txt

@@ -3,6 +3,6 @@ asgiref==3.2.10
 asyncpg==0.20.1
 black==19.10b0
 colorama==0.4.3
-ujson==3.0.0
+orjson==3.3.0
 Jinja2==2.11.2
 targ==0.1.*