python pexpect ssh_python-pexpect_02ssh

#!/usr/bin/env python"""This runs a command on a remote host usingSSH. At the prompts enter hostname,

user, password and the command."""import pexpect

import getpass, os

#user: ssh 主机的用户名

#host：ssh 主机的域名

#password：ssh 主机的密码

#command：即将在远端 ssh 主机上运行的命令

def ssh_command (user, host, password, command):"""This runs a command on the remote host. This could also be done with the

pxsshclass, but this demonstrates what that classdoes at a simpler level.

This returns a pexpect.spawnobject. This handles the case when you tryto

connect to anew host and ssh asks you if you want to accept the publickey

fingerprint andcontinueconnecting.""" ssh_newkey = 'Are you sure you want to continue connecting'# 为 ssh 命令生成一个 spawn 类的子程序对象.

child= pexpect.spawn('ssh -l %s %s %s'%(user, host, command))

i= child.expect([pexpect.TIMEOUT, ssh_newkey, 'password:'])

# 如果登录超时，打印出错信息，并退出.if i == 0: # Timeout

print'ERROR!'print'SSH could not login. Here is what SSH said:'print child.before, child.afterreturnNone

# 如果 ssh 没有publickey，接受它.if i == 1: # SSH does not have the publickey. Just accept it.

child.sendline ('yes')

child.expect ('password:')

i= child.expect([pexpect.TIMEOUT, 'password:'])if i == 0: # Timeout

print'ERROR!'print'SSH could not login. Here is what SSH said:'print child.before, child.afterreturnNone

# 输入密码.

child.sendline(password)returnchild

def main ():

# 获得用户指定 ssh 主机域名.

host= raw_input('Hostname:')

# 获得用户指定 ssh 主机用户名.

user= raw_input('User:')

# 获得用户指定 ssh 主机密码.

password=getpass.getpass()

# 获得用户指定 ssh 主机上即将运行的命令.

command= raw_input('Enter the command:')

child=ssh_command (user, host, password, command)

# 匹配 pexpect.EOF

child.expect(pexpect.EOF)

# 输出命令结果.

print child.beforeif __name__ == '__main__':try:

main()

except Exception, e:

print str(e)

traceback.print_exc()

os._exit(1)