fix(db): add NOT VALID in constraint for bib_type_site (#36)

Cannot use alembic to do this because need sqlalchemy >= 1.4.32
PnX-SI · Jan 17, 2023 · 0e15017 · 0e15017
1 parent 2834de1
commit 0e15017
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/backend/gn_module_monitoring/migrations/b53bafb13ce8_create_bib_type_site.py b/backend/gn_module_monitoring/migrations/b53bafb13ce8_create_bib_type_site.py
@@ -36,12 +36,26 @@ def upgrade():
         schema=monitorings_schema,
     )
 
-    op.create_check_constraint(
-        "ck_bib_type_site_id_nomenclature",
-        "bib_type_site",
-        f"{nomenclature_schema}.check_nomenclature_type_by_mnemonique(id_nomenclature,'TYPE_SITE')",
-        schema=monitorings_schema,
+    # FIXME: if sqlalchemy >= 1.4.32, it should work with postgresql_not_valid=True: cleaner
+    # op.create_check_constraint(
+    #     "ck_bib_type_site_id_nomenclature",
+    #     "bib_type_site",
+    #     f"{nomenclature_schema}.check_nomenclature_type_by_mnemonique(id_nomenclature,'TYPE_SITE')",
+    #     schema=monitorings_schema,
+    #     postgresql_not_valid=True
+    # )
+    statement = sa.text(
+        f"""
+        ALTER TABLE {monitorings_schema}.bib_type_site 
+        ADD
+          CONSTRAINT ck_bib_type_site_id_nomenclature CHECK (
+            {nomenclature_schema}.check_nomenclature_type_by_mnemonique(
+              id_nomenclature, 'TYPE_SITE' :: character varying
+            )
+          ) NOT VALID
+        """
     )
+    op.execute(statement)
 
 
 def downgrade():