diff options
author | Yong Liu <yong.liu@intel.com> | 2014-09-10 13:25:52 +0800 |
---|---|---|
committer | Yong Liu <yong.liu@intel.com> | 2014-09-10 13:25:52 +0800 |
commit | ddbe46a51d4d5352737a1077b25cfe283a4d1fb2 (patch) | |
tree | 99d53ffb5bb5c1aba8c8b4fda39a9eece3763a0b /framework/ssh_connection.py |
import dcts first version
Diffstat (limited to 'framework/ssh_connection.py')
-rw-r--r-- | framework/ssh_connection.py | 39 |
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) |