diff options
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/provision.py')
-rw-r--r-- | lib/sqlalchemy/dialects/postgresql/provision.py | 15 |
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() |