python寫的一個服務器自動巡檢工具

因爲公司服務器需要每半個月巡檢一次，還要將巡檢結果生成圖片貼到巡檢報告中，每次手動操作相當麻煩。恰好剛學python，於是使用python和java寫了一個半自動化的程序，javaq的作用是爲了生成巡檢圖片。python可以自動依次登錄服務器，執行巡檢並在本地生成文本格式的巡檢結果，java讀取這些文本巡檢結果生成圖片。

下面是代碼，歡迎討論。

#!/usr/bin/evn python
#*-* encoding:utf-8 -*-
#Filename:ssh.py
#自動登錄服務器，實現服務器巡檢工作
import os
import sys
import paramiko

#設置一下字符編碼
reload(sys)
sys.setdefaultencoding('utf-8')

#使用public key的登錄服務器，將巡檢結果輸出到特定的目錄中
def login_by_pubkey(serverHost,serverPort,userName,keyFile):
		known_host = "~/.ssh/known_hosts"
	ssh = paramiko.SSHClient();
	ssh.load_system_host_keys(known_host)
	#設置默認接收主機信任的策略，但是可能報告“不信任主機的”異常
	ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

	print 'Connectting host %s......' % serverHost
	ssh.connect(serverHost,serverPort,username = userName,key_filename = keyFile)
	print 'Connect host %s sucess' % serverHost

	fname = '~/xunjian/result_%s' % serverHost
	f = file(fname,'w')
	#執行系統命令，獲取輸出
	stdin, stdout, stderr = ssh.exec_command('df -h')
	#print stdout.readlines()
	f.write('step1:check disk:\n')
	for line in stdout.readlines():
		if len(line) > 0:
			print line
			f.write(line)

	vmstat_stdin,vmstat_stdout,vmstat_stderr = ssh.exec_command('vmstat 2 10')
	f.write('step2:check system:\n')
	for line in vmstat_stdout.readlines():
		if len(line) > 0:
			f.write(line)

	process_stdin,process_stdout,process_stderr = ssh.exec_command('ps -aux | grep java | top 10')
	f.write('step3:check process:\n')
	for line in process_stdout.readlines():
		if len(line) > 0:
			f.write(line)

	#關閉文件和ssh連接
	f.close()
	ssh.close()
	print 'say bye to host %s' % serverHost

	#生成截圖文件(採用Java實現，需要調用本地的Java文件,依賴了commons-io.jar)
	print 'generate image file of %s' % serverHost
	try:
		java_cmd = '/usr/bin/env java -cp commons-io-2.1.jar:img.jar com.*.*.*.CeateCheckPic %s' % fname
		os.system(java_cmd)
	except Exception, e:
		print 'error when generate image file of %s : %s' % (serverHost,e)
	finally:
		print '===generate image file of %s over===' % serverHost

def login_by_prikey():
	pass

if __name__ == '__main__':
	#如果有多個服務器，這個列表中需要配置多條這種配置,實際使用中請將 ip,port,username,public key path替換下面的變量
	ips = ['#ip#,#port#,#user#,#pubkey_path#']

	for ip in ips:
		host,port,user,path = ip.split(',')

		print '==========start %s============' % host
		login_by_pubkey(host,int(port),user,path)
		print '>>>>>>>>>>end %s<<<<<<<<<<<<<<' % host

附件是生成巡檢圖片的一個簡單的jar，大家可以參考一下。

• img.jar (2.1 KB)
• 下載次數: 47