add ssh modules

pythonmao · pythonmao · commit bf0aef4d93de · 2017-03-02T15:26:16.000+08:00
diff --git a/test_ssh/ssh.py b/test_ssh/ssh.py
@@ -0,0 +1,45 @@
+import paramiko
+import six
+
+
+class SSHClient(object):
+    def __init__(self, host, user, password, port=None, key_contents=None,
+                 key_filename=None, timeout=None):
+        try:
+            self.ssh = paramiko.SSHClient()
+            self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+            self.host = host
+            if key_contents:
+                data = six.StringIO(key_contents)
+                if "BEGIN RSA PRIVATE" in key_contents:
+                    pkey = paramiko.RSAKey.from_private_key(data)
+                elif "BEGIN DSA PRIVATE" in key_contents:
+                    pkey = paramiko.DSSKey.from_private_key(data)
+                else:
+                    # Can't include the key contents - secure material.
+                    raise ValueError("Invalid private key")
+            else:
+                pkey = None
+            self.ssh.connect(host,
+                             username=user,
+                             password=password,
+                             port=port if port else 22,
+                             pkey=pkey,
+                             key_filename=key_filename if not pkey else None,
+                             timeout=timeout if timeout else 0)
+        except Exception as e:
+            print 'error'
+
+    def close(self):
+        if self.ssh:
+            self.ssh.close()
+
+    def exec_command(self, command):
+        stdin, stdout, stderr = self.ssh.exec_command(command)
+        if stderr:
+            print 'command error'
+
+        return stdout
+
+ssh = SSHClient('10.240.198.57', 'root', 'Passw0rd')
+ssh.exec_command('ls /')
diff --git a/test_ssh/ssh_subprocess.py b/test_ssh/ssh_subprocess.py
@@ -0,0 +1,35 @@
+import subprocess
+import signal
+from oslo_log import log
+
+LOG = log.getLogger(__name__)
+
+
+def _subprocess_setup():
+    # Python installs a SIGPIPE handler by default. This is usually not what
+    # non-Python subprocesses expect.
+    signal.signal(signal.SIGPIPE, signal.SIG_DFL)
+
+
+def _subprocess_popen(args, stdin=None, stdout=None, stderr=None, shell=False,
+                      env=None, preexec_fn=_subprocess_setup, close_fds=True):
+    return subprocess.Popen(args, shell=shell, stdin=stdin, stdout=stdout,
+                            stderr=stderr, preexec_fn=preexec_fn,
+                            close_fds=close_fds, env=env)
+
+
+class SSHClient(object):
+    def __init__(self, host, user, passwd, script):
+        self.host = host
+        self.user = user
+        self.passwd = passwd
+        self.script = script
+
+    def exec_command(self, cmd):
+        LOG.info("Exec command: %s", cmd)
+        cmd = self.script + ' ' + self.host + ' ' + self.user \
+              + ' ' + self.passwd + ' ' + '"' + cmd + '"'
+        return _subprocess_popen(cmd, shell=True,
+                                 stdin=subprocess.PIPE,
+                                 stdout=subprocess.PIPE,
+                                 stderr=subprocess.PIPE)
