[ADD] validate_fk_constraints function

This is useful in cases where you need to delete a lot of records,
and you speed the deletion up by disabling FK triggers,
but you want to validate integrity after deletion.

openupgradelib/openupgrade.py

@@ -195,6 +195,7 @@ def do_raise(error):
     "cow_templates_mark_if_equal_to_upstream",
     "cow_templates_replicate_upstream",
     "clean_transient_models",
+    "validate_fk_constraints",
 ]
 
 
@@ -3547,3 +3548,32 @@ def clean_transient_models(cr):
                 cr.execute(query)
         except Exception as e:
             logger.warning("Failed to clean transient table %s\n%s", table_name, str(e))
+
+
+def validate_fk_constraints(cr, table, column):
+    """
+    Validate all FK constraints to a column in table.
+
+    This is useful in cases where you need to delete a lot of records,
+    and you speed the deletion up by disabling FK triggers,
+    but you want to validate integrity after deletion.
+
+    Can only be run as superuser.
+
+    :param cr: Database cursor.
+    """
+    cr.execute(
+        query,
+        {
+            "table": table,
+            "column": column,
+        },
+    )
+    for schema, table, constraint in cr.fetchall():
+        cr.execute(
+            "update pg_constraint set convalidated = false where conname = %s",
+            (constraint,),
+        )
+        cr.execute(
+            "alter table %s validate constraint %s", (AsIs(table), AsIs(constraint))
+        )