S01E12

Author: ansipunk

databases/backends/psycopg.py

@@ -1,6 +1,5 @@
 import typing
 
-import orjson
 import psycopg
 import psycopg.adapt
 import psycopg.types
@@ -20,15 +19,33 @@
     TransactionBackend,
 )
 
+try:
+    import orjson
+
+    def load(data):
+        return orjson.loads(data)
+
+    def dump(data):
+        return orjson.dumps(data)
+
+except ImportError:
+    import json
+
+    def load(data):
+        return json.loads(data.decode("utf-8"))
+
+    def dump(data):
+        return json.dumps(data).encode("utf-8")
+
 
 class JsonLoader(psycopg.adapt.Loader):
     def load(self, data):
-        return orjson.loads(data)
+        return load(data)
 
 
 class JsonDumper(psycopg.adapt.Dumper):
     def dump(self, data):
-        return orjson.dumps(data)
+        return dump(data)
 
 
 class PsycopgBackend(DatabaseBackend):

requirements.txt

@@ -1,5 +1,8 @@
 -e .
 
+# Speedups
+orjson==3.9.15
+
 # Async database drivers
 asyncmy==0.2.9
 aiomysql==0.2.0

setup.py

@@ -58,6 +58,7 @@ def get_packages(package):
         "psycopg3": ["psycopg", "psycopg-pool"],
         "sqlite": ["aiosqlite"],
         "aiosqlite": ["aiosqlite"],
+        "orjson": ["orjson"],
     },
     classifiers=[
         "Development Status :: 3 - Alpha",