CondaInterface/CondaExecDialog.py

# -*- coding: utf-8 -*-

# Copyright (c) 2019 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing a dialog to show the output of a conda execution.
"""

from __future__ import unicode_literals
try:
    str = unicode
except NameError:
    pass

import json

from PyQt5.QtCore import pyqtSlot, QProcess, QTimer
from PyQt5.QtWidgets import QDialog, QDialogButtonBox, QAbstractButton

from E5Gui import E5MessageBox

from .Ui_CondaExecDialog import Ui_CondaExecDialog

import Preferences
from Globals import dataString


class CondaExecDialog(QDialog, Ui_CondaExecDialog):
    """
    Class implementing a dialog to show the output of a conda execution.
    """
    def __init__(self, configuration, venvManager, parent=None):
        """
        Constructor
        
        @param configuration dictionary containing the configuration parameters
            as returned by the command configuration dialog
        @type dict
        @param venvManager reference to the virtual environment manager
        @type VirtualenvManager
        @param parent reference to the parent widget
        @type QWidget
        """
        super(CondaExecDialog, self).__init__(parent)
        self.setupUi(self)
        
        self.__finishCommandMethods = {
            "create": self.__finishCreate,
        }
        
        self.__venvName = configuration["logicalName"]
        self.__venvManager = venvManager
        self.__condaCommand = configuration["command"]
        
        self.__process = None
        self.__condaExe = Preferences.getConda("CondaExecutable")
        if not self.__condaExe:
            self.__condaExe = "conda"
    
    @pyqtSlot(QAbstractButton)
    def on_buttonBox_clicked(self, button):
        """
        Private slot called by a button of the button box clicked.
        
        @param button button that was clicked
        @type QAbstractButton
        """
        if button == self.buttonBox.button(QDialogButtonBox.Close):
            self.accept()
        elif button == self.buttonBox.button(QDialogButtonBox.Cancel):
            self.__finish(1, 0)
    
    def start(self, arguments):
        """
        Public slot to start the conda command.
        
        @param arguments commandline arguments for conda program
        @type list of str
        """
        self.errorGroup.hide()
        self.progressLabel.hide()
        self.progressBar.hide()
        
        self.contents.clear()
        self.errors.clear()
        self.progressLabel.clear()
        self.progressBar.setValue(0)
        
        self.__bufferedStdout = None
        self.__json = "--json" in arguments
        self.__firstProgress = True
        self.__lastFetchFile = ""
        
        self.__process = QProcess()
        self.__process.readyReadStandardOutput.connect(self.__readStdout)
        self.__process.readyReadStandardError.connect(self.__readStderr)
        self.__process.finished.connect(self.__finish)
        
        self.__process.start(self.__condaExe, arguments)
        procStarted = self.__process.waitForStarted(5000)
        if not procStarted:
            E5MessageBox.critical(
                self,
                self.tr("Conda Execution"),
                self.tr("""The conda executable could not be started. Is it"""
                        """ configured correctly?"""))
            self.__finish(1, 0)
        else:
            self.__logOutput(self.tr("Operation started.\n"))
    
    def __finish(self, exitCode, exitStatus, giveUp=False):
        """
        Private slot called when the process finished.
        
        It is called when the process finished or
        the user pressed the button.
        
        @param exitCode exit code of the process (integer)
        @param exitStatus exit status of the process (QProcess.ExitStatus)
        @keyparam giveUp flag indicating to not start another attempt (boolean)
        """
        if self.__process is not None and \
           self.__process.state() != QProcess.NotRunning:
            self.__process.terminate()
            QTimer.singleShot(2000, self.__process.kill)
            self.__process.waitForFinished(3000)
        
        self.buttonBox.button(QDialogButtonBox.Close).setEnabled(True)
        self.buttonBox.button(QDialogButtonBox.Cancel).setEnabled(False)
        self.buttonBox.button(QDialogButtonBox.Close).setDefault(True)
        
        self.progressLabel.hide()
        self.progressBar.hide()
        
        self.__logOutput(self.tr("Operation finished.\n"))
        if self.__json:
            if self.__bufferedStdout:
                index = self.__bufferedStdout.find("{")
                rindex = self.__bufferedStdout.rfind("}")
                self.__bufferedStdout = self.__bufferedStdout[index:rindex + 1]
                try:
                    jsonDict = json.loads(self.__bufferedStdout)
                except Exception as error:
                    self.__logError(str(error))
                    return
                
                if "error" in jsonDict:
                    self.__logError(jsonDict["error"])
                elif "success" in jsonDict and jsonDict["success"]:
                    self.__finishCommandMethods[self.__condaCommand](jsonDict)
                else:
                    self.__logError(
                        self.tr("Conda command '{0}' did not return success.")
                        .format(self.__condaCommand))
        elif self.__bufferedStdout:
            self.__logOutput(self.__bufferedStdout)
    
    def __finishCreate(self, resultDict):
        """
        Private method to finish the 'create' command.
        
        @param resultDict dictionary containing the 'create' result data
        @type dict
        """
        if "actions" in resultDict and \
                "PREFIX" in resultDict["actions"]:
            prefix = resultDict["actions"]["PREFIX"]
        elif "prefix" in resultDict:
            prefix = resultDict["prefix"]
        elif "dst_prefix" in resultDict:
            prefix = resultDict["dst_prefix"]
        else:
            prefix = ""
        self.__venvManager.addVirtualEnv(self.__venvName,
                                         prefix,
                                         isConda=True)
    
    def __setProgressValues(self, jsonDict, progressType):
        """
        Private method to set the value of the progress bar.
        
        @param jsonDict dictionary containing the progress info
        @type dict
        @param progressType action type to check for
        @type str
        @return flag indicating success
        @rtype bool
        """
        if progressType in jsonDict and "progress" in jsonDict:
            if jsonDict["maxval"] == 1:
                self.progressBar.setMaximum(100)
                # percent values
                self.progressBar.setValue(
                    int(jsonDict["progress"] * 100))
                parts = jsonDict["fetch"].split("|")
                filename = parts[0].strip()
                filesize = parts[1].strip()
            else:
                self.progressBar.setMaximum(jsonDict["maxval"])
                self.progressBar.setValue(jsonDict["progress"])
                filename = jsonDict["fetch"].strip()
                filesize = dataString(int(jsonDict["maxval"]))
            
            self.progressLabel.setText(
                self.tr("{0} (Size: {1})").format(filename, filesize))
            
            if progressType == "fetch":
                if filename != self.__lastFetchFile:
                    self.__logOutput(
                        self.tr("Fetching {0} ...").format(filename))
                    self.__lastFetchFile = filename
                elif jsonDict["finished"]:
                    self.__logOutput(self.tr("Done.\n"))
            
            if self.__firstProgress:
                self.progressLabel.show()
                self.progressBar.show()
                self.__firstProgress = False
            
            return True
        
        return False
    
    def __readStdout(self):
        """
        Private slot to handle the readyReadStandardOutput signal.
        
        It reads the output of the process, formats it and inserts it into
        the contents pane.
        """
        all_stdout = str(self.__process.readAllStandardOutput(),
                         Preferences.getSystem("IOEncoding"),
                         'replace')
        all_stdout = all_stdout.replace("\x00", "")
        if self.__json:
            for stdout in all_stdout.splitlines():
                try:
                    jsonDict = json.loads(stdout.replace("\x00", "").strip())
                    if self.__setProgressValues(jsonDict, "fetch"):
                        # nothing to do anymore
                        pass
                    elif "progress" not in jsonDict:
                        if self.__bufferedStdout is None:
                            self.__bufferedStdout = stdout
                        else:
                            self.__bufferedStdout += stdout
                except (TypeError, ValueError):
                    if self.__bufferedStdout is None:
                        self.__bufferedStdout = stdout
                    else:
                        self.__bufferedStdout += stdout
        else:
            self.__logOutput(all_stdout)
    
    def __readStderr(self):
        """
        Private slot to handle the readyReadStandardError signal.
        
        It reads the error output of the process and inserts it into the
        error pane.
        """
        self.__process.setReadChannel(QProcess.StandardError)
        
        while self.__process.canReadLine():
            stderr = str(self.__process.readLine(),
                         Preferences.getSystem("IOEncoding"),
                         'replace')
            self.__logError(stderr)
    
    def __logOutput(self, stdout):
        """
        Private method to log some output.
        
        @param stdout output string to log
        @type str
        """
        self.contents.insertPlainText(stdout)
        self.contents.ensureCursorVisible()
    
    def __logError(self, stderr):
        """
        Private method to log an error.
        
        @param stderr error string to log
        @type str
        """
        self.errorGroup.show()
        self.errors.insertPlainText(stderr)
        self.errors.ensureCursorVisible()