Skip to content

Commit

Permalink
db manager: request credentials directly when connection failed (fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jef-n committed Jan 24, 2015
1 parent f8f2713 commit 62b2c1f
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 40 deletions.
43 changes: 36 additions & 7 deletions python/plugins/db_manager/db_plugins/postgis/connector.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,8 @@
from ..connector import DBConnector
from ..plugin import ConnectionError, DbError, Table

from qgis.core import QgsCredentials

import os
import psycopg2
import psycopg2.extensions
Expand All @@ -45,23 +47,50 @@ def __init__(self, uri):

self.host = uri.host() or os.environ.get('PGHOST')
self.port = uri.port() or os.environ.get('PGPORT')
self.user = uri.username() or os.environ.get('PGUSER') or os.environ.get('USER')
self.dbname = uri.database() or os.environ.get('PGDATABASE') or self.user
self.passwd = uri.password() or os.environ.get('PGPASSWORD')

username = uri.username() or os.environ.get('PGUSER') or os.environ.get('USER')
password = uri.password() or os.environ.get('PGPASSWORD')

try:
self.connection = psycopg2.connect( self._connectionInfo().encode('utf-8') )
self.connection.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
except self.connection_error_types(), e:
raise ConnectionError(e)
err = str(e)
uri = self.uri()
conninfo = uri.connectionInfo()

for i in range(3):
(ok, username, password) = QgsCredentials.instance().get(conninfo, username, password, err)
if not ok:
raise ConnectionError(e)

if username:
uri.setUsername( username )

if password:
uri.setPassword( password )

try:
self.connection = psycopg2.connect( uri.connectionInfo().encode('utf-8') )
QgsCredentials.instance().put(conninfo, username, password)
except self.connection_error_types(), e:
if i == 2:
raise ConnectionError(e)

err = str(e)

self.connection.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)

c = self._execute(None, u"SELECT current_user")
self.user = self._fetchone(c)
self._close_cursor(c)

self._checkSpatial()
self._checkRaster()
self._checkGeometryColumnsTable()
self._checkRasterColumnsTable()

def _connectionInfo(self):
return unicode(self._uri.connectionInfo())
return unicode(self.uri().connectionInfo())

def _checkSpatial(self):
""" check whether postgis_version is present in catalog """
Expand Down Expand Up @@ -158,7 +187,7 @@ def fieldTypes(self):

def getDatabasePrivileges(self):
""" db privileges: (can create schemas, can create temp. tables) """
sql = u"SELECT has_database_privilege(%(d)s, 'CREATE'), has_database_privilege(%(d)s, 'TEMP')" % { 'd' : self.quoteString(self.dbname) }
sql = u"SELECT has_database_privilege(current_database(), 'CREATE'), has_database_privilege(current_database(), 'TEMP')"
c = self._execute(None, sql)
res = self._fetchone(c)
self._close_cursor(c)
Expand Down
34 changes: 1 addition & 33 deletions python/plugins/db_manager/db_plugins/postgis/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,6 @@

from ..html_elems import HtmlParagraph, HtmlList, HtmlTable

from qgis.core import QgsCredentials


def classFactory():
return PostGisDBPlugin

Expand Down Expand Up @@ -96,39 +93,10 @@ def connect(self, parent=None):

uri.setUseEstimatedMetadata(useEstimatedMetadata)

err = u""
try:
return self.connectToUri(uri)
except ConnectionError, e:
err = str(e)

# ask for valid credentials
max_attempts = 3
for i in range(max_attempts):
(ok, username, password) = QgsCredentials.instance().get(uri.connectionInfo(), username, password, err)

if not ok:
return False

if service != "":
uri.setConnection(service, database, username, password, sslmode)
else:
uri.setConnection(host, port, database, username, password, sslmode)

try:
self.connectToUri(uri)
except ConnectionError, e:
if i == max_attempts-1: # failed the last attempt
raise e
err = str(e)
continue

QgsCredentials.instance().put(uri.connectionInfo(), username, password)

return True

return False

return False

class PGDatabase(Database):
def __init__(self, connection, uri):
Expand Down

0 comments on commit 62b2c1f

Please sign in to comment.