w1w 发表于 2018-8-7 10:08:22

python 实现端口连通性检测

# -*- 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(" ")
  
      Sip=string.split(" ")
  
      Dip=string.split(" ")
  
      Port=string.split(" ")
  
      Result=string.split(" ").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, e.args)
  
      sys.exit(1)
  
def check_port(value,content):
  
    ager={}
  
    ager=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])
  
      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()
  
                value=[]
  
                break
  
            line=line.strip().split(",")
  
            value=value+line
  
            ager=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.start()
  
    for index in files:
  
      threads.join()
页: [1]
查看完整版本: python 实现端口连通性检测