工作需要检查线上所有服务器ngxin的host配置，是否都添加禁止访问目录中带/.svn/和以tar.gz、tar、zip、等结尾url，如果没有则添加，由于线上Nginx服务器将近百台，每台的nginx配置至少10几个，手工检查太慢了，本人也不想浪费太多时间做这些无用功。故用python写了一个检测脚本。来完成这些无聊事情。

    想用python完成这些事情，思路大概为：先备份每台服务器原来的配置，然后遍历每台服务器内所有host配置，正则匹配host配置看是否已经添加相关配置，如有则跳过，遇到有但不全或没有的则在对应的位置上插入相应的配置。除文本插入时我遇到一个问题外，其他比较简单。因为python目前没有对文本进行插入操作的模块。整个插入过程需要自己一步一步实现。

    当时我考虑文本插入思路：把原配置文件载入内存，正则找到要插入的位置，以此中心把原文件分成两部分，开辟新内存空间按顺序先存放第一部内容，然后存放出插入的内容，然后存放第二部内容。最后flush到硬盘。即通俗说法把原配置根据正则进行重定向。

    我了解实现这种重定向有两个方法，第一种: 打开原文件，根据规则重定向到新建的文件上。第二种：使用fileinput对原文件重定向。

    下面是代码，实现过程大概为：先判断目录nginx配置目录，如果存在则备份，然后遍历所有配置文件，匹配每个配置文件是否已添加相关内容，如果没有添加或者之前添加了部分内容，找到配置文件第一个花括号“}”，下一行插入相应的内容。

#!/usr/bin/python#coding:utf-8#Author by Qfeian@20140215"""First backup file and then update all nginx configurein the path "/usr/local/nginx/conf/vhosts".Insert rule is to find the configure of the first "}"after inserting content"""import fileinputimport osimport reimport sysimport stringimport timefrom subprocess import calls_all = """    location ~* ^(.*)\/\.svn\/ {                deny all;        }        location ~* \.(tar|gz|zip|tgz|sh)$ {                deny all;        }"""s_svn = """    location ~* ^(.*)\/\.svn\/ {                deny all;        }"""s_tar = """        location ~* \.(tar|gz|zip|tgz|sh)$ {                deny all;        }"""def file_insert(fname, str):        r = ur'}'        f = open(fname)        old = f.read()        num = int(re.search(r, old).start())        f_input = fileinput.input(fname, inplace=1)        #for line in fileinput.input(fname, inplace=1):        for line in f_input:            if r in line:                print line.rstrip()                print "\n"+str+"\n"                f.seek(num+2)                print f.read()                break            else:                print line.rstrip()        f.close()        f_input.close()        #print "OK! The %s configure has been sucessfull added!" % fname        print "Ok! 配置文件%s已经添加成功！" % fnamedef file_list(f_dir):    #Check the content of configure,and add the correcsponding content    rsvn = ur'\\\/\\.svn\\\/'    rtar = ur'\(tar\|gz\|zip\|tgz\|sh\)'    if os.path.exists(f_dir): #check dir        for f_name in os.listdir(f_dir): #list the dir all configure file           f_path = os.path.join(f_dir,f_name) #Get the filename full path           f1 = open(f_path)           f1_old = f1.read()           f1.close()           if re.findall(rsvn, f1_old) and re.findall(rtar, f1_old):               #print "Notice!  %s have been added ,ignore this configure ...." % f_path               print "Notice! %s 已添加过相关的配置，忽略此配置文件" % f_path               continue           elif re.findall(rsvn, f1_old):               file_insert(f_path, s_tar)           elif re.findall(rtar, f1_old):               file_insert(f_path, s_svn)           else:               file_insert(f_path, s_all)    else:        print "Warnning! dir %s isn't exists!" % f_dirdef file_backup(f_dir):    # If the file has been backed up with a minute before,will not back up    bak_dir = "/data/nginx_config_bak/" + time.strftime('%Y%m%d_%H_%M')    if not os.path.exists(bak_dir):        os.makedirs(bak_dir)        if os.path.exists(f_dir):            cp = "cp -rp"            cmd = "%s %s %s" % (cp, f_dir, bak_dir)            call(cmd, shell=True) #backup the configureif __name__ == "__main__":    f_dir = "/usr/local/nginx/conf/vhosts"    file_backup(f_dir)    file_list(f_dir)    print "-"*90,"\n所有Nginx配置文件更新完成!\n"

参考：