python案例1---修改haproxy的配置文件

运用的知识点：

　　1.python简单数据结构的使用：列表、字典等

　　2.python两个模块的使用：os和json

　　　　　　1.os.rename('文件1'，'文件2')

　　　　　　2.json完成自动识别字典、列表，并识别后进行自动转换

　　3.python函数的定义和调用

　　4.标志位的灵活运用：flag和haswrite分别用于找backend的record和判断是否已经将记录写入到文件

　　5.python基本语法的使用:for循环、if...else....判断

　　6.python文件的操作的基本使用：with open('文件1','模式') as obj1,open('文件2','模式') as obj2

 

　　

代码：

#!/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.oldboy.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.oldboy.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"

转载自：http://www.cnblogs.com/jachy/p/4983246.html