#!/usr/bin/env python

'''
需要注意的几点问题
1、pxssh类,存在于pexpect模块里,如果没有安装此模块请自行安装,如果已经存在此模块,但是没有pxssh类,经验证,那就是此模块的版本太低。请更新至新版本
2、关于os.system()中的命令,这是在执行一个shell命令,且本身sshpass 是实现在执行scp文件下发的时候是为了自动输入对端机器的账户密码,如果sshpass没有安装,请自行安装。
'''

#!/usr/bin/env python
from pexpect import pxssh
import os

username="root"
passwd="1qaz#EDC"

try:
    for i in range(151,156):
        s = pxssh.pxssh()
        ipaddr= '192.168.100.%s' % i    #生成一个完整的IP地址
        os.environ['ip']=str(ipaddr)    #变量互换,python变量可以在shell里拿到
        s.login(ipaddr,username,passwd)     #登录主机(ip,用户,密码)
        print 'host:%s login complete!' % ipaddr
        s.sendline('mkdir -p /root/.ssh/')    #运行命令,创建一个隐藏ssh目录
        os.system('sshpass -p 1qaz#EDC scp /root/.ssh/authorized_keys root@$ip:/root/.ssh/')    #下发公钥文件:authorized_keys
        print 'host:%s Task running ok!' % ipaddr
        s.prompt()    #匹配系统提示符
        s.logout()    #断开SSH连接
except pxssh.ExceptionPxssh, e:
    print "pxssh failed on login"
    print str(e)