IntenseWebs/pgadmin4
3
0
Fork 0
mirror of https://github.com/pgadmin-org/pgadmin4.git synced 2025-02-25 18:55:31 -06:00
Code Issues Packages Projects Releases Wiki Activity

Handle comments on databases with datallowconn=False correctly. Fixes #2024

Browse Source
This commit is contained in:
Murtuza Zabuawala 2017-01-16 15:42:25 +00:00 committed by Dave Page
parent 0d4bc0d27d
commit e431eb9d8e
1 changed files with 65 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

140
web/pgadmin/browser/server_groups/servers/databases/__init__.py
View File

@ -134,9 +134,17 @@ class DatabaseView(PGChildNodeView):
self.conn = self.manager.connection() self.conn = self.manager.connection()
elif 'did' in kwargs: elif 'did' in kwargs:
self.conn = self.manager.connection(did=kwargs['did']) self.conn = self.manager.connection(did=kwargs['did'])
# If connection to database is not allowed then
# provide generic connection
if kwargs['did'] in self.conn.manager.db_info:
self._db = self.conn.manager.db_info[kwargs['did']]
if self._db['datallowconn'] is False:
self.conn = self.manager.connection()
else: else:
self.conn = self.manager.connection() self.conn = self.manager.connection()
ver = self.manager.version ver = self.manager.version
# we will set template path for sql scripts # we will set template path for sql scripts
if ver >= 90300: if ver >= 90300:
@ -559,8 +567,8 @@ class DatabaseView(PGChildNodeView):
if 'name' not in data: if 'name' not in data:
data['name'] = data['old_name'] data['name'] = data['old_name']
status = self.manager.release(did=did)
for action in ["rename_database", "tablespace"]: for action in ["rename_database", "tablespace"]:
status = self.manager.release(did=did)
SQL = self.get_offline_sql(gid, sid, data, did, action) SQL = self.get_offline_sql(gid, sid, data, did, action)
SQL = SQL.strip('\n').strip(' ') SQL = SQL.strip('\n').strip(' ')
if SQL and SQL != "": if SQL and SQL != "":
@ -568,21 +576,6 @@ class DatabaseView(PGChildNodeView):
if not status: if not status:
return internal_server_error(errormsg=msg) return internal_server_error(errormsg=msg)
if not self.manager.db_info[did]['datallowconn']:
return jsonify(
node=self.blueprint.generate_browser_node(
did,
sid,
data['name'],
"icon-database-not-connected",
connected=False,
allowConn=False
)
)
self.conn = self.manager.connection(database=data['name'], auto_reconnect=True)
status, errmsg = self.conn.connect()
SQL = self.get_online_sql(gid, sid, data, did) SQL = self.get_online_sql(gid, sid, data, did)
SQL = SQL.strip('\n').strip(' ') SQL = SQL.strip('\n').strip(' ')
if SQL and SQL != "": if SQL and SQL != "":
@ -595,7 +588,11 @@ class DatabaseView(PGChildNodeView):
did, did,
sid, sid,
data['name'], data['name'],
"pg-icon-{0}".format(self.node_type) "pg-icon-{0}".format(self.node_type) if
self._db['datallowconn'] else
"icon-database-not-connected",
connected=False,
allowConn=False
) )
) )
@ -639,7 +636,6 @@ class DatabaseView(PGChildNodeView):
return internal_server_error(errormsg=msg) return internal_server_error(errormsg=msg)
return make_json_response(success=1) return make_json_response(success=1)
@check_precondition(action="msql") @check_precondition(action="msql")
@ -693,8 +689,7 @@ class DatabaseView(PGChildNodeView):
for action in ["rename_database", "tablespace"]: for action in ["rename_database", "tablespace"]:
SQL += self.get_offline_sql(gid, sid, data, did, action) SQL += self.get_offline_sql(gid, sid, data, did, action)
if rset['rows'][0]['datallowconn']: SQL += self.get_online_sql(gid, sid, data, did)
SQL += self.get_online_sql(gid, sid, data, did)
else: else:
SQL += self.get_new_sql(gid, sid, data, did) SQL += self.get_new_sql(gid, sid, data, did)
@ -715,33 +710,30 @@ class DatabaseView(PGChildNodeView):
acls = [] acls = []
SQL_acl = '' SQL_acl = ''
if ('datallowconn' in data and data['datallowconn']) or \ try:
'datallowconn' not in data: acls = render_template(
try: "/".join([self.template_path, 'allowed_privs.json'])
acls = render_template( )
"/".join([self.template_path, 'allowed_privs.json']) acls = json.loads(acls, encoding='utf-8')
except Exception as e:
current_app.logger.exception(e)
# Privileges
for aclcol in acls:
if aclcol in data:
allowedacl = acls[aclcol]
data[aclcol] = parse_priv_to_db(
data[aclcol], allowedacl['acl']
) )
acls = json.loads(acls, encoding='utf-8')
except Exception as e:
current_app.logger.exception(e)
# Privileges SQL_acl = render_template(
for aclcol in acls: "/".join([self.template_path, 'grant.sql']),
if aclcol in data: data=data, conn=self.conn
allowedacl = acls[aclcol] )
data[aclcol] = parse_priv_to_db(
data[aclcol], allowedacl['acl']
)
SQL_acl = render_template(
"/".join([self.template_path, 'grant.sql']),
data=data, conn=self.conn
)
conn = self.manager.connection()
SQL = render_template( SQL = render_template(
"/".join([self.template_path, 'create.sql']), "/".join([self.template_path, 'create.sql']),
data=data, conn=conn data=data, conn=self.conn
) )
SQL += "\n" SQL += "\n"
SQL += SQL_acl SQL += SQL_acl
@ -847,46 +839,44 @@ class DatabaseView(PGChildNodeView):
if not status: if not status:
return internal_server_error(errormsg=res) return internal_server_error(errormsg=res)
if res['rows'][0]['datallowconn']: SQL = render_template(
SQL = render_template( "/".join([self.template_path, 'acl.sql']),
"/".join([self.template_path, 'acl.sql']), did=did, conn=self.conn
did=did, conn=self.conn )
) status, dataclres = self.conn.execute_dict(SQL)
status, dataclres = self.conn.execute_dict(SQL) if not status:
if not status: return internal_server_error(errormsg=dataclres)
return internal_server_error(errormsg=dataclres) res = self.formatdbacl(res, dataclres['rows'])
res = self.formatdbacl(res, dataclres['rows'])
SQL = render_template( SQL = render_template(
"/".join([self.template_path, 'defacl.sql']), "/".join([self.template_path, 'defacl.sql']),
did=did, conn=self.conn did=did, conn=self.conn
) )
status, defaclres = self.conn.execute_dict(SQL) status, defaclres = self.conn.execute_dict(SQL)
if not status: if not status:
return internal_server_error(errormsg=defaclres) return internal_server_error(errormsg=defaclres)
res = self.formatdbacl(res, defaclres['rows']) res = self.formatdbacl(res, defaclres['rows'])
result = res['rows'][0] result = res['rows'][0]
if result['datallowconn']: SQL = render_template(
SQL = render_template( "/".join([self.template_path, 'get_variables.sql']),
"/".join([self.template_path, 'get_variables.sql']), did=did, conn=self.conn
did=did, conn=self.conn )
) status, res1 = self.conn.execute_dict(SQL)
status, res1 = self.conn.execute_dict(SQL) if not status:
if not status: return internal_server_error(errormsg=res1)
return internal_server_error(errormsg=res1)
# Get Formatted Security Labels # Get Formatted Security Labels
if 'seclabels' in result: if 'seclabels' in result:
# Security Labels is not available for PostgreSQL <= 9.1 # Security Labels is not available for PostgreSQL <= 9.1
frmtd_sec_labels = parse_sec_labels_from_db(result['seclabels']) frmtd_sec_labels = parse_sec_labels_from_db(result['seclabels'])
result.update(frmtd_sec_labels) result.update(frmtd_sec_labels)
# Get Formatted Variables # Get Formatted Variables
frmtd_variables = parse_variables_from_db(res1['rows']) frmtd_variables = parse_variables_from_db(res1['rows'])
result.update(frmtd_variables) result.update(frmtd_variables)
sql_header = "-- Database: {0}\n\n-- ".format(result['name']) sql_header = "-- Database: {0}\n\n-- ".format(result['name'])
if hasattr(str, 'decode'): if hasattr(str, 'decode'):