Capture connection error , for #16

rundeck-plugins · Dec 27, 2018 · fca7a50 · fca7a50
1 parent 08775e2
commit fca7a50
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/contents/winrm_session.py b/contents/winrm_session.py
@@ -122,14 +122,20 @@ def __init__(self, session, shell, command ):
         self.shell = shell
 
     def get_response(self):
-        if self.shell == "cmd":
-            response = self.session.run_cmd(self.exec_command, out_stream=self.o_stream, err_stream=self.e_stream)
-            self.o_std = response.std_out
-            self.e_std = response.std_err
-            self.stat = response.status_code
-
-        if self.shell == "powershell":
-            response = self.session.run_ps(self.exec_command, out_stream=self.o_stream, err_stream=self.e_stream)
-            self.o_std = response.std_out
-            self.e_std = response.std_err
-            self.stat = response.status_code
+        try:
+            if self.shell == "cmd":
+                response = self.session.run_cmd(self.exec_command, out_stream=self.o_stream, err_stream=self.e_stream)
+                self.o_std = response.std_out
+                self.e_std = response.std_err
+                self.stat = response.status_code
+
+            if self.shell == "powershell":
+                response = self.session.run_ps(self.exec_command, out_stream=self.o_stream, err_stream=self.e_stream)
+                self.o_std = response.std_out
+                self.e_std = response.std_err
+                self.stat = response.status_code
+
+        except Exception as e:
+            print(e)
+            self.e_std = e
+            self.stat=-1