def __init__(self, curs, table_name):
"""Initializes class by loading info about table_name from database."""
super(TableStruct, self).__init__(curs, table_name)
self.table_name = table_name
# fill args
schema, name = skytools.fq_name_parts(table_name)
args = {
'schema': schema,
'table': name,
'fqname': self.fqname,
'fq2name': skytools.quote_literal(self.fqname),
'oid': skytools.get_table_oid(curs, table_name),
'pg_class_oid': skytools.get_table_oid(curs,
'pg_catalog.pg_class'),
}
# load table struct
self.col_list = self._load_elem(curs, self.name, args, TColumn)
# if db is GP then read also table distribution keys
if skytools.exists_table(curs, "pg_catalog.gp_distribution_policy"):
self.dist_key_list = self._load_elem(curs, self.name, args,
TGPDistKey)
else:
self.dist_key_list = None
self.object_list = [
TTable(table_name, self.col_list, self.dist_key_list)
]
self.seq_list = []
# load seqs
for col in self.col_list:
if col.seqname:
fqname = quote_fqident(col.seqname)
owner = self.fqname + '.' + quote_ident(col.name)
seq_args = {
'fqname': fqname,
'fq2name': skytools.quote_literal(fqname),
'owner': skytools.quote_literal(owner),
}
self.seq_list += self._load_elem(curs, col.seqname, seq_args,
TSeq)
self.object_list += self.seq_list
# load additional objects
to_load = [
TColumnDefault, TConstraint, TIndex, TTrigger, TRule, TGrant,
TOwner, TParent
]
for eclass in to_load:
self.object_list += self._load_elem(curs, self.name, args, eclass)
def __init__(self, curs, table_name):
"""Initializes class by loading info about table_name from database."""
BaseStruct.__init__(self, curs, table_name)
self.table_name = table_name
# fill args
schema, name = skytools.fq_name_parts(table_name)
args = {
'schema': schema,
'table': name,
'fqname': self.fqname,
'fq2name': skytools.quote_literal(self.fqname),
'oid': skytools.get_table_oid(curs, table_name),
'pg_class_oid': skytools.get_table_oid(curs, 'pg_catalog.pg_class'),
}
# load table struct
self.col_list = self._load_elem(curs, self.name, args, TColumn)
# if db is GP then read also table distribution keys
if skytools.exists_table(curs, "pg_catalog.gp_distribution_policy"):
self.dist_key_list = self._load_elem(curs, self.name, args,
TGPDistKey)
else:
self.dist_key_list = None
self.object_list = [ TTable(table_name, self.col_list,
self.dist_key_list) ]
self.seq_list = []
# load seqs
for col in self.col_list:
if col.seqname:
fqname = quote_fqident(col.seqname)
owner = self.fqname + '.' + quote_ident(col.name)
seq_args = { 'fqname': fqname, 'owner': skytools.quote_literal(owner) }
self.seq_list += self._load_elem(curs, col.seqname, seq_args, TSeq)
self.object_list += self.seq_list
# load additional objects
to_load = [TColumnDefault, TConstraint, TIndex, TTrigger,
TRule, TGrant, TOwner, TParent]
for eclass in to_load:
self.object_list += self._load_elem(curs, self.name, args, eclass)
def get_column_list(curs, tbl):
"""Get list of columns in right order."""
oid = skytools.get_table_oid(curs, tbl)
q = """SELECT a.attname FROM pg_attribute a
WHERE a.attrelid = %s
AND a.attnum > 0 AND NOT a.attisdropped
ORDER BY a.attnum"""
curs.execute(q, [oid])
list = []
for row in curs.fetchall():
list.append(row[0])
return list
def get_pkey_list(curs, tbl):
"""Get list of pkey fields in right order."""
oid = skytools.get_table_oid(curs, tbl)
q = """SELECT k.attname FROM pg_index i, pg_attribute k
WHERE i.indrelid = %s AND k.attrelid = i.indexrelid
AND i.indisprimary AND k.attnum > 0 AND NOT k.attisdropped
ORDER BY k.attnum"""
curs.execute(q, [oid])
list = []
for row in curs.fetchall():
list.append(row[0])
return list
def get_column_list(curs, tbl):
"""Get list of columns in right order."""
oid = skytools.get_table_oid(curs, tbl)
q = """SELECT a.attname FROM pg_attribute a
WHERE a.attrelid = %s
AND a.attnum > 0 AND NOT a.attisdropped
ORDER BY a.attnum"""
curs.execute(q, [oid])
list = []
for row in curs.fetchall():
list.append(row[0])
return list
def get_pkey_list(curs, tbl):
"""Get list of pkey fields in right order."""
oid = skytools.get_table_oid(curs, tbl)
q = """SELECT k.attname FROM pg_index i, pg_attribute k
WHERE i.indrelid = %s AND k.attrelid = i.indexrelid
AND i.indisprimary AND k.attnum > 0 AND NOT k.attisdropped
ORDER BY k.attnum"""
curs.execute(q, [oid])
list = []
for row in curs.fetchall():
list.append(row[0])
return list
def find_column_types(curs, table):
table_oid = skytools.get_table_oid(curs, table)
if table_oid == None:
return None
key_sql = """
SELECT k.attname FROM pg_index i, pg_attribute k
WHERE i.indrelid = %d AND k.attrelid = i.indexrelid
AND i.indisprimary AND k.attnum > 0 AND NOT k.attisdropped
""" % table_oid
# find columns
q = """
SELECT a.attname as name,
CASE WHEN k.attname IS NOT NULL
THEN 'k' ELSE 'v' END AS type
FROM pg_attribute a LEFT JOIN (%s) k ON (k.attname = a.attname)
WHERE a.attrelid = %d AND a.attnum > 0 AND NOT a.attisdropped
ORDER BY a.attnum
""" % (key_sql, table_oid)
curs.execute(q)
rows = curs.dictfetchall()
return rows
