split-secret/scripts/classes/Cli.py

# This class offers easy logik to execute cli commands
# @author Kevin Veen-Birkenbach [kevin@veen.world]

import subprocess

class Cli(object):
    
    def __init__(self):
        self.command = ''
        self.output  = []
        pass
    
    def executeCommand(self,command):
        self.command = command
        process = subprocess.Popen([command], stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
        out, err = process.communicate()
        stdout = out.splitlines()
        self.output = []
        for line in stdout:
            self.output.append(line.decode("utf-8"))
        if process.wait() > bool(0):
            raise Exception("Error for: \nCommand:<<" + command + ">>\nOutput:<<" + out + ">>\nExitcode:<<" + err + ">>")
        return self.output
    
    def getOutputString(self):
        return str('\n'.join(self.output))

    def getCommandString(self):
        return self.command