get_db() only assigns the db after authentication (#257)

This commit is contained in:
Ross Lawley 2013-04-30 15:05:41 +00:00
parent 9c1cd81adb
commit 68f760b563

View File

@ -137,11 +137,12 @@ def get_db(alias=DEFAULT_CONNECTION_NAME, reconnect=False):
if alias not in _dbs: if alias not in _dbs:
conn = get_connection(alias) conn = get_connection(alias)
conn_settings = _connection_settings[alias] conn_settings = _connection_settings[alias]
_dbs[alias] = conn[conn_settings['name']] db = conn[conn_settings['name']]
# Authenticate if necessary # Authenticate if necessary
if conn_settings['username'] and conn_settings['password']: if conn_settings['username'] and conn_settings['password']:
_dbs[alias].authenticate(conn_settings['username'], db.authenticate(conn_settings['username'],
conn_settings['password']) conn_settings['password'])
_dbs[alias] = db
return _dbs[alias] return _dbs[alias]