test0908
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Simplified introspection methods for PostgreSQL.

This commit is contained in:
Nick Pope 2018-09-12 01:18:16 +01:00 committed by Tim Graham
parent 245c36d7b6
commit f87f9c5f63
1 changed files with 18 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
django/db/backends/postgresql/introspection.py
View File

@ -46,10 +46,13 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
WHERE c.relkind IN ('r', 'v') WHERE c.relkind IN ('r', 'v')
AND n.nspname NOT IN ('pg_catalog', 'pg_toast') AND n.nspname NOT IN ('pg_catalog', 'pg_toast')
AND pg_catalog.pg_table_is_visible(c.oid)""") AND pg_catalog.pg_table_is_visible(c.oid)
return [TableInfo(row[0], {'r': 't', 'v': 'v'}.get(row[1])) """)
for row in cursor.fetchall() mapping = {'r': 't', 'v': 'v'}
if row[0] not in self.ignored_tables] return [
TableInfo(row[0], mapping[row[1]])
for row in cursor.fetchall() if row[0] not in self.ignored_tables
]
def get_table_description(self, cursor, table_name): def get_table_description(self, cursor, table_name):
""" """
@ -70,12 +73,11 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
] ]
def get_sequences(self, cursor, table_name, table_fields=()): def get_sequences(self, cursor, table_name, table_fields=()):
sequences = []
cursor.execute(""" cursor.execute("""
SELECT s.relname as sequence_name, col.attname SELECT s.relname as sequence_name, col.attname
FROM pg_class s FROM pg_class s
JOIN pg_namespace sn ON sn.oid = s.relnamespace JOIN pg_namespace sn ON sn.oid = s.relnamespace
JOIN pg_depend d ON d.refobjid = s.oid AND d.refclassid='pg_class'::regclass JOIN pg_depend d ON d.refobjid = s.oid AND d.refclassid = 'pg_class'::regclass
JOIN pg_attrdef ad ON ad.oid = d.objid AND d.classid = 'pg_attrdef'::regclass JOIN pg_attrdef ad ON ad.oid = d.objid AND d.classid = 'pg_attrdef'::regclass
JOIN pg_attribute col ON col.attrelid = ad.adrelid AND col.attnum = ad.adnum JOIN pg_attribute col ON col.attrelid = ad.adrelid AND col.attnum = ad.adnum
JOIN pg_class tbl ON tbl.oid = ad.adrelid JOIN pg_class tbl ON tbl.oid = ad.adrelid
@ -85,9 +87,10 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
AND n.nspname = 'public' AND n.nspname = 'public'
AND tbl.relname = %s AND tbl.relname = %s
""", [table_name]) """, [table_name])
for row in cursor.fetchall(): return [
sequences.append({'name': row[0], 'table': table_name, 'column': row[1]}) {'name': row[0], 'table': table_name, 'column': row[1]}
return sequences for row in cursor.fetchall()
]
def get_relations(self, cursor, table_name): def get_relations(self, cursor, table_name):
""" """
@ -101,15 +104,11 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
LEFT JOIN pg_class c2 ON con.confrelid = c2.oid LEFT JOIN pg_class c2 ON con.confrelid = c2.oid
LEFT JOIN pg_attribute a1 ON c1.oid = a1.attrelid AND a1.attnum = con.conkey[1] LEFT JOIN pg_attribute a1 ON c1.oid = a1.attrelid AND a1.attnum = con.conkey[1]
LEFT JOIN pg_attribute a2 ON c2.oid = a2.attrelid AND a2.attnum = con.confkey[1] LEFT JOIN pg_attribute a2 ON c2.oid = a2.attrelid AND a2.attnum = con.confkey[1]
WHERE c1.relname = %s WHERE c1.relname = %s AND con.contype = 'f'
AND con.contype = 'f'""", [table_name]) """, [table_name])
relations = {} return {row[1]: (row[2], row[0]) for row in cursor.fetchall()}
for row in cursor.fetchall():
relations[row[1]] = (row[2], row[0])
return relations
def get_key_columns(self, cursor, table_name): def get_key_columns(self, cursor, table_name):
key_columns = []
cursor.execute(""" cursor.execute("""
SELECT kcu.column_name, ccu.table_name AS referenced_table, ccu.column_name AS referenced_column SELECT kcu.column_name, ccu.table_name AS referenced_table, ccu.column_name AS referenced_column
FROM information_schema.constraint_column_usage ccu FROM information_schema.constraint_column_usage ccu
@ -121,9 +120,9 @@ class DatabaseIntrospection(BaseDatabaseIntrospection):
ON ccu.constraint_catalog = tc.constraint_catalog ON ccu.constraint_catalog = tc.constraint_catalog
AND ccu.constraint_schema = tc.constraint_schema AND ccu.constraint_schema = tc.constraint_schema
AND ccu.constraint_name = tc.constraint_name AND ccu.constraint_name = tc.constraint_name
WHERE kcu.table_name = %s AND tc.constraint_type = 'FOREIGN KEY'""", [table_name]) WHERE kcu.table_name = %s AND tc.constraint_type = 'FOREIGN KEY'
key_columns.extend(cursor.fetchall()) """, [table_name])
return key_columns return cursor.fetchall()
def get_constraints(self, cursor, table_name): def get_constraints(self, cursor, table_name):
""" """