查找nginx 配置文件中有效的域名

2019独角兽企业重金招聘Python工程师标准>>>

#!/usr/bin/env python
##重庆话语科技---运维部--莲花子
##python/shell
##群号:35666534   欢迎你
import multiprocessing
import re,string,sys,urllib2,urllib,json,time,operator,optparse,os,fnmatch
import socket
menu=optparse.OptionParser()
menu.add_option("-l","--leixin",help="nginx config type")
menu.add_option("-s","--source",help="single source nginx config")
menu.add_option("-p","--path",help="source nginx config path")
opt,argv=menu.parse_args()
def check_s_r(file=None):
                BUF_SIZE = 1024
                domain=[]
                out=[]
                server_name="NO"
                bigfile = open(file,'r')
                tmp_lines = bigfile.readlines(BUF_SIZE)
                while tmp_lines:
                                for line in tmp_lines:
                                        if re.findall(r'(?<=server_name)\s+(.*);',line):
                                                server_name=re.findall(r'(?<=server_name)\s+(.*);',line)[0]
                                                for name in re.split(r'(\s+)',server_name):
                                                        if name.strip():
               domain.append(name.strip())
      return domain
                                tmp_lines = bigfile.readlines(BUF_SIZE)

def chkping(url):
    try: 
        ip=socket.gethostbyname(url)
        if not ip in ['0.0.0.0','127.0.0.1','10.0.0.1']:
            return(url,True)
        else:
            return (url,False)
    except:
            return (url,False)

def find(_path=None,_type=None):
        fuck={}
 pool = multiprocessing.Pool(processes = 7)
        for root, dirs, files in os.walk(_path):
                for f in files:
                        if fnmatch.fnmatch(f,_type):
    try:
     domain=check_s_r(os.path.join(root,f))
     for idomain in domain:
      if not fuck.has_key(f):
       fuck.update({idomain:f})
    except:
     pass
 s=pool.imap(chkping,fuck)
 for i in s:
  if i[1] == False:
   fuck.pop(i[0])
        pool.close()
        pool.join()
        sorted_x=sorted(fuck.iteritems(), key=lambda p:(p[0].split(".")[-2:5],p[0]),reverse=False ) 
        for i in sorted_x:
                        print "{0:-<30}\t{1:_>1}".format(i[0],i[1]);
def check_cmd():
        if opt.path != None and opt.leixin !=None:
                        find(opt.path,opt.leixin)
        elif opt.source != None :
                        check_s_r(opt.source)

check_cmd()

 

转载于:https://my.oschina.net/u/1176204/blog/888585