1 files changed, 39 insertions, 0 deletions
diff --git a/framework/ssh_connection.py b/framework/ssh_connection.py
new file mode 100644
index 0000000..dfdad2f
--- /dev/null
+++ b/framework/ssh_connection.py
@@ -0,0 +1,39 @@
+# <COPYRIGHT_TAG>
+
+from ssh_pexpect import SSHPexpect
+from settings import USERNAME
+
+
+class SSHConnection(object):
+
+    """
+    Module for create session to host.
+    Implement send_expect/copy function upper SSHPexpet module.
+    """
+
+    def __init__(self, host, session_name):
+        self.session = SSHPexpect(host, USERNAME)
+        self.name = session_name
+
+    def init_log(self, logger):
+        self.logger = logger
+        self.logger.config_execution(self.name)
+        self.session.init_log(logger, self.name)
+
+    def send_expect(self, cmds, expected, timeout=15):
+        self.logger.info(cmds)
+        out = self.session.send_expect(cmds, expected, timeout)
+        self.logger.debug(out)
+        return out
+
+    def close(self):
+        self.session.close()
+
+    def isalive(self):
+        return self.session.isalive()
+
+    def copy_file_from(self, filename, password=''):
+        self.session.copy_file_from(filename, password)
+
+    def copy_file_to(self, filename, password=''):
+        self.session.copy_file_to(filename, password)