automatically provision hstore for pg13+

Change-Id: I5cd7e9e9ab8a1dae2bd467a1e4299d7f26183301

1 files changed, 12 insertions, 3 deletions

diff --git a/lib/sqlalchemy/dialects/postgresql/provision.py b/lib/sqlalchemy/dialects/postgresql/provision.py
index 5c6b9fcea..582157604 100644
--- a/lib/sqlalchemy/dialects/postgresql/provision.py
+++ b/lib/sqlalchemy/dialects/postgresql/provision.py
@@ -148,9 +148,18 @@ def _upsert(cfg, table, returning, set_lambda=None):
     return stmt
 
 
+_extensions = [
+    ("citext", (13,)),
+    ("hstore", (13,)),
+]
+
+
 @post_configure_engine.for_db("postgresql")
 def _create_citext_extension(url, engine, follower_ident):
     with engine.connect() as conn:
-        if conn.dialect.server_version_info >= (13,):
-            conn.execute(text("CREATE EXTENSION IF NOT EXISTS citext"))
-            conn.commit()
+        for extension, min_version in _extensions:
+            if conn.dialect.server_version_info >= min_version:
+                conn.execute(
+                    text(f"CREATE EXTENSION IF NOT EXISTS {extension}")
+                )
+                conn.commit()