def update_script(self):
"""Generates a script that updates/alters an object of the inheriting type"""
data = self._update_query_data()
return templating.render_template(
templating.get_template_path(self._mxin_template_root,
'update.sql',
self._mxin_server_version),
self._mxin_macro_root, **data)
def get_index_constraints(server, did, tid, ctype, cid=None) -> (dict):
server_version = server.connection.server_version
sql = templating.render_template(
templating.get_template_path(GET_INDEX_CONSTRAINT_TEMPLATE_ROOT, 'properties.sql', server_version),
macro_roots=None,
did=did, tid=tid, cid=cid, constraint_type=ctype
)
result = server.connection.execute_dict(sql)
for idx_cons in result[1]:
sql = templating.render_template(
templating.get_template_path(
GET_INDEX_CONSTRAINT_TEMPLATE_ROOT,
'get_constraint_cols.sql',
server_version),
macro_roots=None,
cid=idx_cons['oid'],
colcnt=idx_cons['col_count'])
constraint_cols = server.connection.execute_dict(sql)
columns = []
for r in constraint_cols[1]:
columns.append({"column": r['column'].strip('"')})
idx_cons['columns'] = columns
# INCLUDE clause in index is supported from PG-11+
if server_version[0] >= 11:
sql = templating.render_template(
templating.get_template_path(
GET_INDEX_CONSTRAINT_TEMPLATE_ROOT,
'get_constraint_include.sql',
server_version),
macro_roots=None,
cid=idx_cons['oid'])
constraint_cols = server.connection.execute_dict(sql)
idx_cons['include'] = [col['colname'] for col in constraint_cols[1]]
return result[1]
def create_script(self):
"""Generates a script that creates an object of the inheriting type"""
data = self._create_query_data()
template_root = self._template_root(self._server)
sql = templating.render_template(templating.get_template_path(
template_root, 'create.sql', self._server_version),
macro_roots=self._macro_root(),
**data)
cols, rows = self._server.connection.execute_dict(sql)
script = rows[0]["Create Function"]
return script
def get_check_constraints(server, tid):
server_version = server.connection.server_version
sql = templating.render_template(
templating.get_template_path(GET_CHECK_CONSTRAINT_TEMPLATE_ROOT, 'properties.sql', server_version),
macro_roots=None,
tid=tid
)
result = server.connection.execute_dict(sql)
return result[1]
def get_exclusion_constraints(server, did, tid):
server_version = server.connection.server_version
sql = templating.render_template(
templating.get_template_path(GET_EXCLUSION_CONSTRAINT_TEMPLATE_ROOT, 'properties.sql', server_version),
macro_roots=None,
did=did, tid=tid
)
result = server.connection.execute_dict(sql)
for ex in result[1]:
sql = templating.render_template(
templating.get_template_path(GET_EXCLUSION_CONSTRAINT_TEMPLATE_ROOT, 'get_constraint_cols.sql', server_version),
macro_roots=None,
cid=ex['oid'], colcnt=ex['col_count']
)
res = server.connection.execute_dict(sql)
columns = _get_columns(res)
ex['columns'] = columns
# INCLUDE clause in index is supported from PG-11+
if server_version[0] >= 11:
sql = templating.render_template(
templating.get_template_path(GET_EXCLUSION_CONSTRAINT_TEMPLATE_ROOT, 'get_constraint_include.sql', server_version),
macro_roots=None,
cid=ex['oid']
)
res = server.connection.execute_dict(sql)
ex['include'] = [col['colname'] for col in res[1]]
if ex.get('amname', '') == "":
ex['amname'] = 'btree'
return result[1]
def _sequence_property_generator(self):
template_root = self._template_root(self._server)
# Setup the parameters for the query
template_vars = self.template_vars
# Render and execute the template
sql = templating.render_template(
templating.get_template_path(template_root, 'properties.sql',
self._server.version),
self._macro_root(), **template_vars)
cols, rows = self._server.connection.execute_dict(sql)
if len(rows) > 0:
return rows[0]
def _column_property_generator(self):
template_root = self._template_root(self._server)
# Setup the parameters for the query
template_vars = self.template_vars
# Render and execute the template
sql = templating.render_template(
templating.get_template_path(template_root, 'properties.sql', self._server.version),
self._macro_root(),
**template_vars
)
cols, rows = self._server.connection.execute_dict(sql)
for row in rows:
if row['name'] == self._name:
return row
def get_parent(server, tid, template_path=None):
server_version = server.connection.server_version
sql = templating.render_template(
templating.get_template_path(
GET_FOREIGN_CONSTRAINT_TEMPLATE_ROOT,
'get_parent.sql',
server_version),
macro_roots=None,
tid=tid)
rset = server.connection.execute_2darray(sql)
schema = ''
table = ''
if 'rows' in rset and len(rset['rows']) > 0:
schema = rset['rows'][0][0]
table = rset['rows'][0][1]
return schema, table