python脚本监控磁盘空间
写了个python小程序，监控磁盘空间，前面部分网上也有很多，写博客的目的是记录下来，已供自己后面使用，思路就是用pexpect 这个模块，ssh到不同的机器上，查到磁盘空间，最后对查到的结果进行处理，然后存到mysql数据中。以下是代码：
#coding=utf8
import pexpect
import getpass,os,sys
import re,datetime, time
def ssh_command (user, host, password, command):
ssh_newkey = 'Are you sure you want to continue connecting'
child = pexpect.spawn('ssh -l %s %s %s'%(user, host, command))
i = child.expect([pexpect.TIMEOUT, ssh_newkey, 'password: '])
if i == 0: # Timeout
print child.before, child.after
return None
# 如果 ssh 没有 public key，接受它.
if i == 1: # SSH does not have the public key. Just accept it.
child.sendline ('yes')
child.expect ('password: ')
i = child.expect([pexpect.TIMEOUT, 'password: '])
if i == 0: # Timeout
print child.before, child.after
return None
# 输入密码.
child.sendline(password)
return child
def main (date):
base='ip文件'
wbase='保存文件名'
file=open(base)
f=open(wbase,"w")
#raw_command = raw_input("输入命令: ")
while 1:
lines = file.readlines(10000)
if not lines:
break
for line in lines:
serverM=line.split(',')
child = ssh_command (serverM[0],serverM[1],serverM[2], 'df -h')
# 匹配 pexpect.EOF
child.expect(pexpect.EOF)
# 输出命令结果.
#print serverM[1]+" 磁盘空间如下:"
#print child.before
f.write(serverM[1])
f.write(child.before)
f.close()
file.close()
readLine(wbase,date)
def readLine(base,date):
file = open(base)
a=[]
while 1:
lines = file.readlines(10000)
if not lines:
break
for line in lines:
ob=re.search('\d*[.]\d*[.]\d*[.]\d*',line)
oe=re.search('/export',line)
if ob:
a.append(ob.group(0))
if oe:
a.append(line[len(line)-13:len(line)-10])
a.append(line[len(line)-19:len(line)-14])
a.append(line[len(line)-25:len(line)-20])
file.close()
mysql=""" " """ %date
os.system(mysql)
i= len(a)/4
while i>0:
i=i-1
mysql1=""" mysql -hip -P3358 -u用户名-p密码 -D数据库 -e "insert into 表名 values(%s,'%s','%s','%s','%s')" """ %(date,a[0+4*i],a[3+4*i],a[2+4*i],a[1+4*i])
#print a[0+4*i]
#print a[1+4*i]
#print a[2+4*i]
#print mysql
os.system(mysql1)
if __name__ == '__main__':
if (len(sys.argv) > 1):
y_date = sys.argv[1]
else:
y_date = (datetime.datetime.now() - datetime.timedelta(days=1)).strftime("%Y%m%d")
try:
main(y_date)
except Exception, e:
print str(e)
traceback.print_exc()
os._exit(1)