python 实现端口连通性检测

w1w

# -*- coding: utf-8 -*-　　
#!/bin/env python
　　
#AUTHOR:karl
　　
#DATE:2018-1-19
　　
#VERSION:V1.0
　　
######################
　　
import time
　　
import os
　　
import paramiko
　　
import datetime
　　
import sys
　　
import MySQLdb
　　
import threading
　　
#######################################
　　
#首次插入数据是REG=0，更新数据REG为其他
　　
#######################################
　　
REG=1
　　
private_key = paramiko.RSAKey.from_private_key_file('/home/appdeploy/.ssh/id_rsa')
　　
def Data_mysql(info):
　　
    try:
　　
        string=info
　　
        Pip=string.split(" ")[0]
　　
        Sip=string.split(" ")[1]
　　
        Dip=string.split(" ")[2]
　　
        Port=string.split(" ")[3]
　　
        Result=string.split(" ")[4].replace('\n','')
　　
        con = MySQLdb.connect(host='localhost',user="root",passwd="******",db="zabbix",port=3306,charset="utf8")
　　
        dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
　　
        cursor=con.cursor()
　　
        valuse=(dt,Result,Pip,Sip,Dip,Port)
　　
        if REG == 0:
　　
            sql="""INSERT INTO Balant_telnet (create_time,telnet_result,telnet_physicalip,telnet_sourceip,telnet_desip,telnet_port) VALUES (%s,%s,%s,%s,%s,%s) """
　　
            result=cursor.execute(sql,valuse)
　　
        else:
　　
            sql="""UPDATE Balant_telnet set create_time='%s',telnet_result=%s where telnet_physicalip='%s' and telnet_sourceip='%s' and telnet_desip='%s' and telnet_port='%s';""" %(valuse)
　　
            result=cursor.execute(sql)
　　
        cursor.close()
　　
        con.commit()
　　
        con.close()
　　
    except MySQLdb.Error, e:
　　
        print "Error %d: %s" % (e.args[0], e.args[1])
　　
        sys.exit(1)
　　
def check_port(value,content):
　　
    ager={}
　　
    ager[value]=content
　　
    ssh=paramiko.SSHClient()
　　
    try:
　　
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
　　
        ssh.connect(value,port=22,username='appdeploy',pkey=private_key,timeout=5)
　　
        cmd="sh /home/appdeploy/monitor.sh {value}".format(value=ager[ager.keys()[index]])
　　
        stdin,stdout,stderr = ssh.exec_command(cmd)
　　
        for file_d in stdout.readlines():
　　
            Data_mysql(file_d)
　　
    except:
　　
        result=cmd+','+'failed'+'\n'
　　
    ssh.close()
　　
###################################
　　
# 对配置文件进行参数解析
　　
###################################
　　
def get_parameter():
　　
    ager={}
　　
    value=[]
　　
    with open("monitor_config") as context:
　　
        for line in context:
　　
            while not "," in list(line):
　　
                host=line.strip()[1:-1]
　　
                value=[]
　　
                break
　　
            line=line.strip().split(",")
　　
            value=value+line
　　
            ager[str(host)]=value
　　
    return ager
　　
if __name__ == '__main__':
　　
    threads=[]
　　
    parm=get_parameter()
　　
    files=range(len(parm.keys()))
　　
    for index,content in parm.items():
　　
        t = threading.Thread(target=check_port,args=(index,content))
　　
        threads.append(t)
　　
    for index in files:
　　
        threads[index].start()
　　
    for index in files:
　　
        threads[index].join()