database.py 文件源码

def connect(self):
        """ Connect to the database and set some parameters.
        """
        if Database.database is None:
            Database.database = QtSql.QSqlDatabase("QPSQL")
            Database.database.setHostName(os.environ.get(
                "DATABASE_HOST",
                settings.DB_HOST
            ))
            Database.database.setPort(int(os.environ.get(
                "DATABASE_PORT",
                settings.DB_PORT,
            )))
            Database.database.setUserName(os.environ.get(
                "DATABASE_USER",
                settings.USERNAME
            ))
            Database.database.setPassword(os.environ.get(
                "DATABASE_PASSWORD",
                settings.PASSWORD
            ))
            Database.database.setDatabaseName(os.environ.get(
                "DATABASE_NAME",
                settings.DBNAME
            ))
            if not Database.database.open():
                if rapi.utils.check_x11():
                    # We need this to create an app before opening a window.
                    import gui.utils
                    self.tmp = QtWidgets.QApplication(sys.argv)
                    gui.utils.error("Error", "Can't join database")
                print("Can't join database")
                sys.exit(1)