commit 2d84e5276630cc4a8a7ead5a84898d98dbed4387
author Steve McIntyre <steve.mcintyre@linaro.org> Thu Feb 12 06:44:42 2015 +0000
committer Steve McIntyre <steve.mcintyre@linaro.org> Thu Feb 12 06:44:42 2015 +0000
tree 73ce02b92d2ae9777398505ee31c997f430afd10
parent ad12cd7576e3014f095b352f416039a19d25ee2c

Handle timeouts and disconnects in switch_save_running_config() too

Change-Id: Ida782863c42d5f841e8847ac614235c180dcaead

drivers/CiscoCatalyst.py

diff --git a/drivers/CiscoCatalyst.py b/drivers/CiscoCatalyst.py
index 15b4feb..fedf865 100644
--- a/drivers/CiscoCatalyst.py
+++ b/drivers/CiscoCatalyst.py
@@ -85,7 +85,7 @@
             self.connection.expect("startup-config")
             self._cli("startup-config")
             self.connection.expect("OK")
-        except PExpectError:
+        except (PExpectError, pexpect.EOF):
             # recurse on error
             self._switch_connect()
             self.switch_save_running_config()