python(pexpect)自动ssh脚本
python(pexpect)简单的自动ssh脚本
安装python的pexpect模块
# apt-get install python-pip
# pip install pexpect
# vi ssh.py
按a或i进入编辑模式
#!/usr/bin/python
#/usr/bin/python
import pexpect
child=pexpect.spawn('ssh [email protected]',timeout=5)
ssh_newkey='Are you sure you want to continue connecting'
Down='No route to host'
i=child.expect([ssh_newkey,'password:',pexpect.EOF,pexpect.TIMEOUT,Down]) （下面的0、1、2、3分别对应“[]”里的序列）
if i == 0:
child.sendline('yes')
child.sendline('123456')
child.sendline('exit')
if i == 1:
child.sendline('123456')
child.sendline('exit')
if i == 2:
f=open('/log/ssh/ssh.log','a')
f.write('10.0.0.22 ssh(nis) is Bad')
f.write('\n')
f.close()
if i == 3:
f=open('/log/ssh/ssh.log','a')
f.write('10.0.0.22 ssh(nis) is Bad')
f.write('\n')
f.close()
if i == 4:
f=open('/log/ssh/ssh.log','a')
f.write('10.0.0.22 ssh(nis) is down')
f.write('\n')
f.close()
child.interact()
child.close()
按Esc键退出编辑模式
:wq（保存并退出）
# python ssh.py
脚本解释：
1、如果第1次连接，输入yes，然后再输入密码，如果不是第1次连接，输入密码
2、EOF表示服务器（ip）不存在或非linux类，TIMEOUT表连接超时
3、如果如果找不到该服务器或连接超时，将信息追加写入/log/ssh/ssh.log（建议第1台覆盖，后面追加，r为读，w为写即覆盖，a为追加）
4、if的下面内容前要缩进一格
再配合bash脚本
# vi mvlog.sh (因为python里log文件名是死的，可以借助bash让其更灵活)
#!/bin/bash
Date=`date +%Y%m%d`
Date2=`date +%Y%m%d -d "1 day ago"`
Date3=`date +%Y%m%d -d "2 day ago"`
Ssh=/log/ssh
if [ -f $Ssh/ssh_$Date2.log ];then
/bin/rm -rf /log/ssh/ssh_$Date3.log
else
/bin/mv /log/ssh/ssh.log /log/ssh/ssh_$Date2.log
/usr/bin/touch /log/ssh/ssh.log
fi
:wq
# vi ssh.sh（这个主要根据log文件内容及更新时间判断有服务器报警并发邮件通知）
#!/bin/bash
Date=`date +%H:%M`
Hour=`date +%H`
Hourr=`date +%M`
Size=`/bin/ls -l /log/ssh/ | grep ssh.log | awk -F[:" "]+ '{print $5}'`
Hour2=`/bin/ls -l /log/ssh/ | grep ssh.log | awk -F[:" "]+ '{print $8}'`
Hour22=`/bin/ls -l /log/ssh/ | grep ssh.log | awk -F[:" "]+ '{print $9}'`
if [ $Size -ge "4" ] && [ $Hour = "$Hour2" ] && [ $Hourr = "$Hour22" ]
then
/bin/echo "time $Date" >> /log/ssh/ssh.log
/usr/bin/mail -s "Android ssh(nis) is bad at $Date" [email protected] < /log/ssh/ssh.log
/usr/bin/mail -s "Android ssh(nis) is bad" [email protected] < /log/ssh/ssh.log
fi