Return exit code of executed commands

Signed-off-by: Jonatan Schlag <jonatan.schlag@ipfire.org>

diff --git a/test.py b/test.py
index 54d256df85da7767846bd00fa4aac483bb5c7e02..d5be079ad1d5f72b78e180c077bd636e3b899bb5 100755 (executable)
--- a/test.py
+++ b/test.py
@@ -355,7 +355,7 @@ class connection():
         self.con.flush()
 
     def command(self, command):
-        self.write("{}\n".format(command))
+        self.write("{}; echo \"END: $?\"\n".format(command))
 
         # We need to read out the prompt for this command first
         # If we do not do this we will break the loop immediately
@@ -367,6 +367,13 @@ class connection():
             data = self.readline()
             self.log_console_line(data.decode())
 
+        # We saved our exit code in data (the last line)
+        self.log.debug(data.decode())
+        data = data.decode().replace("END: ", "")
+        self.log.debug(data)
+        self.log.debug(data.strip())
+        return data.strip()
+
 
 # A class which define and undefine a virtual network based on an xml file
 class network():