def desconnexio(self):
if self.db is None:
return
conName = self.db.connectionName()
self.db.close()
self.db = None
QSqlDatabase.removeDatabase(conName)
def dbLogDesconnexio(self):
if not self.intranet:
return
try:
if self.dbLog is not None:
conName = self.dbLog.connectionName()
self.dbLog.close()
self.dbLog = None
QSqlDatabase.removeDatabase(conName)
except Exception:
self.dbLog = None
def test_open_sqlite_db(self):
existing_db = get_projects_db()
random_name = ''.join(random.choice(string.lowercase) for i in range(9))
db = open_sqlite_db(existing_db.databaseName(), random_name)
try:
self.assertEqual(db.databaseName(), existing_db.databaseName())
except:
raise
finally:
db.close()
QSqlDatabase.removeDatabase(random_name)
def get_db(self):
if QSqlDatabase.contains("midb"):
QSqlDatabase.removeDatabase("midb")
db = QSqlDatabase.addDatabase("QPSQL", "midb")
db.setHostName('localhost')
db.setPort(5432)
db.setDatabaseName('')
db.setUserName('')
db.setPassword('')
return db
def dbDisconnect():
if not QvNovageo.app.intranet:
return
try:
if QvNovageo.db is not None:
name = QvNovageo.db.connectionName()
QvNovageo.db.close()
QvNovageo.db = None
QSqlDatabase.removeDatabase(name)
except Exception:
QvNovageo.db = None
def get_db(self, cfg):
if QSqlDatabase.contains("midb"):
if self.db:
if self.db.isOpen():
self.db.close()
QSqlDatabase.removeDatabase("midb")
self.db = QSqlDatabase.addDatabase("QPSQL", "midb")
self.db.setHostName(cfg.get('host', 'localhost'))
self.db.setPort(int(cfg.get('port', 5432)))
self.db.setDatabaseName(cfg.get('dbname', ''))
self.db.setUserName(cfg.get('user', ''))
self.db.setPassword(cfg.get('pswd', ''))
self.query = QSqlQuery(self.db)
