在 python 中填充 n 级 PostgreSQL table

Question

我无法思考如何填充 n 级 table（在我的例子中是 3 级），我使用 python 从查询中获取数据，但我不确定如何填充我的新 table resources 因为它引用了自己。对此方法的任何反馈将不胜感激！

在运行我的 python 文件中的以下查询之后，我得到以下 table

SELECT distinct c.table_catalog AS "Database", c.table_schema AS "Schema", c.table_name AS "Table"
FROM information_schema.columns c
WHERE c.table_schema != 'information_schema' AND c.table_schema != 'pg_catalog' AND c.table_schema != 'pg_internal' AND c.table_schema not like '% %'
ORDER BY c.table_schema, c.table_name;


Database  Schema            Table
____________________________________
dev       BigBangTheory     SomeTable1
dev       BigBangTheory     SomeTable2
dev       BigBangTheory     SomeTable3
dev       Walle             AnotherTable100
dev       Walle             AnotherTable200
dev       StarWars          SpaceTablexxx
dev       StarWars          SpaceTableyyy
stage     BigBangTheory     SomeTable1
stage     BigBangTheory     SomeTable2
stage     BigBangTheory     SomeTable3
stage     Walle             AnotherTable100
stage     Walle             AnotherTable200
stage     StarWars          SpaceTablexxx
stage     StarWars          SpaceTableyyy

我还有另一个 table 想要使用上述结果填充。我要填充的 table 看起来像这样：

CREATE TABLE IF NOT EXISTS resources
(
"id" SERIAL NOT NULL PRIMARY KEY,
"type" varchar(100) NOT NULL,             
"name" varchar(100) NOT NULL,    
"parent" int,
FOREIGN KEY (parent) REFERENCES resources (id)
);

所以我的目标是 table resources 看起来像这样：

id      type         name                   parent
____________________________________________________
1       database     dev                    NULL
2       schema       BigBangTheory          1
3       table        SomeTable1             2
4       table        SomeTable2             2
5       table        SomeTable3             2
6       schema       Walle                  1
7       table        AnotherTable100        6
8       table        AnotherTable200        6
9       schema       StarWars               1
10      table        SpaceTablexxx          9
11      table        SpaceTableyyy          9

12      database     stage                  NULL
13      schema       BigBangTheory          12
14      table        SomeTable1             13
15      table        SomeTable2             13
16      table        SomeTable3             13
17      schema       Walle                  12
18      table        AnotherTable100        17
19      table        AnotherTable200        17
20      schema       StarWars               12
21      table        SpaceTablexxx          20
22      table        SpaceTableyyy          20

提前致谢！感谢所有反馈 <3

Answer 1

作为初学者：您可以直接从 information_schema.tables 而不是 information_schema.columns 获取所需的信息（每个 table 只有一行，因此需要 distinct).

然后：在 Postgres 中，您可以在单个查询中执行您想要的操作，使用带有 returning 子句的级联常见 table 表达式到 Postgres 中的 insert 语句。可以

逻辑是先插入顶级对象（数据库）和 return 生成的序列，然后插入模式（使用数据库序列），最后插入 table。

with 
    info as (
        select c.table_catalog, c.table_schema, c.table_name
        from information_schema.tables
        where 
            c.table_schema not in ('information_schema', 'pg_catalog', 'pg_internal')
            and c.table_schema not like '% %'
    ),
    dbs as (
        insert into resources (type, name)
        select distinct 'database', table_catalog 
        from info
        returning id, name
    ),
    schemas as (
        insert into resources(type, name, parent)
        select distinct 'schema', i.table_schema, d.id
        from info i
        inner join dbs d on d.name = i.table_catalog
        returning id, name, parent
    )
insert into resources(type, name, parent)
select 'table', table_name, s.id
from info i
inner join schemas s on s.name = i.table_schema
inner join dbs d on d.id = s.parent and d.name = i.table_catalog

请注意，最后一个 insert 同时加入 schemas 和 dbs；这是为了正确处理生活在不同模式中的 "homonym" table。

Here 是一个演示（我使用 table 来模拟您初始查询的结果）。

在 python 中填充 n 级 PostgreSQL table

Populating n level PostgreSQL table in python

python

sql

postgresql

database-design

sql-insert