python学习之路之案例1（修改haproxy的配置文件,并实现下线、上线的功能）

global
log 127.0.0.1 local2
daemon
maxconn 256
log 127.0.0.1 local2 info
defaults
log global
mode http
timeout connect 5000ms
timeout client 50000ms
timeout server 50000ms
option  dontlognull

listen stats :8888
stats enable
stats uri       /admin
stats auth      admin:1234

frontend jachy.org
bind 0.0.0.0:80
option httplog
option httpclose
option  forwardfor
log global
acl www hdr_reg(host) -i www.jachy.org
use_backend www.jachy.org if www

backend www.jachy.org
server 100.1.7.9 100.1.7.9 weight 20 maxconn 3000

#!/usr/bin/env python
# -*- coding:utf-8 -*-
#导入2个模块json和os
import  json,os

#定义fetch函数，并同时传入我们指定backend参数，此参数代表backend名
def fetch(backend):
flag = False #定义flag，目的是为了后面是否取可用的backend
fetch_list=[]  #定义空列表，目的是为了后面将取出的backend信息存储在此列表里面

#以读的形式打开'ha'文件,一行一行读，然后将我们指定backend读取到列表里面
#最后返回装有我们指定backend信息的列表
with open('ha') as obj:
for line in obj:
if line.strip() == "backend %s" %(backend):
flag = True
continue
if line.strip().startswith('backend'):
flag=False
if flag and line.strip():
fetch_list.append(line.strip())

return fetch_list

#定义add1函数，并同时传入参数dict_info,此字典参数里面包含我们要传入的server相关信息
def add1(dict_info):

backend_title = dict_info.get("backend") #要插入的backend的名称
context_title = "backend %s" %(backend_title)  #要插入backend整个字段
record_title = dict_info["record"]
#要插入的记录信息
context_record = "server %s %s weight %s maxconn %s" %(record_title["server"],record_title["server"],record_title["weight"],record_title["maxconn"])

#将backend获取，并存储在fetch_list列表里面
fetch_list=fetch(backend_title)

#如果要插入的backend存在
#1.首先设定2个标志：flag和has_write
#flag:用于找到要添加的backend下的内容
#has_write:用于判断fetch_list中的内容是否重新写入到了文件
#2.遍历源文件ha：
#1首先逐行读取，并同时进行判断，
# 1.判断backend是否已经找到，先把backend写入文件，然后找到后将flag设置为True，最后跳出本次循环
# 2.判断找到的backend里面的record是否为空，若为空将flag设置为False
# 3.对flag进行判断
#1.将flag为True的record从fetch_list写入到文件中，在这里面同时要进行是否已经写入的判断，即定义has_write的作用
#2.若flag为False，就直接写入新文件，即将非列表里的内容直接写入

if fetch_list:
flag=False
has_write = False
with open('ha') as read_obj,open('ha.new','w') as write_obj:
for line in read_obj:
if line.strip() == context_title:
write_obj.write("\n"+line)
flag=True
continue
if flag and line.startswith('backend'):
flag = False
if flag:
for new_line in fetch_list:
if not has_write:
temp = "%s%s" %(" "*8,new_line)
write_obj.write(temp)
has_write=True
else:
write_obj.write(line)

else:
#如果要插入的backend不存在
#分2部分写入：
# 1部分.从ha文件里面直接读取，并同时将读取的内容写入新的文件ha.new
# 2部分.将新的backend和context信息，直接写到新的文件ha.new的文件尾部
with open('ha') as read_obj,open('ha.new','w') as write_obj:
for line in read_obj:
write_obj.write(line)
write_obj.write("\n"+context_title+"\n")
temp=" "*8+context_record+"\n"
write_obj.write(temp)

#将原文件ha改名备用文件为ha.bak，即将ha下线
#将新文件ha.new改名为线上文件ha，即将ha.new上线
os.rename('ha','ha.bak')
os.rename('ha.new','ha')

#删除函数
def delete(dict_info):
#1.首先获取用户的记录并以列表的形式存储，以便于我们调用
del_backend = dict_info["backend"]
del_record = dict_info["record"]

context_title = "backend %s" %(del_backend)
context_record = "server %s %s weight %s maxconn %s" %(del_record["server"],del_record["server"],del_record["weight"],del_record["maxconn"])

#调用上面定义的fetch()函数，得到包含backend内容的列表
fetch_list = fetch(del_backend)
#判断1：若列表为空：直接返回
if not fetch_list:
return
#判断2：否则不为空
else:
#判断1：若用户输入的server信息不在fetch_list里面，直接返回
if context_record not in fetch_list:
print "your server message is not exists"
return
#判断2：若输入的server在fetch_list里面，就将此server信息从列表中移除
else:
fetch_list.remove(context_record)

#逐行从ha文件中读取，并同时进行判断
#核心思想：
#首先边读取、边判断、边写入，
# 判断其实就是我们自己输入的backend，找到后将列表fetch_list信息写入到新的文件里面。
# 其余的非列表的内容直接从源文件ha写到新文件ha.new即可
with open('ha','r') as read_obj,open('ha.new','w') as write_obj:
flag = False
has_write = False
for line in read_obj:
if line.strip() == context_title:
write_obj.write(line)
flag = True
continue
if flag and line.startswith('backend'):
flag = False
if flag:
if not has_write:
print fetch_list
for new_line in fetch_list:
temp = "%s%s\n" %(" "*8,new_line)
write_obj.write(temp)
has_write = True
else:
write_obj.write(line)

#将原文件ha改名备用文件为ha.bak，即将ha下线
#将新文件ha.new改名为线上文件ha，即将ha.new上线
os.rename('ha','ha.bak')
os.rename('ha.new','ha')

"""
s='{"backend":"ttt.jachy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}'
data_info=json.loads(s)
add1(data_info)
#delete(data_info)
"""

#主函数
if __name__ == "__main__":
#获取用户的操作选择
print u"1.获取记录\n2.添加记录\n3.删除记录\n"
select_num=raw_input("plz input number:")

#根据用户的选择，进行调用函数
if select_num == "1":
backend = raw_input("plz input backend:")
fetch_list = fetch(backend)
for i in fetch_list:
print i
else:
print "plz input like this example:"
print '{"backend":"ttt.jachy.org","record":{"server":"100.1.7.9","weight":"20","maxconn":"3000"}}'
data_str = raw_input("plz input record>>>")
data_dict = json.loads(data_str)
if select_num == "2":
add1(data_dict)
elif select_num == "3":
delete(data_dict)
else:
print "input error"