#!/usr/bin/python
# -*- coding: utf-8 -*-

'''https://www.ibm.com/developerworks/cn/linux/l-cn-pexpect2/index.html'''

import pexpect
import types


username = "root"
ip = "192.168.***.***"
password = "****"
pex = pexpect.spawn('ssh %s@%s' % (username, ip))


def _check(pattern, timeout=120):
  i = pex.expect(pattern, timeout=timeout)
  return i


def sendcr(cmd):
  if pex == None:
    return 0
  n = pex.send("%s\r" % cmd)
  return n


def getexec(cmd):

  child = pexpect.spawn(cmd)
  child.expect(pexpect.EOF)
  return child.before


if __name__ == '__main__':

  checklist1 = [["(", False],
         ["(", False],
         ["(", True],
         # ["(",True],    # hpux
         ["(", True], # windows
         [".+[>$#]\s*$", True], # debian
         ["(", True],
         ["(", False],
         ["(", False],
         ["(", "yes"],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         ["(", False],
         # ["(",terminal_type],
         ["This private key will be ignored.", False],
         ["(", False],
         ["(", "\r"],
         ["(", 'Y'],
         [pexpect.EOF, False],
         [pexpect.TIMEOUT, False],
         ["(", password],
         ["(", password],
         ["(", password],
         ["(", username],
         ["(", username],
         ["(", True]]

  checklist2 = [i[0] for i in checklist1]
  while True:
    i = _check(checklist2)
    print i, checklist1[i], checklist1[i][1]
    if (type(checklist1[i][1]) is types.BooleanType):
      if type(checklist1[i][1]):
        break
    else:
      sendcr(checklist1[i][1])

  cmd = "ls -l /etc/rsyslog.conf"
  result = getexec(cmd)
  print "result", result

26 ['(?i)assword', '***'] ***
5 ['(?i)Last login', True] True
result -rw-r--r--. 1 root root 3167 Mar 13 11:24 /etc/rsyslog.conf