Added support to view trigger function under the respective trigger node. Fixes #2519

This commit is contained in:
Akshay Joshi
2020-10-12 13:49:54 +05:30
parent 365ec0ba9f
commit 556278dbc5
9 changed files with 148 additions and 15 deletions

View File

@@ -1151,6 +1151,14 @@ class FunctionView(PGChildNodeView, DataTypeReader, SchemaDiffObjectCompare):
else:
object_type = 'function'
# We are showing trigger functions under the trigger node.
# It may possible that trigger is in one schema and trigger
# function is in another schema, so to show the SQL we need to
# change the schema id i.e scid.
if self.node_type == 'trigger_function' and \
scid != resp_data['pronamespace']:
scid = resp_data['pronamespace']
# Get Schema Name from its OID.
self._get_schema_name_from_oid(resp_data)

View File

@@ -38,9 +38,26 @@ define('pgadmin.node.trigger_function', [
dialogHelp: url_for('help.static', {'filename': 'trigger_function_dialog.html'}),
label: gettext('Trigger function'),
collection_type: 'coll-trigger_function',
canEdit: function(itemData, item) {
let node = pgBrowser.treeMenu.findNodeByDomElement(item);
if (!node || node.parentNode.getData()._type === 'trigger')
return false;
return true;
},
hasSQL: true,
showMenu: function(itemData, item) {
let node = pgBrowser.treeMenu.findNodeByDomElement(item);
if (!node || node.parentNode.getData()._type === 'trigger')
return false;
return true;
},
hasDepends: true,
hasStatistics: true,
url_jump_after_node: 'schema',
Init: function() {
/* Avoid mulitple registration of menus */
if (this.initialized)
@@ -68,7 +85,6 @@ define('pgadmin.node.trigger_function', [
enable: 'canCreate',
},
]);
},
model: pgBrowser.Node.Model.extend({
idAttribute: 'oid',

View File

@@ -0,0 +1,9 @@
SELECT p.oid AS tfuncoid, p.proname AS tfunction,
p.pronamespace AS tfuncschoid, n.nspname AS tfuncschema, l.lanname
FROM pg_trigger t
LEFT OUTER JOIN pg_proc p ON p.oid=t.tgfoid
LEFT OUTER JOIN pg_namespace n ON n.oid = p.pronamespace
LEFT OUTER JOIN pg_language l ON l.oid=p.prolang
WHERE NOT tgisinternal
AND tgrelid = {{tid}}::OID
AND t.oid = {{trid}}::OID

View File

@@ -0,0 +1,9 @@
SELECT p.oid AS tfuncoid, p.proname AS tfunction,
p.pronamespace AS tfuncschoid, n.nspname AS tfuncschema, l.lanname
FROM pg_trigger t
LEFT OUTER JOIN pg_proc p ON p.oid=t.tgfoid
LEFT OUTER JOIN pg_namespace n ON n.oid = p.pronamespace
LEFT OUTER JOIN pg_language l ON l.oid=p.prolang
WHERE NOT tgisinternal
AND tgrelid = {{tid}}::OID
AND t.oid = {{trid}}::OID

View File

@@ -121,7 +121,7 @@ class TriggerModule(CollectionNodeModule):
"""
Load the module node as a leaf node
"""
return False
return True
@property
def module_use_template_javascript(self):
@@ -280,6 +280,11 @@ class TriggerView(PGChildNodeView, SchemaDiffObjectCompare):
)
self.template_path = 'triggers/sql/{0}/#{1}#'.format(
self.manager.server_type, self.manager.version)
self.trigger_function_template_path = \
'trigger_functions/{0}/sql/#{1}#'.format(
self.manager.server_type, self.manager.version)
# Store server type
self.server_type = self.manager.server_type
# We need parent's name eg table name and schema name
@@ -293,6 +298,69 @@ class TriggerView(PGChildNodeView, SchemaDiffObjectCompare):
return wrap
@check_precondition
def get_children_nodes(self, manager, **kwargs):
"""
Function is used to get the child nodes.
:param manager:
:param kwargs:
:return:
"""
nodes = []
scid = kwargs.get('scid')
tid = kwargs.get('tid')
trid = kwargs.get('trid')
try:
SQL = render_template(
"/".join([self.template_path, 'get_function_oid.sql']),
tid=tid, trid=trid
)
status, rset = self.conn.execute_2darray(SQL)
if not status:
return internal_server_error(errormsg=rset)
if len(rset['rows']) == 0:
return gone(
gettext("Could not find the specified trigger function"))
# For language EDB SPL we should not display any node.
if rset['rows'][0]['lanname'] != 'edbspl':
trigger_function_schema_oid = rset['rows'][0]['tfuncschoid']
sql = render_template("/".join(
[self.trigger_function_template_path, self._NODE_SQL]),
scid=trigger_function_schema_oid,
fnid=rset['rows'][0]['tfuncoid']
)
status, res = self.conn.execute_2darray(sql)
if not status:
return internal_server_error(errormsg=rset)
if len(res['rows']) == 0:
return gone(gettext(
"Could not find the specified trigger function"))
row = res['rows'][0]
func_name = row['name']
# If trigger function is from another schema then we should
# display the name as schema qulified name.
if scid != trigger_function_schema_oid:
func_name = \
rset['rows'][0]['tfuncschema'] + '.' + row['name']
trigger_func = current_app.blueprints['NODE-trigger_function']
nodes.append(trigger_func.generate_browser_node(
row['oid'], trigger_function_schema_oid,
gettext(func_name),
icon="icon-trigger_function", funcowner=row['funcowner'],
language=row['lanname'], inode=False)
)
except Exception as e:
return internal_server_error(errormsg=str(e))
return nodes
@check_precondition
def get_trigger_functions(self, gid, sid, did, scid, tid, trid=None):
"""