Example #1
0
    def __init__(self, mWindow):
        super(TrayIcon, self).__init__(mWindow)
        self.setIcon(self._icon(config.mainIcon))
        self.setVisible(True)

        self._mainWindow = mWindow
        self._mainWindow.setWindowIcon(self._icon("main_logo_color.png"))

        self._doHandler = DOHandler()
        self._updateChecker = UpdateChecker(self._mainWindow)
        self._updateChecker.quitProgram.connect(self._quit)

        self._data = []
        self._dInfos = []
        self._menu = None

        self._quitAction = QtWidgets.QAction("Quit", self)
        self._quitAction.setIcon(self._icon("quit.png"))
        self._quitAction.triggered.connect(self._quit)

        self._aboutAction = QtWidgets.QAction("About", self)
        self._aboutAction.setIcon(self._icon("about.png"))
        self._aboutAction.triggered.connect(self._about)

        self._settingsAction = QtWidgets.QAction("Preferences", self)
        self._settingsAction.setIcon(self._icon("settings.png"))
        self._settingsAction.triggered.connect(self._settings)

        self._createAction = QtWidgets.QAction("Create droplet (web)", self)
        self._createAction.setIcon(self._icon("create.png"))
        self._createAction.triggered.connect(self._createDroplet)

        self._bugAction = QtWidgets.QAction("Report bug/request", self)
        self._bugAction.setIcon(self._icon("bug.png"))
        self._bugAction.triggered.connect(self._report)

        self._updateAction = QtWidgets.QAction("Check for update", self)
        self._updateAction.setIcon(self._icon("update.png"))
        self._updateAction.triggered.connect(self.update)

        self._helpMenu = QtWidgets.QMenu(self._mainWindow)
        self._helpMenu.addAction(self._updateAction)
        self._helpMenu.addAction(self._bugAction)
        self._helpMenu.addAction(self._aboutAction)

        self._helpAction = QtWidgets.QAction("Help", self)
        self._helpAction.setIcon(self._icon("help.png"))
        self._helpAction.setMenu(self._helpMenu)

        self._updateThread = UpdateThread(self)
        self._updateThread.updated.connect(self._updateData)
        self._updateThread.start()

        clientId = config.value('clientId', "")
        apiKey = config.value('apiKey', "")

        self.activated.connect(self._popupMenu)

        if not clientId or not apiKey:
            self._settings()

        self._updateMenu()
Example #2
0
class TrayIcon(QtWidgets.QSystemTrayIcon):
    def __init__(self, mWindow):
        super(TrayIcon, self).__init__(mWindow)
        self.setIcon(self._icon(config.mainIcon))
        self.setVisible(True)

        self._mainWindow = mWindow
        self._mainWindow.setWindowIcon(self._icon("main_logo_color.png"))

        self._doHandler = DOHandler()
        self._updateChecker = UpdateChecker(self._mainWindow)
        self._updateChecker.quitProgram.connect(self._quit)

        self._data = []
        self._dInfos = []
        self._menu = None

        self._quitAction = QtWidgets.QAction("Quit", self)
        self._quitAction.setIcon(self._icon("quit.png"))
        self._quitAction.triggered.connect(self._quit)

        self._aboutAction = QtWidgets.QAction("About", self)
        self._aboutAction.setIcon(self._icon("about.png"))
        self._aboutAction.triggered.connect(self._about)

        self._settingsAction = QtWidgets.QAction("Preferences", self)
        self._settingsAction.setIcon(self._icon("settings.png"))
        self._settingsAction.triggered.connect(self._settings)

        self._createAction = QtWidgets.QAction("Create droplet (web)", self)
        self._createAction.setIcon(self._icon("create.png"))
        self._createAction.triggered.connect(self._createDroplet)

        self._bugAction = QtWidgets.QAction("Report bug/request", self)
        self._bugAction.setIcon(self._icon("bug.png"))
        self._bugAction.triggered.connect(self._report)

        self._updateAction = QtWidgets.QAction("Check for update", self)
        self._updateAction.setIcon(self._icon("update.png"))
        self._updateAction.triggered.connect(self.update)

        self._helpMenu = QtWidgets.QMenu(self._mainWindow)
        self._helpMenu.addAction(self._updateAction)
        self._helpMenu.addAction(self._bugAction)
        self._helpMenu.addAction(self._aboutAction)

        self._helpAction = QtWidgets.QAction("Help", self)
        self._helpAction.setIcon(self._icon("help.png"))
        self._helpAction.setMenu(self._helpMenu)

        self._updateThread = UpdateThread(self)
        self._updateThread.updated.connect(self._updateData)
        self._updateThread.start()

        clientId = config.value('clientId', "")
        apiKey = config.value('apiKey', "")

        self.activated.connect(self._popupMenu)

        if not clientId or not apiKey:
            self._settings()

        self._updateMenu()

    def _icon(self, filename):
        return QtGui.QIcon(rPath(filename))

    def _dropletMenu(self, idx):
        dropletMenu = QtWidgets.QMenu(self._mainWindow)

        copyIPAction = QtWidgets.QAction("Copy IP to clipboard", self._menu)
        copyIPAction.setIcon(self._icon("ip.png"))
        copyIPAction.triggered.connect(lambda y, x=idx: self._ipToClipboard(x))

        resetRootAction = QtWidgets.QAction("Reset root password", self._menu)
        resetRootAction.setIcon(self._icon("password.png"))
        resetRootAction.triggered.connect(lambda y, x=idx: self._resetRoot(x))

        renameDropletAction = QtWidgets.QAction("Rename", self._menu)
        renameDropletAction.setIcon(self._icon("rename.png"))
        renameDropletAction.triggered.connect(lambda y, x=idx: self._renameDroplet(x))

        destroyDropletAction = QtWidgets.QAction("Destroy", self._menu)
        destroyDropletAction.setIcon(self._icon("destroy.png"))
        destroyDropletAction.triggered.connect(lambda y, x=idx: self._destroyDroplet(x))

        dropletMenu.addAction(copyIPAction)

        if self._dInfos[idx]['status'] == 'active':

            sshAction = QtWidgets.QAction("Open SSH connection", self._menu)
            sshAction.setIcon(self._icon("ssh.png"))
            sshAction.triggered.connect(lambda y, x=idx: self._openSSH(x))

            rebootAction = QtWidgets.QAction("Power cycle", self._menu)
            rebootAction.setIcon(self._icon("reboot.png"))
            rebootAction.triggered.connect(lambda y, x=idx: self._powerCycle(x))

            shutDownAction = QtWidgets.QAction("Power Off", self._menu)
            shutDownAction.setIcon(self._icon("shutdown.png"))
            shutDownAction.triggered.connect(lambda y, x=idx: self._powerOff(x))

            dropletMenu.addAction(sshAction)
            dropletMenu.addSeparator()
            dropletMenu.addAction(renameDropletAction)
            dropletMenu.addAction(resetRootAction)
            dropletMenu.addAction(rebootAction)
            dropletMenu.addAction(shutDownAction)

        else:
            startAction = QtWidgets.QAction("Power On", self._menu)
            startAction.setIcon(self._icon("start.png"))
            startAction.triggered.connect(lambda y, x=idx: self._powerOn(x))

            dropletMenu.addSeparator()
            dropletMenu.addAction(renameDropletAction)
            dropletMenu.addAction(resetRootAction)
            dropletMenu.addAction(startAction)

        dropletMenu.addSeparator()
        dropletMenu.addAction(destroyDropletAction)

        return dropletMenu

    def _checkResult(self, commandName, dropletName, result):
        if result['status'] != "OK":
            if 'message' in result:
                if 'pending' in result['message']:
                    msg = "%s is in pending stage. Please try again later" % dropletName
                else:
                    msg = result['message']
                self._message(msg)

    def _renameDroplet(self, idx):
        dropletName = self._dInfos[idx]['name']
        dropletId = self._dInfos[idx]['id']
        rd = RenameDialog(dropletName, self._mainWindow)
        rd.showNormal()
        rd.activateWindow()
        rd.exec_()
        if rd.result:
            result = self._doHandler.rename(dropletId, rd.result)
            self._checkResult("Rename", dropletName, result)

    def _resetRoot(self, idx):
        dropletName = self._dInfos[idx]['name']
        action = "reboot" if self._dInfos[idx]['status'] == 'active' else "power on"
        result = self._question("Reset root password command will %s the %s. Proceed?" % (action, dropletName))
        if result:
            dropletId = self._dInfos[idx]['id']
            result = self._doHandler.resetRoot(dropletId)
            self._checkResult("Reset root", dropletName, result)

    def _openSSH(self, idx):
        userName = config.value('userName', "root")
        ipAddress = self._dInfos[idx]['ip_address']
        sshPort = config.value('sshPort', 22)
        command = config.sshCommand
        if platform == "win32":
            command = command % (userName, ipAddress, sshPort)
            DETACHED_PROCESS = 0x00000008
            subprocess.Popen(command.split(' '), creationflags=DETACHED_PROCESS)
        else:
            command = command % (sshPort, userName, ipAddress)
            os.system(command)

    def _powerOn(self, idx):
        dropletName = self._dInfos[idx]['name']
        dropletId = self._dInfos[idx]['id']
        result = self._doHandler.powerOn(dropletId)
        self._checkResult("Power on", dropletName, result)

    def _powerCycle(self, idx):
        dropletName = self._dInfos[idx]['name']
        result = self._question("Are you sure that you want to power cycle the %s?" % dropletName)
        if result:
            dropletId = self._dInfos[idx]['id']
            result = self._doHandler.powerCycle(dropletId)
            self._checkResult("Power cycle", dropletName, result)

    def _destroyDroplet(self, idx):
        dropletName = self._dInfos[idx]['name']
        result = self._question("Are you sure that you want to DESTROY the %s? This command is irreversible!" % dropletName)
        if result:
            dropletName = self._dInfos[idx]['name']
            dropletId = self._dInfos[idx]['id']
            result = self._doHandler.destroy(dropletId)
            self._checkResult("Destroy", dropletName, result)

    def _powerOff(self, idx):
        dropletName = self._dInfos[idx]['name']
        result = self._question("Are you sure that you want to power off the %s?" % dropletName)
        if result:
            dropletName = self._dInfos[idx]['name']
            dropletId = self._dInfos[idx]['id']
            result = self._doHandler.powerOff(dropletId)
            self._checkResult("Power off", dropletName, result)

    def _updateData(self, result):
        if self._data != result:
            self._dInfos = []
            self._data = result
            if self._data and 'droplets' in self._data:
                self._dInfos = self._data['droplets']
            self._updateMenu()
        gc.collect()

    def _updateMenu(self):
        menuVisible = False
        if self._menu:
            self._menu.removeAction(self._createAction)
            self._menu.removeAction(self._settingsAction)
            self._menu.removeAction(self._quitAction)
            self._menu.removeAction(self._aboutAction)
            self._menu.removeAction(self._updateAction)
            self._menu.removeAction(self._helpAction)
            menuVisible = self._menu.isVisible()
            self._menu.hide()
            self._menu.setParent(None)
            self._menu.deleteLater()

        self._menu = CustomMenu()

        if not self._data:
            infoMsg = "Connecting..."
            self.setIcon(self._icon("main_logo_gray_error.png"))
        elif "OK" in self._data['status']:
            infoMsg = "Connected"
            self.setIcon(self._icon(config.mainIcon))
        elif "ERROR" in self._data['status']:
            infoMsg = self._data['message']
            self.setIcon(self._icon("main_logo_gray_error.png"))
        else:
            infoMsg = self._data['status']
            self.setIcon(self._icon("main_logo_gray_error.png"))

        infoAction = QtWidgets.QAction("  Status: %s" % infoMsg, self._menu)
        infoAction.setEnabled(False)

        self._menu.addAction(infoAction)
        self._menu.addSeparator()

        if len(self._dInfos) > 0:
            for idx, dInfo in enumerate(self._dInfos):
                dropletAction = QtWidgets.QAction(dInfo['name'], self._menu)

                if dInfo['status'] == 'active':
                    if dInfo['locked']:
                        dropletAction.setIcon(self._icon("locked.png"))
                        dropletAction.setToolTip("Locked (operation in progress)")
                    else:
                        dropletAction.setIcon(self._icon("active.png"))
                        dropletAction.setToolTip("Running")
                else:
                    dropletAction.setIcon(self._icon("inactive.png"))
                    dropletAction.setToolTip("Stopped")

                if not dInfo['locked']:
                    dropletAction.setMenu(self._dropletMenu(idx))

                self._menu.addAction(dropletAction)

            self._menu.addSeparator()

        self._menu.addAction(self._createAction)
        self._menu.addSeparator()
        self._menu.addAction(self._settingsAction)
        self._menu.addAction(self._helpAction)
        self._menu.addSeparator()
        self._menu.addAction(self._quitAction)

        if menuVisible:
            self._popupMenu()

    def _popupMenu(self):
        if self._menu:
            if self._menu.isVisible():
                self._menu.hide()
            rect = self.geometry()
            self._menu.popup(QtCore.QPoint(rect.x(), rect.y()))
            if sys.platform == "win32":
                self._menu.activateWindow()

    def _ipToClipboard(self, idx):
        ipAddress = self._dInfos[idx]['ip_address']
        clipboard = QtWidgets.QApplication.clipboard()
        mimeData = QtCore.QMimeData()
        mimeData.setText(ipAddress)
        clipboard.setMimeData(mimeData)
        QtWidgets.QApplication.processEvents()

    def __messageBox(self, msg, mIcon):
        mBox = QtWidgets.QMessageBox(self._mainWindow)
        mBox.setWindowTitle("DO Manager")
        mBox.setText(msg)
        mBox.setWindowFlags(mBox.windowFlags() | Qt.WindowStaysOnTopHint)
        mBox.setIcon(mIcon)
        return mBox

    def _message(self, msg, error=False):
        self.showMessage("DO Manager", msg)

    def _question(self, msg):
        mBox = self.__messageBox(msg, QtWidgets.QMessageBox.Question)
        mBox.setStandardButtons(mBox.Yes | mBox.No)
        return mBox.exec_() == mBox.Yes

    def _createDroplet(self):
        os.system("%s https://cloud.digitalocean.com/droplets/new" % config.openCommand)

    def _settings(self):
        pd = PreferencesDialog(self._mainWindow)
        pd.showNormal()
        pd.activateWindow()
        pd.exec_()

    def _about(self):
        ad = AboutDialog(self._mainWindow)
        ad.showNormal()
        ad.activateWindow()
        ad.exec_()

    def _report(self):
        os.system("%s %s" % (config.openCommand, config.reportUrl))

    def update(self, silent=False):
        self._updateChecker.check(silent)

    def _quit(self):
        if sys.platform == "win32":
            self.hide()
            QtWidgets.QApplication.processEvents()
        QtWidgets.QApplication.quit()