208 lines
6.6 KiB
Python
208 lines
6.6 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2015, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
"""Defines views for management of server groups"""
|
|
|
|
from abc import ABCMeta, abstractmethod
|
|
import traceback
|
|
from flask import Blueprint, Response, current_app, request, render_template
|
|
from flask.ext.babel import gettext
|
|
from flask.ext.security import current_user, login_required
|
|
from pgadmin import current_blueprint
|
|
from pgadmin.utils.ajax import make_json_response
|
|
from pgadmin.browser import BrowserPluginModule
|
|
from pgadmin.utils.menu import MenuItem
|
|
from pgadmin.settings.settings_model import db, ServerGroup
|
|
import config
|
|
|
|
|
|
|
|
class ServerGroupModule(BrowserPluginModule):
|
|
|
|
NODE_TYPE = "server-group"
|
|
|
|
def get_own_menuitems(self):
|
|
return {
|
|
'standard_items': [
|
|
ServerGroupMenuItem(action="drop", priority=10, function="drop_server_group"),
|
|
ServerGroupMenuItem(action="rename", priority=10, function="rename_server_group")
|
|
],
|
|
'create_items': [
|
|
ServerGroupMenuItem(name="create_server_group",
|
|
label=gettext('Server Group...'),
|
|
priority=10,
|
|
function="create_server_group",
|
|
types=[self.node_type])
|
|
],
|
|
'context_items': [
|
|
ServerGroupMenuItem(name="delete_server_group",
|
|
label=gettext('Delete server group'),
|
|
priority=10,
|
|
onclick='drop_server_group(item);'),
|
|
ServerGroupMenuItem(name="rename_server_group",
|
|
label=gettext('Rename server group...'),
|
|
priority=10,
|
|
onclick='rename_server_group(item);')
|
|
]
|
|
}
|
|
|
|
|
|
@property
|
|
def jssnippets(self):
|
|
snippets = [render_template("server_groups/server_groups.js")]
|
|
for module in self.submodules:
|
|
snippets.extend(module.jssnippets)
|
|
return snippets
|
|
|
|
def get_nodes(self, **kwargs):
|
|
"""Return a JSON document listing the server groups for the user"""
|
|
groups = ServerGroup.query.filter_by(user_id=current_user.id)
|
|
# TODO: Move this JSON generation to a Server method
|
|
# this code is duplicated somewhere else
|
|
for group in groups:
|
|
yield {
|
|
"id": "%s/%d" % (self.node_type, group.id),
|
|
"label": group.name,
|
|
"icon": "icon-%s" % self.node_type,
|
|
"inode": True,
|
|
"_type": self.node_type
|
|
}
|
|
|
|
@property
|
|
def node_type(self):
|
|
return self.NODE_TYPE
|
|
|
|
|
|
|
|
class ServerGroupMenuItem(MenuItem):
|
|
|
|
def __init__(self, **kwargs):
|
|
kwargs.setdefault("type", ServerGroupModule.NODE_TYPE)
|
|
super(ServerGroupMenuItem, self).__init__(**kwargs)
|
|
|
|
|
|
class ServerGroupPluginModule(BrowserPluginModule):
|
|
"""
|
|
Base class for server group plugins.
|
|
"""
|
|
|
|
__metaclass__ = ABCMeta
|
|
|
|
|
|
@abstractmethod
|
|
def get_nodes(self, servergroup):
|
|
pass
|
|
|
|
|
|
# Initialise the module
|
|
blueprint = ServerGroupModule( __name__, static_url_path='')
|
|
|
|
@blueprint.route("/<server_group>")
|
|
@login_required
|
|
def get_nodes(server_group):
|
|
"""Build a list of treeview nodes from the child nodes."""
|
|
nodes = []
|
|
for module in current_blueprint.submodules:
|
|
nodes.extend(module.get_nodes(server_group=server_group))
|
|
return make_json_response(data=nodes)
|
|
|
|
|
|
@blueprint.route('/add/', methods=['POST'])
|
|
@login_required
|
|
def add():
|
|
"""Add a server group node to the settings database"""
|
|
success = 1
|
|
errormsg = ''
|
|
data = { }
|
|
|
|
if request.form['name'] != '':
|
|
servergroup = ServerGroup(user_id=current_user.id, name=request.form['name'])
|
|
|
|
try:
|
|
db.session.add(servergroup)
|
|
db.session.commit()
|
|
except Exception as e:
|
|
success = 0
|
|
errormsg = e.message
|
|
|
|
else:
|
|
success = 0
|
|
errormsg = gettext('No server group name was specified')
|
|
|
|
if success == 1:
|
|
data['id'] = servergroup.id
|
|
data['name'] = servergroup.name
|
|
|
|
return make_json_response(success=success,
|
|
errormsg=errormsg,
|
|
info=traceback.format_exc(),
|
|
result=request.form,
|
|
data=data)
|
|
|
|
@blueprint.route('/delete/', methods=['POST'])
|
|
@login_required
|
|
def delete():
|
|
"""Delete a server group node in the settings database"""
|
|
success = 1
|
|
errormsg = ''
|
|
|
|
if request.form['id'] != '':
|
|
# There can be only one record at most
|
|
servergroup = ServerGroup.query.filter_by(user_id=current_user.id, id=int(request.form['id'])).first()
|
|
|
|
if servergroup is None:
|
|
success = 0
|
|
errormsg = gettext('The specified server group could not be found.')
|
|
else:
|
|
try:
|
|
db.session.delete(servergroup)
|
|
db.session.commit()
|
|
except Exception as e:
|
|
success = 0
|
|
errormsg = e.message
|
|
|
|
else:
|
|
success = 0
|
|
errormsg = gettext('No server group was specified.')
|
|
|
|
return make_json_response(success=success,
|
|
errormsg=errormsg,
|
|
info=traceback.format_exc(),
|
|
result=request.form)
|
|
|
|
@blueprint.route('/rename/', methods=['POST'])
|
|
@login_required
|
|
def rename():
|
|
"""Rename a server group node in the settings database"""
|
|
success = 1
|
|
errormsg = ''
|
|
|
|
if request.form['id'] != '':
|
|
# There can be only one record at most
|
|
servergroup = ServerGroup.query.filter_by(user_id=current_user.id, id=int(request.form['id'])).first()
|
|
|
|
if servergroup is None:
|
|
success = 0
|
|
errormsg = gettext('The specified server group could not be found.')
|
|
else:
|
|
try:
|
|
servergroup.name = request.form['name']
|
|
db.session.commit()
|
|
except Exception as e:
|
|
success = 0
|
|
errormsg = e.message
|
|
|
|
else:
|
|
success = 0
|
|
errormsg = gettext('No server group was specified.')
|
|
|
|
return make_json_response(success=success,
|
|
errormsg=errormsg,
|
|
info=traceback.format_exc(),
|
|
result=request.form)
|