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):

databases/core.py

@@ -49,7 +49,7 @@ class Database:
         "postgresql+asyncpg": "databases.backends.asyncpg:AsyncpgBackend",
         "postgresql+psycopg": "databases.backends.psycopg:PsycopgBackend",
         "mysql": "databases.backends.mysql:MySQLBackend",
-        "mysql+aiomysql": "databases.backends.asyncmy:MySQLBackend",
+        "mysql+aiomysql": "databases.backends.mysql:MySQLBackend",
         "mysql+asyncmy": "databases.backends.asyncmy:AsyncMyBackend",
         "sqlite": "databases.backends.sqlite:SQLiteBackend",
         "sqlite+aiosqlite": "databases.backends.sqlite:SQLiteBackend",

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",