diff options
| -rw-r--r-- | resources/libraries/python/ssh.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/resources/libraries/python/ssh.py b/resources/libraries/python/ssh.py index ad5fb27b85..a94eec4e91 100644 --- a/resources/libraries/python/ssh.py +++ b/resources/libraries/python/ssh.py @@ -67,6 +67,17 @@ class SSH(object): format(self._ssh.get_transport().getpeername())) logger.debug('Connections: {0}'.format(str(SSH.__existing_connections))) + def disconnect(self, node): + """Close SSH connection to the node. + + :param node: The node to disconnect from. + :type node: dict + """ + node_hash = self._node_hash(node) + if node_hash in SSH.__existing_connections: + ssh = SSH.__existing_connections.pop(node_hash) + ssh.close() + def exec_command(self, cmd, timeout=10): """Execute SSH command on a new channel on the connected Node. |