From 23a83a342ad6d820ee5749ebccda04e54c373f7d Mon Sep 17 00:00:00 2001 From: Julian David Rath Date: Mon, 6 Mar 2023 08:43:44 +0100 Subject: [PATCH] Creating CITEXT extension in provision.py --- lib/sqlalchemy/dialects/postgresql/provision.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/lib/sqlalchemy/dialects/postgresql/provision.py b/lib/sqlalchemy/dialects/postgresql/provision.py index 4609701a2c..8bb47d958b 100644 --- a/lib/sqlalchemy/dialects/postgresql/provision.py +++ b/lib/sqlalchemy/dialects/postgresql/provision.py @@ -11,6 +11,7 @@ from ...testing.provision import drop_all_schema_objects_post_tables from ...testing.provision import drop_all_schema_objects_pre_tables from ...testing.provision import drop_db from ...testing.provision import log +from ...testing.provision import post_configure_engine from ...testing.provision import prepare_for_drop_tables from ...testing.provision import set_default_schema_on_connection from ...testing.provision import temp_table_keyword_args @@ -145,3 +146,9 @@ def _upsert(cfg, table, returning, set_lambda=None): stmt = stmt.returning(*returning) return stmt + + +@post_configure_engine.for_db("postgresql") +def _create_citext_extension(url, engine, follower_ident): + with engine.begin() as conn: + conn.execute(text("CREATE EXTENSION IF NOT EXISTS citext")) -- 2.47.2