整个案例运用的知识点:
1.python基本数据结构的使用:列表、字典等
2.python两个模块的使用:os和json
1.os.rename(‘文件1‘,‘文件2‘)
2.json完成自动识别字典、列表,并识别后进行自动转换
3.python函数的定义和调用
4.标志位的灵活运用:
1.flag用于找backend
2.haswrite用于找record和判断是否已经将记录写入到文件
5.python基本语法的使用:
1.for循环
2.if...else....判断
6.python文件的操作的基本使用:
1.with open(‘文件1‘,‘模式‘) as obj1,open(‘文件2‘,‘模式‘) as obj2
整个代码的中心思想:
1.设置2个文件:源文件ha和终文件ha.new(整个过程是:边读边写)
1.源文件ha用于提供读
2.终文件ha.new用于提供写
2.上线、下线的巧妙应用:
1.下线:将源文件ha rename 成备用文件ha.bak
2.上线:将终文件ha.new rename 成新的配置文件ha
配置文件(即ha):
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
代码实现的功能:
1.查找ha文件中指定backend下的后端主机列表信息
2.增加ha文件中指定backend以及backend后端主机列表信息
3.删除ha文件中指定backend以及backend后端主机列表信息
代码:
#!/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"
python学习之路之案例1(修改haproxy的配置文件)
原文地址:http://jachy.blog.51cto.com/10625380/1715829