python修改nginx配置文件_python切割nginx配置文件的脚本

#!/usr/bin/env python

import os

import sys

def text_with_color(color,text):

def base(color,text):

color_code = {"black":30,

"red":31,

"green":32,

"yellow":33,

"blue":34,

"white":47

}

return "\033[;%sm%s\033[0m" %(color_code[color],text)

return base(color,text)

def read_nginx_file(file_name):

try:

with open(file_name) as f:

all_content = f.readlines()

except Exception,err:

print "Open nginx config file ERROR,err_msg:",err

sys.exit(3)

temp = all_content.pop()

count = 0

head = []

servers = {}

server_count = 1

while count < len(all_content):

if not all_content[count].strip().replace(" ","") == "server{":

if len(all_content[count].strip()) != 0:

head.append(all_content[count])

count += 1

else:

break

while count < len(all_content):

servers[server_count] = []

servers[server_count].append(all_content[count])

count += 1

while count < len(all_content):

if not all_content[count].strip().replace(" ","") == "server{":

if len(all_content[count].strip()) != 0:

servers[server_count].append(all_content[count])

count += 1

else:

server_count += 1

break

head.extend(["\n"," include conf.d/*.conf","\n","}"])

return head,servers

def write_main_conf_file(old_file_name,config):

os.rename(old_file_name,old_file_name + ".back")

main_file = open("nginx.conf","wb")

for i in config:

main_file.write(i)

main_file.close()

def write_server_conf_file(server):

if not os.path.exists("conf.d"):

os.mkdir("conf.d")

for i in server:

if "server_name" in i.split():

server_file = open("conf.d" + "/" + i.split()[1].strip(";") + ".conf","wb")

for i in server:

server_file.write(i)

server_file.close()

def file_format_clear(config):

count = 0

for i in config:

config[count] = i.strip()+"\n"

count += 1

space = " "

indent_count = 0

count = 0

for i in config:

if "{" in config[count]:

config[count] = space * indent_count + i

indent_count += 1

elif "}" in config[count]:

config[count] = space * (indent_count -1 ) + i

indent_count -= 1

else:

config[count] = space * indent_count + i

count += 1

if __name__ == '__main__':

file_name = "nginx.conf"

head,servers = read_nginx_file(file_name) #load nginx config file

file_format_clear(head) #format clear

for i in servers.keys():

file_format_clear(servers[i])

write_main_conf_file(file_name,head) #write main config file

for i in servers.keys(): #write server config file

write_server_conf_file(servers[i])

print text_with_color("green","Nginx config file cut success.")

print text_with_color("yellow","Please copy nginx.conf and conf.d to your nginx conf directory.")