Gracefully close the SSH session

ktbyers · Jan 17, 2016 · 0ccab2b · 0ccab2b
1 parent 34937ce
commit 0ccab2b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/netmiko/ssh_connection.py b/netmiko/ssh_connection.py
@@ -69,3 +69,10 @@ def exit_enable_mode(self, exit_command='disable'):
             if self.check_enable_mode():
                 raise ValueError("Failed to exit enable mode.")
         return output
+
+    def cleanup(self):
+        '''
+        Gracefully exit the SSH session.
+        '''
+        self.exit_config_mode()
+        self.remote_conn.sendall("exit\n")