Don't overwrite psycopg2 adapters when creating the connection (#113)

Don't overwrite psycopg2 adapters when creating the connection
procrastinate-org · Nov 29, 2019 · 6afaa82 · 6afaa82
2 parents 81fe265 + a779a95
commit 6afaa82
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/procrastinate/aiopg_connector.py b/procrastinate/aiopg_connector.py
@@ -16,6 +16,11 @@ def wrap_json(arguments: Dict[str, Any]):
 
 
 def get_connection(dsn="", **kwargs) -> Awaitable[aiopg.Connection]:
+    # tell aiopg not to register adapters for hstore & json by default, as
+    # those are registered at the module level and could overwrite previously
+    # defined adapters
+    kwargs.setdefault("enable_json", False)
+    kwargs.setdefault("enable_hstore", False)
     return aiopg.connect(dsn=dsn, **kwargs)