fuel/deploy/ssh_client.py

   1 import paramiko
   2 import common
   3 import scp
   4
   5 TIMEOUT = 600
   6 log = common.log
   7 err = common.err
   8
   9
  10 class SSHClient(object):
  11
  12     def __init__(self, host, username, password):
  13         self.host = host
  14         self.username = username
  15         self.password = password
  16         self.client = None
  17
  18     def open(self, timeout=TIMEOUT):
  19         self.client = paramiko.SSHClient()
  20         self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  21         self.client.connect(self.host, username=self.username,
  22                             password=self.password, look_for_keys=False,
  23                             timeout=timeout)
  24
  25     def close(self):
  26         if self.client is not None:
  27             self.client.close()
  28             self.client = None
  29
  30     def __enter__(self):
  31         self.open()
  32         return self
  33
  34     def __exit__(self, type, value, traceback):
  35         self.close()
  36
  37     def exec_cmd(self, command, sudo=False, timeout=TIMEOUT, check=True):
  38         if sudo and self.username != 'root':
  39             command = "sudo -S -p '' %s" % command
  40         stdin, stdout, stderr = self.client.exec_command(command,
  41                                                          timeout=timeout)
  42         if sudo:
  43             stdin.write(self.password + '\n')
  44             stdin.flush()
  45         response = stdout.read().strip()
  46         error = stderr.read().strip()
  47
  48         if check:
  49             if error:
  50                 self.close()
  51                 err(error)
  52             else:
  53                 return response
  54         return response, error
  55
  56     def run(self, command):
  57         transport = self.client.get_transport()
  58         transport.set_keepalive(1)
  59         chan = transport.open_session()
  60         chan.exec_command(command)
  61         while not chan.exit_status_ready():
  62             if chan.recv_ready():
  63                 data = chan.recv(1024)
  64                 while data:
  65                     log(data.strip())
  66                     data = chan.recv(1024)
  67
  68             if chan.recv_stderr_ready():
  69                 error_buff = chan.recv_stderr(1024)
  70                 while error_buff:
  71                     log(error_buff.strip())
  72                     error_buff = chan.recv_stderr(1024)
  73
  74     def scp_get(self, remote, local='.', dir=False):
  75         try:
  76             with scp.SCPClient(self.client.get_transport()) as _scp:
  77                 _scp.get(remote, local, dir)
  78         except Exception as e:
  79             err(e)
  80
  81     def scp_put(self, local, remote='.', dir=False):
  82         try:
  83             with scp.SCPClient(self.client.get_transport()) as _scp:
  84                 _scp.put(local, remote, dir)
  85         except Exception as e:
  86             err(e)