Issue #6: Handle exceptions gracefully

dnaeon · Apr 15, 2015 · d68995d · d68995d
1 parent dce1fb1
commit d68995d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/pvc/core.py b/src/pvc/core.py
@@ -27,6 +27,8 @@
 
 """
 
+import pyVmomi
+
 import requests
 requests.packages.urllib3.disable_warnings()
 
@@ -129,9 +131,14 @@ def login(self):
                 self.dialog.set_background_title(background_title)
                 return True
             except Exception as e:
+                if isinstance(e, pyVmomi.vim.MethodFault):
+                    msg = e.msg
+                else:
+                    msg = e
+
                 self.dialog.msgbox(
                     title='Login failed',
-                    text='Failed to login to {}\n\n{}\n'.format(self.agent.host, e.msg)
+                    text='Failed to login to {}\n\n{}\n'.format(self.agent.host, msg)
                 )
 
     def run(self):