是否有一种自动方法可以从许多表中删除与给定条件匹配的许多行？

Question

我有一个 Postgres 数据库，其中包含许多表，每个表包含许多行，这些不同的表之间有许多各种类型的关系。

如果我在所有这些表中添加一个新列 should_delete，是否有一些自动方法可以删除所有行 WHERE should_delete = true，而无需手动指定应从中删除行的每个表，或手动排序查询，以便在引用它们的其他行之前删除行（从而避免由于约束失败而导致的错误，例如，通过在包含外键的行之前删除由外键引用的行，即使两行都包含 should_delete = true 并且应该被删除）？

这最终会导致所有标记为 should_delete = true 的行都被删除，并且没有标记为 should_delete = false 的行被删除，但只有在无法在不违反约束的情况下仅删除标记为 should_delete = true 的行时才会导致错误.

如果没有纯粹的 Postgres 方式来实现这一点，我也在使用 Python 和 Django 来访问这个数据库，所以如果有一个 Python 包可以用来做到这一点也是可以接受的。

Answer 1

尝试按照选择获取包含您的列的表列表

SELECT table_name
FROM information_schema.columns
WHERE column_name = 'should_delete'
ORDER BY table_schema, table_name;

循环遍历结果并继续为每个表执行删除语句

我从这篇文章中获得灵感：https://kb.objectrocket.com/postgresql/postgres-list-tables-with-python-1023

第二种可能性是当您从选择中获得结果时生成带有实际删除语句的 SQL 文件 - 如下所示：

with open("delete_statements.sql", "w") as myFile:
    for table in resultFromPostgresqlSelect:
        myFile.write("DELTE FROM {} WHERE should_delete = true;\n".format(table))

编辑 - 经过几次 cmet 和澄清

您想要的是使用 python 连接到 PostgreSQL 并运行以下查询

SELECT
    tc.table_schema,
    tc.constraint_name,
    tc.table_name,
    kcu.column_name,
    ccu.table_schema AS foreign_table_schema,
    ccu.table_name AS foreign_table_name,
    ccu.column_name AS foreign_column_name
FROM
    information_schema.table_constraints AS tc
    JOIN information_schema.key_column_usage AS kcu
      ON tc.constraint_name = kcu.constraint_name
      AND tc.table_schema = kcu.table_schema
    JOIN information_schema.constraint_column_usage AS ccu
      ON ccu.constraint_name = tc.constraint_name
      AND ccu.table_schema = tc.table_schema
WHERE constraint_type = 'FOREIGN KEY';

它将为您提供以下外键列表和它们所在的表列表 - 以及更重要的是 - 引用这些表的列名

 table_schema | constraint_name | table_name |     column_name      | foreign_table_schema | foreign_table_name | foreign_column_name
--------------+-----------------+------------+----------------------+----------------------+--------------------+---------------------
 public       | fk_table_a_id   | table_b    | reference_table_a_id | public               | table_a            | id

从此，您将遍历每个foreign_table（在本例中为table_a）并选择所有应该删除=true 的ID，然后对该表的引用表执行相同的操作。 一旦你有了这些列表（在我的例子中是两个列表 - 一个来自 table_a 的 ID 列表和一个来自 table_b 的 ID 列表），你将加入它们并只找到两边的那些 - 你可以从两个表中删除的那些行

在我的例子中，我会生成这两个 SQL 并为它们获取结果

SELECT id FROM table_a WHERE should_delete = true;

SELECT reference_table_a_id FROM table_b WHERE should_delete = true;

然后我可以匹配这些 ID，并且我有从 table_a 和 table_b 中删除的列表

或者甚至更好 - 您生成删除查询，该查询将在表之间连接 - 有关详细信息，请参阅以下帖子PostgreSQL delete with inner join

这是解决您问题的pythonic/动态方式。如果您有任何问题/建议，请告诉我

---

在下面添加这两个表的定义

CREATE TABLE table_a (id INT GENERATED ALWAYS AS IDENTITY, something int, should_delete int DEFAULT 0,
PRIMARY KEY(id));

CREATE TABLE table_b (id INT GENERATED ALWAYS AS IDENTITY, something_else int, should_delete int DEFAULT 0, reference_table_a_id INT,
PRIMARY KEY(ID), 
CONSTRAINT fk_table_a_id
 FOREIGN KEY(reference_table_a_id) 
      REFERENCES table_a(id)
 );