#!/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)