def _connectorFor(self, databaseName=None):
if databaseName is None:
databaseName = self.databaseName
if self.spawnedDBUser:
dsn = "{}:dbname={}:{}".format(self.host, databaseName,
self.spawnedDBUser)
elif self.uid is not None:
dsn = "{}:dbname={}:{}".format(self.host, databaseName,
pwd.getpwuid(self.uid).pw_name)
else:
dsn = "{}:dbname={}".format(self.host, databaseName)
kwargs = {}
if self.port:
kwargs["host"] = "{}:{}".format(self.host, self.port)
return DBAPIConnector(pgdb, postgresPreflight, dsn, **kwargs)
def pgConnectorFromConfig(config):
"""
Create a postgres DB-API connector from the given configuration.
"""
import pgdb
return DBAPIConnector(pgdb, postgresPreflight, config.DSN).connect