X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?p=genesis.git;a=blobdiff_plain;f=fuel%2Fdeploy%2Fssh_client.py;fp=fuel%2Fdeploy%2Fssh_client.py;h=0000000000000000000000000000000000000000;hp=0f6b8c7eaea6e74175985cc0718946234809059f;hb=e94d5484ecdf8021245d961479dae07ea2f8321f;hpb=563547b4a9f44090f32c0e17d040114854563760 diff --git a/fuel/deploy/ssh_client.py b/fuel/deploy/ssh_client.py deleted file mode 100644 index 0f6b8c7..0000000 --- a/fuel/deploy/ssh_client.py +++ /dev/null @@ -1,97 +0,0 @@ -############################################################################### -# Copyright (c) 2015 Ericsson AB and others. -# szilard.cserey@ericsson.com -# All rights reserved. This program and the accompanying materials -# are made available under the terms of the Apache License, Version 2.0 -# which accompanies this distribution, and is available at -# http://www.apache.org/licenses/LICENSE-2.0 -############################################################################### - - -import paramiko -import common -import scp - -TIMEOUT = 600 -log = common.log -err = common.err - - -class SSHClient(object): - - def __init__(self, host, username, password): - self.host = host - self.username = username - self.password = password - self.client = None - - def open(self, timeout=TIMEOUT): - self.client = paramiko.SSHClient() - self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) - self.client.connect(self.host, username=self.username, - password=self.password, look_for_keys=False, - timeout=timeout) - - def close(self): - if self.client is not None: - self.client.close() - self.client = None - - def __enter__(self): - self.open() - return self - - def __exit__(self, type, value, traceback): - self.close() - - def exec_cmd(self, command, check=True, sudo=False, timeout=TIMEOUT): - if sudo and self.username != 'root': - command = "sudo -S -p '' %s" % command - stdin, stdout, stderr = self.client.exec_command(command, - timeout=timeout) - if sudo: - stdin.write(self.password + '\n') - stdin.flush() - response = stdout.read().strip() - error = stderr.read().strip() - - if check: - if error: - self.close() - err(error) - else: - return response - return response, error - - def run(self, command): - transport = self.client.get_transport() - transport.set_keepalive(1) - chan = transport.open_session() - chan.exec_command(command) - while not chan.exit_status_ready(): - if chan.recv_ready(): - data = chan.recv(1024) - while data: - log(data.strip()) - data = chan.recv(1024) - - if chan.recv_stderr_ready(): - error_buff = chan.recv_stderr(1024) - while error_buff: - log(error_buff.strip()) - error_buff = chan.recv_stderr(1024) - return chan.recv_exit_status() - - def scp_get(self, remote, local='.', dir=False): - try: - with scp.SCPClient(self.client.get_transport()) as _scp: - _scp.get(remote, local, dir) - except Exception as e: - err(e) - - def scp_put(self, local, remote='.', dir=False): - try: - with scp.SCPClient(self.client.get_transport()) as _scp: - _scp.put(local, remote, dir) - except Exception as e: - err(e)